qu-mongo 0.1.2 → 0.1.3

data/lib/qu/backend/mongo.rb

@@ -3,6 +3,18 @@ require 'mongo'
 module Qu
   module Backend
     class Mongo < Base
+
+      # Number of times to retry connection on connection failure (default: 5)
+      attr_accessor :max_retries
+
+      # Seconds to wait before try to reconnect after connection failure (default: 1)
+      attr_accessor :retry_frequency
+
+      def initialize
+        self.max_retries = 5
+        self.retry_frequency = 1
+      end
+
       def connection
         @connection ||= begin
           uri = URI.parse(ENV['MONGOHQ_URL'].to_s)
@@ -118,8 +130,23 @@ module Qu
       end
 
       def [](name)
-        database["qu.#{name}"]
+        rescue_connection_failure do
+          database["qu:#{name}"]
+        end
       end
+
+      def rescue_connection_failure
+        retries = 0
+        begin
+          yield
+        rescue ::Mongo::ConnectionFailure => ex
+          retries += 1
+          raise ex if retries > max_retries
+          sleep(retry_frequency * retries)
+          retry
+        end
+      end
+
     end
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: qu-mongo
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 29
   prerelease: 
   segments: 
   - 0
   - 1
-  - 2
-  version: 0.1.2
+  - 3
+  version: 0.1.3
 platform: ruby
 authors: 
 - Brandon Keepers
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-10-05 00:00:00 -04:00
+date: 2011-10-10 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -40,12 +40,12 @@ dependencies:
     requirements: 
     - - "="
       - !ruby/object:Gem::Version 
-        hash: 31
+        hash: 29
         segments: 
         - 0
         - 1
-        - 2
-        version: 0.1.2
+        - 3
+        version: 0.1.3
   type: :runtime
   version_requirements: *id002
 - !ruby/object:Gem::Dependency