multithink 0.0.2 → 0.0.3

data/lib/multithink/connection.rb

@@ -6,20 +6,44 @@ include RethinkDB::Shortcuts
 
 class MultiThink::Connection
 
-  def initialize(servers)
+  DEFAULTS = {retries: 10}
+
+  def initialize(servers, options = {})
     @servers = servers
+    options = DEFAULTS.merge(options)
+    @retries = options.fetch(:retries)
     connect
   end
 
   def connect
     # TODO try all servers until we get a connection
     options = @servers.first
-    @conn = r.connect(options)
+    begin
+      @conn = r.connect(options)
+    rescue
+      @tried ||= 0
+      sleep 1
+      retry if (@tried += 1) < @retries
+    end
   end
 
   def run(query)
     # TODO handle connection failure
-    query.run(@conn)
+    begin
+      query.run(@conn)
+    rescue RuntimeError => e
+      reconnect
+    end
+  end
+
+  def reconnect
+    begin
+      # try fast path first
+      @conn.reconnect
+    rescue
+      # if that fails then try get a new connection
+      connect
+    end
   end
 
 end

data/lib/multithink/version.rb

@@ -1,3 +1,3 @@
 class MultiThink
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: multithink
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors: