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

Remove thread-local map and socket map (complexity creep).

This commit is contained in:
Kyle Banker 2011-11-18 15:47:06 -05:00
parent 177fad34ff
commit fa10508f07
11 changed files with 142 additions and 298 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
lib/mongo/connection.rb
View File

@ -481,68 +481,6 @@ module Mongo
@max_bson_size @max_bson_size
end end
def get_local_reader
self.connections ||= {}
if !connected? && self.connections[self.object_id]
self.connections[self.object_id]
else
self.connections[self.object_id] = {}
end
self.connections[self.object_id][:reader] ||= checkout_reader
end
def get_local_writer
self.connections ||= {}
if !connected? && self.connections[self.object_id]
self.connections[self.object_id]
else
self.connections[self.object_id] = {}
end
self.connections[self.object_id][:writer] ||= checkout_writer
end
# Allow the current thread’s connection to return to the pool.
#
# Calling this method allows the socket that has been reserved
# for this thread to be returned to the pool. Other threads will
# then be able to re-use that socket. If your application uses many
# threads, or has long-running threads that infrequently perform MongoDB
# operations, then judicious use of this method can lead to performance gains.
# Care should be taken, however, to make sure that end_request is not called
# in the middle of a sequence of operations in which ordering is important. This
# could lead to unexpected results.
#
# One important case is when a thread is dying permanently. It is best to call
# end_request when you know a thread is finished, as otherwise its socket will
# not be reclaimed.
def end_request
if socket = self.connections[self.object_id][:reader]
checkin(socket)
end
if socket = self.connections[self.object_id][:writer]
checkin(socket)
end
end
# Used to close, check in, or refresh sockets held
# in thread-local variables.
def local_socket_done(socket)
if self.connections[self.object_id][:reader] == socket
if self.read_pool.sockets_low?
checkin(socket)
self.connections[self.object_id][:reader] = nil
end
end
if self.connections[self.object_id][:writer] == socket
if self.primary_pool && self.primary_pool.sockets_low?
checkin(socket)
self.connections[self.object_id][:writer] = nil
end
end
end
# Checkout a socket for reading (i.e., a secondary node). # Checkout a socket for reading (i.e., a secondary node).
# Note: this is overridden in ReplSetConnection. # Note: this is overridden in ReplSetConnection.
def checkout_reader def checkout_reader
@ -560,16 +498,12 @@ module Mongo
# Checkin a socket used for reading. # Checkin a socket used for reading.
# Note: this is overridden in ReplSetConnection. # Note: this is overridden in ReplSetConnection.
def checkin_reader(socket) def checkin_reader(socket)
warn "Connection#checkin_writer is not deprecated and will be removed " +
"in driver v2.0. Use Connection#checkin instead."
checkin(socket) checkin(socket)
end end
# Checkin a socket used for writing. # Checkin a socket used for writing.
# Note: this is overridden in ReplSetConnection. # Note: this is overridden in ReplSetConnection.
def checkin_writer(socket) def checkin_writer(socket)
warn "Connection#checkin_writer is not deprecated and will be removed " +
"in driver v2.0. Use Connection#checkin instead."
checkin(socket) checkin(socket)
end end

16
lib/mongo/cursor.rb
View File

@ -520,10 +520,10 @@ module Mongo
def checkout_socket_from_connection def checkout_socket_from_connection
@checkin_connection = true @checkin_connection = true
if @command || @read_preference == :primary if @command || @read_preference == :primary
@connection.get_local_writer @connection.checkout_writer
else else
@read_pool = @connection.read_pool @read_pool = @connection.read_pool
@connection.get_local_reader @connection.checkout_reader
end end
end end
@ -556,7 +556,11 @@ module Mongo
@read_pool.checkin(sock) @read_pool.checkin(sock)
@checkin_read_pool = false @checkin_read_pool = false
elsif @checkin_connection elsif @checkin_connection
@connection.local_socket_done(sock) if @command || @read_preference == :primary
@connection.checkin_writer(sock)
else
@connection.checkin_reader(sock)
end
@checkin_connection = false @checkin_connection = false
end end
end end
@ -566,7 +570,11 @@ module Mongo
@read_pool.checkin(sock) @read_pool.checkin(sock)
@checkin_read_pool = false @checkin_read_pool = false
else else
@connection.checkin(sock) if @command || @read_preference == :primary
@connection.checkin_writer(sock)
else
@connection.checkin_reader(sock)
end
@checkin_connection = false @checkin_connection = false
end end
end end

48
lib/mongo/networking.rb
View File

@ -25,21 +25,23 @@ module Mongo
connection = opts.fetch(:connection, :writer) connection = opts.fetch(:connection, :writer)
begin
add_message_headers(message, operation) add_message_headers(message, operation)
packed_message = message.to_s packed_message = message.to_s
if connection == :writer if connection == :writer
sock = get_local_writer sock = checkout_writer
else else
sock = get_local_reader sock = checkout_reader
end end
begin
send_message_on_socket(packed_message, sock) send_message_on_socket(packed_message, sock)
local_socket_done(sock) ensure
rescue ConnectionFailure, OperationFailure, OperationTimeout => ex if connection == :writer
checkin(sock) checkin_writer(sock)
raise ex else
checkin_reader(sock)
end
end end
end end
@ -64,13 +66,13 @@ module Mongo
last_error_id = add_message_headers(last_error_message, Mongo::Constants::OP_QUERY) last_error_id = add_message_headers(last_error_message, Mongo::Constants::OP_QUERY)
packed_message = message.append!(last_error_message).to_s packed_message = message.append!(last_error_message).to_s
sock = checkout_writer
begin begin
sock = get_local_writer
send_message_on_socket(packed_message, sock) send_message_on_socket(packed_message, sock)
docs, num_received, cursor_id = receive(sock, last_error_id) docs, num_received, cursor_id = receive(sock, last_error_id)
local_socket_done(sock) checkin_writer(sock)
rescue ConnectionFailure, OperationFailure, OperationTimeout => ex rescue ConnectionFailure, OperationFailure, OperationTimeout => ex
checkin(sock) checkin_writer(sock)
raise ex raise ex
end end
@ -101,17 +103,14 @@ module Mongo
read=:primary, exhaust=false) read=:primary, exhaust=false)
request_id = add_message_headers(message, operation) request_id = add_message_headers(message, operation)
packed_message = message.to_s packed_message = message.to_s
begin
if socket if socket
sock = socket sock = socket
should_checkin = false should_checkin = false
else else
if command if command || read == :primary
sock = get_local_writer sock = checkout_writer
elsif read == :primary
sock = get_local_writer
elsif read == :secondary elsif read == :secondary
sock = get_local_reader sock = checkout_reader
else else
sock = checkout_tagged(read) sock = checkout_tagged(read)
end end
@ -119,12 +118,19 @@ module Mongo
end end
result = '' result = ''
begin
send_message_on_socket(packed_message, sock) send_message_on_socket(packed_message, sock)
result = receive(sock, request_id, exhaust) result = receive(sock, request_id, exhaust)
local_socket_done(sock) if should_checkin ensure
rescue ConnectionFailure, OperationFailure, OperationTimeout => ex if should_checkin
checkin(sock) if should_checkin if command || read == :primary
raise ex checkin_writer(sock)
elsif read == :secondary
checkin_reader(sock)
else
# TODO: sock = checkout_tagged(read)
end
end
end end
result result
end end
@ -281,7 +287,7 @@ module Mongo
total_bytes_sent total_bytes_sent
rescue => ex rescue => ex
close close
raise ConnectionFailure, "Operation failed with the following exception: #{ex}" raise ConnectionFailure, "Operation failed with the following exception: #{ex}:#{ex.message}"
end end
end end

150
lib/mongo/repl_set_connection.rb
View File

@ -105,6 +105,7 @@ module Mongo
# No connection manager by default. # No connection manager by default.
@manager = nil @manager = nil
@pool_mutex = Mutex.new
if ![:sync, :async, false].include?(@refresh_mode) if ![:sync, :async, false].include?(@refresh_mode)
raise MongoArgumentError, raise MongoArgumentError,
@ -129,7 +130,6 @@ module Mongo
@refresh_version = 0 @refresh_version = 0
# Maps # Maps
@sockets_to_pools = {}
@threads_to_sockets = Hash.new { |h, k| h[k] = Hash.new } @threads_to_sockets = Hash.new { |h, k| h[k] = Hash.new }
@tag_map = nil @tag_map = nil
@ -264,13 +264,7 @@ module Mongo
# Close the connection to the database. # Close the connection to the database.
def close def close
@connected = false @connected = false
@manager.close(:soft => true) if @manager
if @refresh_thread
@refresh_thread = nil
end
@manager.close if @manager
@sockets_to_pools.clear
@threads_to_sockets.clear @threads_to_sockets.clear
end end
@ -308,95 +302,6 @@ module Mongo
end end
end end
def get_local_reader
self.connections ||= {}
if !connected? && self.connections[self.object_id]
self.connections[self.object_id]
else
self.connections[self.object_id] = {}
end
socket = self.connections[self.object_id][:reader] ||= checkout_reader
if self.read_pool != @sockets_to_pools[socket]
checkin(socket)
socket = self.connections[self.object_id][:reader] = checkout_reader
end
@threads_to_sockets[Thread.current][:reader] = socket
end
def get_local_writer
self.connections ||= {}
if !connected? && self.connections[self.object_id]
self.connections[self.object_id]
else
self.connections[self.object_id] = {}
end
socket = self.connections[self.object_id][:writer] ||= checkout_writer
if self.primary_pool != @sockets_to_pools[socket]
checkin(socket)
socket = self.connections[self.object_id][:writer] = checkout_writer
end
@threads_to_sockets[Thread.current][:writer] = socket
end
# Allow the current thread’s connection to return to the pool.
#
# Calling this method allows the socket that has been reserved
# for this thread to be returned to the pool. Other threads will
# then be able to re-use that socket. If your application uses many
# threads, or has long-running threads that infrequently perform MongoDB
# operations, then judicious use of this method can lead to performance gains.
# Care should be taken, however, to make sure that end_request is not called
# in the middle of a sequence of operations in which ordering is important. This
# could lead to unexpected results.
#
# One important case is when a thread is dying permanently. It is best to call
# end_request when you know a thread is finished, as otherwise its socket will
# not be reclaimed.
def end_request
if socket = self.connections[self.object_id][:reader]
checkin(socket)
end
if socket = self.connections[self.object_id][:writer]
checkin(socket)
end
end
# Used to close, check in, or refresh sockets held
# in thread-local variables.
def local_socket_done(socket)
if self.connections[self.object_id][:reader] == socket
if self.read_pool.sockets_low? ||
self.read_pool != @sockets_to_pools[socket]
checkin(socket)
self.connections[self.object_id][:reader] = nil
end
end
if self.connections[self.object_id][:writer] == socket
if self.primary_pool &&
(self.primary_pool.sockets_low? ||
self.primary_pool != @sockets_to_pools[socket])
checkin(socket)
self.connections[self.object_id][:writer] = nil
end
end
if (Time.now - @last_cleanup) > CLEANUP_INTERVAL &&
@cleanup_lock.try_lock
@threads_to_sockets.each do |thread, sockets|
if !thread.alive?
checkin(sockets[:reader])
checkin(sockets[:writer])
@threads_to_sockets.delete(thread)
end
end
@cleanup_lock.unlock
end
end
# Checkout a socket for reading (i.e., a secondary node). # Checkout a socket for reading (i.e., a secondary node).
# Note that @read_pool might point to the primary pool # Note that @read_pool might point to the primary pool
# if no read pool has been defined. # if no read pool has been defined.
@ -433,25 +338,21 @@ module Mongo
end end
end end
def checkin(socket) # Checkin a socket used for reading.
if pool = @sockets_to_pools[socket] def checkin_reader(socket)
pool.checkin(socket) if !self.read_pool.checkin(socket) &&
if !@sockets_to_pools[socket] !self.primary_pool.checkin(socket)
close_socket(socket) close_socket(socket)
end end
elsif socket sync_refresh
end
# Checkin a socket used for writing.
def checkin_writer(socket)
if !self.primary_pool.checkin(socket)
close_socket(socket) close_socket(socket)
end end
sync_refresh
@sockets_to_pools.delete(socket)
# Refresh synchronously every @refresh_interval seconds
# if synchronous refresh mode is enabled.
if @refresh_mode == :sync &&
((Time.now - @last_refresh) > @refresh_interval)
refresh
@last_refresh = Time.now
end
end end
def close_socket(socket) def close_socket(socket)
@ -466,7 +367,6 @@ module Mongo
begin begin
if pool if pool
socket = pool.checkout socket = pool.checkout
@sockets_to_pools[socket] = pool
socket socket
end end
rescue ConnectionFailure => ex rescue ConnectionFailure => ex
@ -577,10 +477,8 @@ module Mongo
# Given a pool manager, update this connection's # Given a pool manager, update this connection's
# view of the replica set. # view of the replica set.
def update_config(new_manager) def update_config(new_manager)
old_manager = @manager
@manager = new_manager @manager = new_manager
@seeds = @manager.seeds.dup @seeds = @manager.seeds.dup
@sockets_to_pools.clear
@refresh_version += 1 @refresh_version += 1
end end
@ -591,11 +489,12 @@ module Mongo
if @refresh_mode == :async if @refresh_mode == :async
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 && @connected do while true do
sleep(@refresh_interval) sleep(@refresh_interval)
refresh refresh
end end
end end
@refresh_thread.priority = 1000
end end
@last_refresh = Time.now @last_refresh = Time.now
@ -611,7 +510,6 @@ module Mongo
pool = self.tag_map[{k.to_s => v}] pool = self.tag_map[{k.to_s => v}]
if pool if pool
socket = pool.checkout socket = pool.checkout
@sockets_to_pools[socket] = pool
return socket return socket
end end
end end
@ -620,18 +518,12 @@ module Mongo
"Could not find a connection tagged with #{tags}." "Could not find a connection tagged with #{tags}."
end end
# Checkin a socket used for reading. def sync_refresh
def checkin_reader(socket) if @refresh_mode == :sync &&
warn "ReplSetConnection#checkin_writer is deprecated and will be removed " + ((Time.now - @last_refresh) > @refresh_interval)
"in driver v2.0. Use ReplSetConnection#checkin instead." @last_refresh = Time.now
checkin(socket) refresh
end end
# Checkin a socket used for writing.
def checkin_writer(socket)
warn "ReplSetConnection#checkin_writer is deprecated and will be removed " +
"in driver v2.0. Use ReplSetConnection#checkin instead."
checkin(socket)
end end
end end
end end

1
lib/mongo/util/node.rb
View File

@ -36,7 +36,6 @@ module Mongo
def connect def connect
begin begin
socket = nil socket = nil
puts "Creating connection in node to #{@host}:#{@port}"
if @connection.connect_timeout if @connection.connect_timeout
Mongo::TimeoutHandler.timeout(@connection.connect_timeout, OperationTimeout) do Mongo::TimeoutHandler.timeout(@connection.connect_timeout, OperationTimeout) do
socket = @connection.socket_class.new(@host, @port) socket = @connection.socket_class.new(@host, @port)

29
lib/mongo/util/pool.rb
View File

@ -17,13 +17,11 @@
module Mongo module Mongo
class Pool class Pool
PRUNE_INTERVAL = 300
PING_ATTEMPTS = 6 PING_ATTEMPTS = 6
MAX_PING_TIME = 1_000_000 MAX_PING_TIME = 1_000_000
attr_accessor :host, :port, :address, attr_accessor :host, :port, :address,
:size, :timeout, :safe, :checked_out, :connection, :size, :timeout, :safe, :checked_out, :connection
:sockets_low
# Create a new pool of connections. # Create a new pool of connections.
def initialize(connection, host, port, opts={}) def initialize(connection, host, port, opts={})
@ -50,7 +48,6 @@ module Mongo
# Operations to perform on a socket # Operations to perform on a socket
@socket_ops = Hash.new { |h, k| h[k] = [] } @socket_ops = Hash.new { |h, k| h[k] = [] }
@sockets_low = true
@sockets = [] @sockets = []
@pids = {} @pids = {}
@checked_out = [] @checked_out = []
@ -74,7 +71,7 @@ module Mongo
end end
sockets_to_close.each do |sock| sockets_to_close.each do |sock|
begin begin
sock.close sock.close unless sock.closed?
rescue IOError => ex rescue IOError => ex
warn "IOError when attempting to close socket connected to #{@host}:#{@port}: #{ex.inspect}" warn "IOError when attempting to close socket connected to #{@host}:#{@port}: #{ex.inspect}"
end end
@ -90,10 +87,6 @@ module Mongo
@closed @closed
end end
def sockets_low?
@sockets_low
end
def inspect def inspect
"#<Mongo::Pool:0x#{self.object_id.to_s(16)} @host=#{@host} @port=#{port} " + "#<Mongo::Pool:0x#{self.object_id.to_s(16)} @host=#{@host} @port=#{port} " +
"@ping_time=#{@ping_time} #{@checked_out.size}/#{@size} sockets available.>" "@ping_time=#{@ping_time} #{@checked_out.size}/#{@size} sockets available.>"
@ -156,8 +149,11 @@ module Mongo
# Return a socket to the pool. # Return a socket to the pool.
def checkin(socket) def checkin(socket)
@connection_mutex.synchronize do @connection_mutex.synchronize do
@checked_out.delete(socket) if @checked_out.delete(socket)
@queue.signal @queue.signal
else
return false
end
end end
true true
end end
@ -263,18 +259,7 @@ module Mongo
end end
@connection_mutex.synchronize do @connection_mutex.synchronize do
if @size > 1000 #prune
if @sockets.size > 0.7 * @size
@sockets_low = true
else
@sockets_low = false
end
if (Time.now - @last_pruning) > PRUNE_INTERVAL
prune
@last_pruning = Time.now
end
end
socket = if @checked_out.size < @sockets.size socket = if @checked_out.size < @sockets.size
checkout_existing_socket checkout_existing_socket

64
test/replica_sets/refresh_test.rb
View File

@ -18,13 +18,17 @@ class ReplicaSetRefreshTest < Test::Unit::TestCase
Benchmark.bm do |x| Benchmark.bm do |x|
x.report("Connect") do x.report("Connect") do
10.times do 10.times do
ReplSetConnection.new([self.rs.host, self.rs.ports[0]], [self.rs.host, self.rs.ports[1]], ReplSetConnection.new([self.rs.host, self.rs.ports[0]],
[self.rs.host, self.rs.ports[2]], :refresh_mode => false) [self.rs.host, self.rs.ports[1]],
[self.rs.host, self.rs.ports[2]],
:refresh_mode => false)
end end
end end
@con = ReplSetConnection.new([self.rs.host, self.rs.ports[0]], [self.rs.host, self.rs.ports[1]], @con = ReplSetConnection.new([self.rs.host, self.rs.ports[0]],
[self.rs.host, self.rs.ports[2]], :refresh_mode => false) [self.rs.host, self.rs.ports[1]],
[self.rs.host, self.rs.ports[2]],
:refresh_mode => false)
x.report("manager") do x.report("manager") do
man = Mongo::PoolManager.new(@con, @con.seeds) man = Mongo::PoolManager.new(@con, @con.seeds)
@ -39,8 +43,10 @@ class ReplicaSetRefreshTest < Test::Unit::TestCase
self.rs.kill_all_secondaries self.rs.kill_all_secondaries
rescue_connection_failure do rescue_connection_failure do
@conn = ReplSetConnection.new([self.rs.host, self.rs.ports[0]], [self.rs.host, self.rs.ports[1]], @conn = ReplSetConnection.new([self.rs.host, self.rs.ports[0]],
[self.rs.host, self.rs.ports[2]], :refresh_mode => false) [self.rs.host, self.rs.ports[1]],
[self.rs.host, self.rs.ports[2]],
:refresh_mode => false)
end end
assert_equal [], @conn.secondaries assert_equal [], @conn.secondaries
@ -68,33 +74,51 @@ class ReplicaSetRefreshTest < Test::Unit::TestCase
self.rs.kill_all_secondaries self.rs.kill_all_secondaries
rescue_connection_failure do rescue_connection_failure do
@conn = ReplSetConnection.new([self.rs.host, self.rs.ports[0]], [self.rs.host, self.rs.ports[1]], @conn = ReplSetConnection.new([self.rs.host, self.rs.ports[0]],
[self.rs.host, self.rs.ports[2]], :refresh_interval => 2, :refresh_mode => :async) [self.rs.host, self.rs.ports[1]],
[self.rs.host, self.rs.ports[2]],
:refresh_interval => 2,
:refresh_mode => :sync)
end end
assert_equal [], @conn.secondaries assert_equal [], @conn.secondaries
assert @conn.connected? assert @conn.connected?
assert_equal @conn.read_pool, @conn.primary_pool assert_equal @conn.read_pool, @conn.primary_pool
old_refresh_version = @conn.refresh_version
self.rs.restart_killed_nodes self.rs.restart_killed_nodes
sleep(4) sleep(4)
@conn['foo']['bar'].find_one
@conn['foo']['bar'].insert({:a => 1})
puts "Old: #{old_refresh_version} New: #{@conn.refresh_version}"
assert @conn.read_pool != @conn.primary_pool, "Read pool and primary pool are identical." assert @conn.refresh_version > old_refresh_version,
assert @conn.secondaries.length > 0, "No secondaries have been added." "Refresh version hasn't changed."
assert @conn.secondaries.length > 0,
"No secondaries have been added."
assert @conn.read_pool != @conn.primary_pool,
"Read pool and primary pool are identical."
end end
def test_automated_refresh_with_removed_node def test_automated_refresh_with_removed_node
@conn = ReplSetConnection.new([self.rs.host, self.rs.ports[0]], [self.rs.host, self.rs.ports[1]], @conn = ReplSetConnection.new([self.rs.host, self.rs.ports[0]],
[self.rs.host, self.rs.ports[2]], :refresh_interval => 2, :refresh_mode => :async) [self.rs.host, self.rs.ports[1]],
[self.rs.host, self.rs.ports[2]],
:refresh_interval => 2,
:refresh_mode => :sync)
assert_equal 2, @conn.secondary_pools.length num_secondaries = @conn.secondary_pools.length
assert_equal 2, @conn.secondaries.length old_refresh_version = @conn.refresh_version
n = self.rs.remove_secondary_node n = self.rs.remove_secondary_node
sleep(4) sleep(4)
@conn['foo']['bar'].find_one
puts "Old: #{old_refresh_version} New: #{@conn.refresh_version}"
assert_equal 1, @conn.secondaries.length assert @conn.refresh_version > old_refresh_version,
assert_equal 1, @conn.secondary_pools.length "Refresh version hasn't changed."
assert_equal num_secondaries - 1, @conn.secondaries.length
assert_equal num_secondaries - 1, @conn.secondary_pools.length
self.rs.add_node(n) self.rs.add_node(n)
end end
@ -103,17 +127,19 @@ class ReplicaSetRefreshTest < Test::Unit::TestCase
@conn = ReplSetConnection.new([self.rs.host, self.rs.ports[0]], @conn = ReplSetConnection.new([self.rs.host, self.rs.ports[0]],
[self.rs.host, self.rs.ports[1]], [self.rs.host, self.rs.ports[1]],
[self.rs.host, self.rs.ports[2]], [self.rs.host, self.rs.ports[2]],
:refresh_interval => 2, :refresh_mode => :async) :refresh_interval => 2, :refresh_mode => :sync)
self.rs.add_node self.rs.add_node
sleep(4) sleep(4)
@conn['foo']['bar'].find_one
@conn2 = ReplSetConnection.new([self.rs.host, self.rs.ports[0]], @conn2 = ReplSetConnection.new([self.rs.host, self.rs.ports[0]],
[self.rs.host, self.rs.ports[1]], [self.rs.host, self.rs.ports[1]],
[self.rs.host, self.rs.ports[2]], [self.rs.host, self.rs.ports[2]],
:refresh_interval => 2, :refresh_mode => :async) :refresh_interval => 2, :refresh_mode => :sync)
assert @conn2.secondaries == @conn.secondaries assert @conn2.secondaries.sort == @conn.secondaries.sort,
"Second connection secondaries not equal to first."
assert_equal 3, @conn.secondary_pools.length assert_equal 3, @conn.secondary_pools.length
assert_equal 3, @conn.secondaries.length assert_equal 3, @conn.secondaries.length

13
test/replica_sets/refresh_with_threads_test.rb
View File

@ -18,7 +18,8 @@ class ReplicaSetRefreshWithThreadsTest < Test::Unit::TestCase
@conn = ReplSetConnection.new([self.rs.host, self.rs.ports[0]], @conn = ReplSetConnection.new([self.rs.host, self.rs.ports[0]],
[self.rs.host, self.rs.ports[1]], [self.rs.host, self.rs.ports[1]],
[self.rs.host, self.rs.ports[2]], [self.rs.host, self.rs.ports[2]],
:refresh_interval => 2, :refresh_mode => :async, :refresh_interval => 5,
:refresh_mode => :sync,
:read => :secondary) :read => :secondary)
@duplicate = @conn[MONGO_TEST_DB]['duplicate'] @duplicate = @conn[MONGO_TEST_DB]['duplicate']
@unique = @conn[MONGO_TEST_DB]['unique'] @unique = @conn[MONGO_TEST_DB]['unique']
@ -29,9 +30,9 @@ class ReplicaSetRefreshWithThreadsTest < Test::Unit::TestCase
@unique.create_index("test", :unique => true) @unique.create_index("test", :unique => true)
threads = [] threads = []
100.times do 10.times do
threads << Thread.new do threads << Thread.new do
100.times do |i| 1000.times do |i|
if i % 2 == 0 if i % 2 == 0
assert_raise Mongo::OperationFailure do assert_raise Mongo::OperationFailure do
@unique.insert({"test" => "insert"}, :safe => true) @unique.insert({"test" => "insert"}, :safe => true)
@ -40,17 +41,15 @@ class ReplicaSetRefreshWithThreadsTest < Test::Unit::TestCase
@duplicate.insert({"test" => "insert"}, :safe => true) @duplicate.insert({"test" => "insert"}, :safe => true)
end end
end end
@conn.end_request
end end
end end
self.rs.add_node self.rs.add_node
sleep(4) threads.each {|t| t.join }
config = @conn['admin'].command({:ismaster => 1}) config = @conn['admin'].command({:ismaster => 1})
assert_equal 3, @conn.secondary_pools.length assert_equal 3, @conn.secondary_pools.length
assert_equal 3, @conn.secondaries.length assert_equal 3, @conn.secondaries.length
threads.each {|t| t.join }
end end
end end

3
test/threading_test.rb
View File

@ -39,7 +39,6 @@ class TestThreading < Test::Unit::TestCase
@duplicate.update({"test" => "insert"}, {"$set" => {"test" => "update"}}, :safe => true) @duplicate.update({"test" => "insert"}, {"$set" => {"test" => "update"}}, :safe => true)
times << Time.now - t1 times << Time.now - t1
end end
@@con.end_request
end end
end end
end end
@ -61,7 +60,6 @@ class TestThreading < Test::Unit::TestCase
else else
@duplicate.insert({"test" => "insert"}, :safe => true) @duplicate.insert({"test" => "insert"}, :safe => true)
end end
@@con.end_request
end end
end end
@ -87,7 +85,6 @@ class TestThreading < Test::Unit::TestCase
sum += document["x"] sum += document["x"]
end end
assert_equal 499500, sum assert_equal 499500, sum
@@con.end_request
end end
end end

12
test/unit/read_test.rb
View File

@ -73,12 +73,11 @@ class ReadTest < Test::Unit::TestCase
should "use default value on query" do should "use default value on query" do
@cursor = @col.find({:a => 1}) @cursor = @col.find({:a => 1})
sock = mock() sock = mock()
sock.expects(:close).twice read_pool = stub(:checkin => true)
read_pool = stub(:sockets_low? => false)
@con.stubs(:read_pool).returns(read_pool) @con.stubs(:read_pool).returns(read_pool)
primary_pool = stub(:sockets_low? => false) primary_pool = stub(:checkin => true)
@con.stubs(:primary_pool).returns(primary_pool) @con.stubs(:primary_pool).returns(primary_pool)
@con.expects(:checkout_reader).twice.returns(sock) @con.expects(:checkout_reader).returns(sock)
@con.expects(:receive_message).with do |o, m, l, s, c, r| @con.expects(:receive_message).with do |o, m, l, s, c, r|
r == nil r == nil
end.returns([[], 0, 0]) end.returns([[], 0, 0])
@ -89,10 +88,9 @@ class ReadTest < Test::Unit::TestCase
should "allow override default value on query" do should "allow override default value on query" do
@cursor = @col.find({:a => 1}, :read => :primary) @cursor = @col.find({:a => 1}, :read => :primary)
sock = mock() sock = mock()
sock.expects(:close).twice primary_pool = stub(:checkin => true)
primary_pool = stub(:sockets_low? => false)
@con.stubs(:primary_pool).returns(primary_pool) @con.stubs(:primary_pool).returns(primary_pool)
@con.expects(:checkout_writer).twice.returns(sock) @con.expects(:checkout_writer).returns(sock)
@con.expects(:receive_message).with do |o, m, l, s, c, r| @con.expects(:receive_message).with do |o, m, l, s, c, r|
r == nil r == nil
end.returns([[], 0, 0]) end.returns([[], 0, 0])