class Mongo::Server::ConnectionPool::GenerationManager
@api private
Attributes
server[R]
Public Class Methods
new(server:)
click to toggle source
# File lib/mongo/server/connection_pool/generation_manager.rb, line 24 def initialize(server:) @map = Hash.new { |hash, key| hash[key] = 1 } @pipe_fds = Hash.new { |hash, key| hash[key] = { 1 => IO.pipe } } @server = server @lock = Mutex.new @scheduled_for_close = [] end
Public Instance Methods
bump(service_id: nil)
click to toggle source
# File lib/mongo/server/connection_pool/generation_manager.rb, line 65 def bump(service_id: nil) @lock.synchronize do close_all_scheduled if service_id gen = @map[service_id] += 1 @pipe_fds[service_id] ||= {} @pipe_fds[service_id][gen] = IO.pipe else # When service id is not supplied, one of two things may be # happening; # # 1. The pool is not to a load balancer, in which case we only # need to increment the generation for the nil service_id. # 2. The pool is to a load balancer, in which case we need to # increment the generation for each service. # # Incrementing everything in the map accomplishes both tasks. @map.each do |k, v| gen = @map[k] += 1 @pipe_fds[service_id] ||= {} @pipe_fds[service_id][gen] = IO.pipe end end end end
generation(service_id: nil)
click to toggle source
# File lib/mongo/server/connection_pool/generation_manager.rb, line 34 def generation(service_id: nil) validate_service_id!(service_id) @lock.synchronize do @map[service_id] end end
generation_unlocked(service_id: nil)
click to toggle source
# File lib/mongo/server/connection_pool/generation_manager.rb, line 42 def generation_unlocked(service_id: nil) validate_service_id!(service_id) @map[service_id] end
pipe_fds(service_id: nil)
click to toggle source
# File lib/mongo/server/connection_pool/generation_manager.rb, line 48 def pipe_fds(service_id: nil) @pipe_fds[service_id][@map[service_id]] end
remove_pipe_fds(generation, service_id: nil)
click to toggle source
# File lib/mongo/server/connection_pool/generation_manager.rb, line 52 def remove_pipe_fds(generation, service_id: nil) validate_service_id!(service_id) r, w = @pipe_fds[service_id].delete(generation) w.close # Schedule the read end of the pipe to be closed. We cannot close it # immediately since we need to wait for any Kernel#select calls to # notice that part of the pipe is closed, and check the socket. This # all happens when attempting to read from the socket and waiting for # it to become ready again. @scheduled_for_close << r end
Private Instance Methods
close_all_scheduled()
click to toggle source
Close all fds scheduled for closing.
# File lib/mongo/server/connection_pool/generation_manager.rb, line 106 def close_all_scheduled while pipe = @scheduled_for_close.pop pipe.close end end
validate_service_id!(service_id)
click to toggle source
# File lib/mongo/server/connection_pool/generation_manager.rb, line 93 def validate_service_id!(service_id) if service_id unless server.load_balancer? raise ArgumentError, "Generation scoping to services is only available in load-balanced mode, but the server at #{server.address} is not a load balancer" end else if server.load_balancer? raise ArgumentError, "The server at #{server.address} is a load balancer and therefore does not have a single global generation" end end end