/usr/lib/ruby/vendor_ruby/sequel/extensions/arbitrary_servers.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 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 | # The arbitrary_servers extension allows you to connect to arbitrary
# servers/shards that were not defined when you created the database.
# To use it, you first extend the Database's connection pool with the
# Sequel::ArbitraryServers module:
#
# Sequel.extension :arbitrary_servers
# DB.pool.extend Sequel::ArbitraryServers
#
# Then you can pass arbitrary connection options for the server/shard
# to use as a hash:
#
# DB[:table].server(:host=>'...', :database=>'...').all
#
# Because Sequel can never be sure that the connection will be reused,
# arbitrary connections are disconnected as soon as the outermost block
# that uses them exits. So this example uses the same connection:
#
# DB.transaction(:server=>{:host=>'...', :database=>'...'}) do |c|
# DB.transaction(:server=>{:host=>'...', :database=>'...'}) do |c2|
# # c == c2
# end
# end
#
# But this example does not:
#
# DB.transaction(:server=>{:host=>'...', :database=>'...'}) do |c|
# end
# DB.transaction(:server=>{:host=>'...', :database=>'...'}) do |c2|
# # c != c2
# end
#
# You can use this extension in conjunction with the server_block
# extension:
#
# DB.with_server(:host=>'...', :database=>'...') do
# DB.synchronize do
# # All of these use the host/database given to with_server
# DB[:table].insert(...)
# DB[:table].update(...)
# DB.tables
# DB[:table].all
# end
# end
#
# Anyone using this extension in conjunction with the server_block
# extension may want to do the following to so that you don't need
# to call synchronize separately:
#
# def DB.with_server(*)
# super{synchronize{yield}}
# end
#
# Note that this extension only works with the sharded threaded connection
# pool. If you are using the sharded single connection pool, you need
# to switch to the sharded threaded connection pool before using this
# extension.
module Sequel
module ArbitraryServers
private
# If server is a hash, create a new connection for
# it, and cache it first by thread and then server.
def acquire(thread, server)
if server.is_a?(Hash)
sync{@allocated[thread] ||= {}}[server] = make_new(server)
else
super
end
end
# If server is a hash, the entry for it probably doesn't
# exist in the @allocated hash, so check for existence to
# avoid calling nil.[]
def owned_connection(thread, server)
if server.is_a?(Hash)
if a = sync{@allocated[thread]}
a[server]
end
else
super
end
end
# If server is a hash, return it directly.
def pick_server(server)
if server.is_a?(Hash)
server
else
super
end
end
# If server is a hash, delete the thread from the allocated
# connections for that server. Additionally, if this was the last thread
# using that server, delete the server from the @allocated hash.
def release(thread, conn, server)
if server.is_a?(Hash)
a = @allocated[thread]
a.delete(server)
@allocated.delete(thread) if a.empty?
@disconnection_proc.call(conn) if @disconnection_proc
else
super
end
end
end
end
|