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