github-migration/mongo-ruby-driver
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added thread-local sockets to ReplSetConnection

Browse Source
This commit is contained in:
Kyle Banker 2011-11-03 18:48:01 -04:00
parent 54433502f5
commit 01d1231373
1 changed files with 22 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
lib/mongo/repl_set_connection.rb
View File

@ -304,33 +304,36 @@ module Mongo
end end
def get_local_reader def get_local_reader
Thread.current[:connections] ||= {} self.connections ||= {}
Thread.current[:connections][self.object_id] ||= {} self.connections[self.object_id] ||= {}
Thread.current[:connections][self.object_id][:version] ||= self.refresh_version self.connections[self.object_id][:reader] ||= checkout_reader
Thread.current[:connections][self.object_id][:reader] ||= checkout_reader
end end
def get_local_writer def get_local_writer
Thread.current[:connections] ||= {} self.connections ||= {}
Thread.current[:connections][self.object_id] ||= {} self.connections[self.object_id] ||= {}
Thread.current[:connections][self.object_id][:version] ||= self.refresh_version self.connections[self.object_id][:writer] ||= checkout_writer
Thread.current[:connections][self.object_id][:writer] ||= checkout_writer
end end
# Used to close, check in, or refresh sockets held # Used to close, check in, or refresh sockets held
# in thread-local variables. # in thread-local variables.
def local_socket_done def local_socket_done(socket)
if Thread.current[:connections][self.object_id][:version] != self.refresh_version puts "Done. Threads: #{Thread.list.size}, pool_size: #{self.pool_size}"
checkin(Thread.current[:connections][self.object_id][:reader]) if self.connections[self.object_id][:reader] == socket
Thread.current[:connections][self.object_id][:reader] ||= checkout_reader if self.read_pool.sockets_low?
end puts "***SOCKETS ARE LOW! READER****"
end checkin(socket)
self.connections[self.object_id][:reader] = nil
end
end
def checkin_sockets if self.connections[self.object_id][:writer] == socket
checkin(Thread.current[:connections][self.object_id][:reader]) if self.primary_pool && self.primary_pool.sockets_low?
checkin(Thread.current[:connections][self.object_id][:writer]) puts "***SOCKETS ARE LOW! WRITER****"
Thread.current[:connections][self.object_id][:writer] = nil checkin(socket)
Thread.current[:connections][self.object_id][:reader] = nil self.connections[self.object_id][:writer] = nil
end
end
end end
# Checkout a socket for reading (i.e., a secondary node). # Checkout a socket for reading (i.e., a secondary node).