class Vault::ConnectionPool

Constants

DEFAULTS
VERSION

Public Class Methods

new(options = {}, &block) click to toggle source
# File lib/vault/vendor/connection_pool.rb, line 45
def initialize(options = {}, &block)
  raise ArgumentError, 'Connection pool requires a block' unless block

  options = DEFAULTS.merge(options)

  @size = options.fetch(:size)
  @timeout = options.fetch(:timeout)

  @available = TimedStack.new(@size, &block)
  @key = :"current-#{@available.object_id}"
end
wrap(options, &block) click to toggle source
# File lib/vault/vendor/connection_pool.rb, line 41
def self.wrap(options, &block)
  Wrapper.new(options, &block)
end

Public Instance Methods

checkin() click to toggle source
# File lib/vault/vendor/connection_pool.rb, line 99
def checkin
  conn = pop_connection # mutates stack, must be on its own line
  @available.push(conn) if stack.empty?

  nil
end
checkout(options = {}) click to toggle source
# File lib/vault/vendor/connection_pool.rb, line 87
def checkout(options = {})
  conn = if stack.empty?
    timeout = options[:timeout] || @timeout
    @available.pop(timeout: timeout)
  else
    stack.last
  end

  stack.push conn
  conn
end
shutdown(&block) click to toggle source
# File lib/vault/vendor/connection_pool.rb, line 106
def shutdown(&block)
  @available.shutdown(&block)
end
with(options = {}) { |conn| ... } click to toggle source

MRI

# File lib/vault/vendor/connection_pool.rb, line 60
def with(options = {})
  Thread.handle_interrupt(Exception => :never) do
    conn = checkout(options)
    begin
      Thread.handle_interrupt(Exception => :immediate) do
        yield conn
      end
    ensure
      checkin
    end
  end
end

Private Instance Methods

pop_connection() click to toggle source
# File lib/vault/vendor/connection_pool.rb, line 112
def pop_connection
  if stack.empty?
    raise ConnectionPool::Error, 'no connections are checked out'
  else
    stack.pop
  end
end
stack() click to toggle source
# File lib/vault/vendor/connection_pool.rb, line 120
def stack
  ::Thread.current[@key] ||= []
end