/usr/lib/ruby/vendor_ruby/sequel/connection_pool/sharded_single.rb is in ruby-sequel 3.33.0-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 | # A ShardedSingleConnectionPool is a single threaded connection pool that
# works with multiple shards/servers.
class Sequel::ShardedSingleConnectionPool < Sequel::ConnectionPool
# Initializes the instance with the supplied block as the connection_proc.
#
# The single threaded pool takes the following options:
#
# * :servers - A hash of servers to use. Keys should be symbols. If not
# present, will use a single :default server. The server name symbol will
# be passed to the connection_proc.
# * :servers_hash - The base hash to use for the servers. By default,
# Sequel uses Hash.new(:default). You can use a hash with a default proc
# that raises an error if you want to catch all cases where a nonexistent
# server is used.
def initialize(opts={}, &block)
super
@conns = {}
@servers = opts.fetch(:servers_hash, Hash.new(:default))
add_servers([:default])
add_servers(opts[:servers].keys) if opts[:servers]
end
# Adds new servers to the connection pool. Primarily used in conjunction with master/slave
# or shard configurations. Allows for dynamic expansion of the potential slaves/shards
# at runtime. servers argument should be an array of symbols.
def add_servers(servers)
servers.each{|s| @servers[s] = s}
end
# The connection for the given server.
def conn(server=:default)
@conns[@servers[server]]
end
# Disconnects from the database. Once a connection is requested using
# #hold, the connection is reestablished. Options:
# * :server - Should be a symbol specifing the server to disconnect from,
# or an array of symbols to specify multiple servers.
def disconnect(opts={}, &block)
block ||= @disconnection_proc
(opts[:server] ? Array(opts[:server]) : servers).each{|s| disconnect_server(s, &block)}
end
# Yields the connection to the supplied block for the given server.
# This method simulates the ConnectionPool#hold API.
def hold(server=:default)
begin
server = pick_server(server)
yield(@conns[server] ||= make_new(server))
rescue Sequel::DatabaseDisconnectError
disconnect_server(server, &@disconnection_proc)
raise
end
end
# Remove servers from the connection pool. Primarily used in conjunction with master/slave
# or shard configurations. Similar to disconnecting from all given servers,
# except that after it is used, future requests for the server will use the
# :default server instead.
def remove_servers(servers)
raise(Sequel::Error, "cannot remove default server") if servers.include?(:default)
servers.each do |server|
disconnect_server(server, &@disconnection_proc)
@servers.delete(server)
end
end
# Return an array of symbols for servers in the connection pool.
def servers
@servers.keys
end
# The number of different shards/servers this pool is connected to.
def size
@conns.length
end
private
# Disconnect from the given server, if connected.
def disconnect_server(server, &block)
if conn = @conns.delete(server)
block.call(conn) if block
end
end
# If the server given is in the hash, return it, otherwise, return the default server.
def pick_server(server)
@servers[server]
end
CONNECTION_POOL_MAP[[true, true]] = self
end
|