Adds #with_server support for the sharded threaded connection pool.
Set a default server/shard to use inside the block for the current thread.
# File lib/sequel/extensions/server_block.rb, line 127 def with_server(default_server, read_only_server=default_server) begin set_default_server(default_server, read_only_server) yield ensure clear_default_server end end
Remove the current default server for the current thread, restoring the previous default server.
# File lib/sequel/extensions/server_block.rb, line 145 def clear_default_server t = Thread.current a = sync{@default_servers[t]} a.pop sync{@default_servers.delete(t)} if a.empty? end
Use the server given to #with_server for the given thread, if appropriate.
# File lib/sequel/extensions/server_block.rb, line 153 def pick_server(server) a = sync{@default_servers[Thread.current]} if !a || a.empty? super else case server when :default, nil a[-1][0] when :read_only a[-1][1] else super end end end
Make the given server the new default server for the current thread.
# File lib/sequel/extensions/server_block.rb, line 139 def set_default_server(default_server, read_only_server=default_server) sync{(@default_servers[Thread.current] ||= [])} << [default_server, read_only_server] end