Use retry instead of while loop for rescuing connection failure (quasor)

docs/REPLICA_SETS.md

@@ -38,18 +38,14 @@ every half second and time out after thirty seconds.
 
     # Ensure retry upon failure
     def rescue_connection_failure(max_retries=60)
-        success = false
       retries = 0
-        while !success
       begin
         yield
-            success = true
       rescue Mongo::ConnectionFailure => ex
         retries += 1
-            raise ex if retries >= max_retries
+        raise ex if retries > max_retries
         sleep(0.5)
-          end
-        end
+        retry
       end
     end
 

test/replica_sets/rs_test_helper.rb

@@ -12,17 +12,15 @@ class Test::Unit::TestCase
   # Generic code for rescuing connection failures and retrying operations.
   # This could be combined with some timeout functionality.
   def rescue_connection_failure(max_retries=60)
-    success = false
-    tries   = 0
-    while !success && tries < max_retries
+    retries = 0
     begin
       yield
-        success = true
-      rescue Mongo::ConnectionFailure
-        puts "Rescue attempt #{tries}\n"
-        tries += 1
+    rescue Mongo::ConnectionFailure => ex
+      puts "Rescue attempt #{retries}"
+      retries += 1
+      raise ex if retries > max_retries
       sleep(1)
-      end
+      retry
     end
   end