module Sequel::ThreadedServerBlock

Adds #with_server support for the sharded threaded connection pool.

Public Instance Methods

with_server(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, line 107
def with_server(server)
  begin
    set_default_server(server)
    yield
  ensure
    clear_default_server
  end
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, 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
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, 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
set_default_server(server) click to toggle source

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