module Sequel::ThreadedServerBlock
Adds with_server
support for the sharded threaded connection pool.
Public Instance Methods
with_server(default_server, read_only_server=default_server) { || ... }
click to toggle source
Set a default server/shard to use inside the block for the current thread.
# File lib/sequel/extensions/server_block.rb 133 def with_server(default_server, read_only_server=default_server) 134 begin 135 set_default_server(default_server, read_only_server) 136 yield 137 ensure 138 clear_default_server 139 end 140 end
Private Instance Methods
clear_default_server()
click to toggle source
Remove the current default server for the current thread, restoring the previous default server.
# File lib/sequel/extensions/server_block.rb 151 def clear_default_server 152 t = Thread.current 153 a = sync{@default_servers[t]} 154 a.pop 155 sync{@default_servers.delete(t)} if a.empty? 156 end
pick_server(server)
click to toggle source
Use the server given to with_server
for the given thread, if appropriate.
Calls superclass method
# File lib/sequel/extensions/server_block.rb 159 def pick_server(server) 160 a = sync{@default_servers[Thread.current]} 161 if !a || a.empty? 162 super 163 else 164 # Hash handling required to work when loaded after arbitrary servers plugin. 165 case server 166 when :default, nil 167 v = a[-1][0] 168 v = @servers[v] unless v.is_a?(Hash) 169 v 170 when :read_only 171 v = a[-1][1] 172 v = @servers[v] unless v.is_a?(Hash) 173 v 174 else 175 super 176 end 177 end 178 end
set_default_server(default_server, read_only_server=default_server)
click to toggle source
Make the given server the new default server for the current thread.
# File lib/sequel/extensions/server_block.rb 145 def set_default_server(default_server, read_only_server=default_server) 146 sync{(@default_servers[Thread.current] ||= [])} << [default_server, read_only_server] 147 end