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 107 def with_server(server) begin set_default_server(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 125 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 133 def pick_server(server) a = sync{@default_servers[Thread.current]} if !a || a.empty? super else case server when :default, nil, :read_only a.last 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 119 def set_default_server(server) sync{(@default_servers[Thread.current] ||= [])} << server end