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
132 def with_server(default_server, read_only_server=default_server)
133   set_default_server(default_server, read_only_server)
134   yield
135 ensure
136   clear_default_server
137 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
148 def clear_default_server
149   t = Sequel.current
150   a = sync{@default_servers[t]}
151   a.pop
152   sync{@default_servers.delete(t)} if a.empty?
153 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
156 def pick_server(server)
157   a = sync{@default_servers[Sequel.current]}
158   if !a || a.empty?
159     super
160   else
161     # Hash handling required to work when loaded after arbitrary servers plugin.
162     case server
163     when :default, nil
164       v = a[-1][0]
165       v = @servers[v] unless v.is_a?(Hash)
166       v
167     when :read_only
168       v = a[-1][1]
169       v = @servers[v] unless v.is_a?(Hash)
170       v
171     else
172       super
173     end
174   end
175 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
142 def set_default_server(default_server, read_only_server=default_server)
143   sync{(@default_servers[Sequel.current] ||= [])} << [default_server, read_only_server]
144 end