lib/kestrel/client.rb in kestrel-client-0.7.0 vs lib/kestrel/client.rb in kestrel-client-0.7.1

@@ -160,12 +160,11 @@
       if @shuffle &&
           (@counter > 0 && key != @current_queue) ||
           @counter >= @gets_per_server
         @counter = 0
         @current_queue = key
-        @read_client.quit
-        @read_client.set_servers(servers[rand(@server_count)])
+        @read_client.reset(servers[rand(@server_count)])
       else
         @counter +=1
       end
     end
 
@@ -184,14 +183,14 @@
     def with_retries #:nodoc:
       yield
     rescue *RECOVERABLE_ERRORS
       tries ||= @exception_retry_limit + 1
       tries -= 1
+      @write_client.reset(servers[rand(@server_count)])
+
       if tries > 0
         retry
       else
-        @write_client.quit
-        @write_client.set_servers(servers[rand(@server_count)])
         raise
       end
     end
 
   end