RUBY-386 extend ismaster timeout to Connection

shingara · Dec 13, 2011 · 8fd89eb · 8fd89eb
1 parent c308f9b
commit 8fd89eb
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lib/mongo/connection.rb b/lib/mongo/connection.rb
@@ -615,7 +615,13 @@ def check_is_master(node)
           socket.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
         end
 
-        config = self['admin'].command({:ismaster => 1}, :socket => socket)
+        if @connect_timeout
+          Mongo::TimeoutHandler.timeout(@connect_timeout, OperationTimeout) do
+            config = self['admin'].command({:ismaster => 1}, :socket => socket)
+          end
+        else
+          config = self['admin'].command({:ismaster => 1}, :socket => socket)
+        end
       rescue OperationFailure, SocketError, SystemCallError, IOError => ex
         close
       ensure