Add synchronous refresh; fix connection leak.

lib/mongo/repl_set_connection.rb

@@ -200,28 +200,21 @@ module Mongo
       @max_bson_size = manager.max_bson_size
     end
 
-    # If ismaster doesn't match our current view
+    # Refresh the current replica set configuration.
-    # then create a new PoolManager, passing in our
+    def refresh(opts={})
-    # existing view. It should be able to do the diff.
+      return false if !connected?
-    # Then take out the connection lock and replace
-    # our current values.
-    def refresh
-      return if !connected?
-
-      if !Thread.current[:background]
-        Thread.current[:background] = PoolManager.new(self, @seeds)
-      end
-
-      background_manager = Thread.current[:background]
 
       # Return if another thread is already in the process of refreshing.
       return if sync_exclusive?
 
       sync_synchronize(:EX) do
         log(:debug, "Refreshing...")
-        background_manager.connect
+        @background_manager ||= PoolManager.new(self, @seeds)
-        update_config(background_manager)
+        @background_manager.connect
+        update_config(@background_manager)
       end
+
+      return true
     end
 
     def connected?
@@ -281,7 +274,7 @@ module Mongo
 
         if @nodes
           @nodes.each do |member|
-            member.disconnect
+            member.close
           end
         end
 
@@ -339,13 +332,16 @@ module Mongo
     private
 
     def initiate_auto_refresh
-      return unless @auto_refresh
+      if @auto_refresh
-      return if @refresh_thread && @refresh_thread.alive?
+        return if @refresh_thread && @refresh_thread.alive?
-      @refresh_thread = Thread.new do
+        @refresh_thread = Thread.new do
-        while true do
+          while true do
-          sleep(@refresh_interval)
+            sleep(@refresh_interval)
-          refresh
+            refresh
+          end
         end
+      else
+        @last_refresh = Time.now
       end
     end
 
@@ -444,6 +440,11 @@ module Mongo
           socket.close
         end
       end
+
+      if !@auto_refresh &&
+        ((Time.now - @last_refresh) > @refresh_interval)
+        refresh
+      end
     end
   end
 end

lib/mongo/util/node.rb

@@ -20,10 +20,6 @@ module Mongo
     end
     alias :== :eql?
 
-    def close
-      self.socket.close if self.socket
-    end
-
     def host_string
       address
     end
@@ -60,7 +56,7 @@ module Mongo
       self.socket = socket
     end
 
-    def disconnect
+    def close
       if self.socket
         self.socket.close
         self.socket = nil

lib/mongo/util/pool.rb

@@ -53,7 +53,7 @@ module Mongo
 
     def close
       @connection_mutex.synchronize do
-        (@sockets - @checked_out).each do |sock|
+        @sockets.each do |sock|
           begin
             sock.close
           rescue IOError => ex
@@ -140,6 +140,7 @@ module Mongo
       rescue => ex
         socket.close if socket
         raise ConnectionFailure, "Failed to connect to host #{@host} and port #{@port}: #{ex}"
+        @node.close if @node
       end
 
       # If any saved authentications exist, we want to apply those

lib/mongo/util/pool_manager.rb

@@ -9,6 +9,7 @@ module Mongo
       @connection = connection
       @seeds = seeds
       @refresh_node = nil
+      @previously_connected = false
     end
 
     def inspect
@@ -16,16 +17,28 @@ module Mongo
     end
 
     def connect
+      if @previously_connected
+        close
+      end
+
       initialize_data
       members = connect_to_members
       initialize_pools(members)
       update_seed_list(members)
+
       @members = members
+      @previously_connected = true
     end
 
-    private
+    def healthy?
+      if !@refresh_node || !refresh_node.set_config
+        return false
+      end
 
-    def initialize_data
+      #if refresh_node.node_list
+    end
+
+    def close
       begin
         if @primary_pool
           @primary_pool.close
@@ -45,7 +58,11 @@ module Mongo
 
         rescue ConnectionFailure
       end
+    end
 
+    private
+
+    def initialize_data
       @primary = nil
       @primary_pool = nil
       @read_pool = nil
@@ -71,6 +88,7 @@ module Mongo
           members << node
         end
       end
+      seed.close
 
       if members.empty?
         raise ConnectionFailure, "Failed to connect to any given member."
@@ -174,7 +192,7 @@ module Mongo
         if node.connect && node.set_config
           return node
         else
-          node.disconnect
+          node.close
         end
       end