Class: Bundler::ConnectionPool

Inherits:
Object
  • Object
show all
Defined in:
lib/bundler/vendor/connection_pool/lib/connection_pool.rb,
lib/bundler/vendor/connection_pool/lib/connection_pool.rb,
lib/bundler/vendor/connection_pool/lib/connection_pool/version.rb,
lib/bundler/vendor/connection_pool/lib/connection_pool/wrapper.rb

Overview

Generic connection pool class for sharing a limited number of objects or network connections among many threads. Note: pool elements are lazily created.

Example usage with block (faster):

@pool = Bundler::ConnectionPool.new { Redis.new }
@pool.with do |redis|
  redis.lpop('my-list') if redis.llen('my-list') > 0
end

Using optional timeout override (for that single invocation)

@pool.with(timeout: 2.0) do |redis|
  redis.lpop('my-list') if redis.llen('my-list') > 0
end

Example usage replacing an existing connection (slower):

$redis = Bundler::ConnectionPool.wrap { Redis.new }

def do_work
  $redis.lpop('my-list') if $redis.llen('my-list') > 0
end

Accepts the following options:

  • :size - number of connections to pool, defaults to 5

  • :timeout - amount of time to wait for a connection if none currently available, defaults to 5 seconds

Direct Known Subclasses

Persistent::Net::HTTP::Persistent::Pool

Defined Under Namespace

Classes: Error, PoolShuttingDownError, TimedStack, TimeoutError, Wrapper

Constant Summary collapse

DEFAULTS =
{size: 5, timeout: 5}
VERSION =
"2.3.0"

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(options = {}, &block) ⇒ ConnectionPool

Returns a new instance of ConnectionPool.

Raises:

  • (ArgumentError)


45
46
47
48
49
50
51
52
53
54
55
56
# File 'lib/bundler/vendor/connection_pool/lib/connection_pool.rb', line 45

def initialize(options = {}, &block)
  raise ArgumentError, "Connection pool requires a block" unless block

  options = DEFAULTS.merge(options)

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

  @available = TimedStack.new(@size, &block)
  @key = :"pool-#{@available.object_id}"
  @key_count = :"pool-#{@available.object_id}-count"
end

Instance Attribute Details

#sizeObject (readonly)

Size of this connection pool



117
118
119
# File 'lib/bundler/vendor/connection_pool/lib/connection_pool.rb', line 117

def size
  @size
end

Class Method Details

.wrap(options, &block) ⇒ Object



41
42
43
# File 'lib/bundler/vendor/connection_pool/lib/connection_pool.rb', line 41

def self.wrap(options, &block)
  Wrapper.new(options, &block)
end

Instance Method Details

#availableObject

Number of pool entries available for checkout at this instant.



120
121
122
# File 'lib/bundler/vendor/connection_pool/lib/connection_pool.rb', line 120

def available
  @available.length
end

#checkinObject



82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
# File 'lib/bundler/vendor/connection_pool/lib/connection_pool.rb', line 82

def checkin
  if ::Thread.current[@key]
    if ::Thread.current[@key_count] == 1
      @available.push(::Thread.current[@key])
      ::Thread.current[@key] = nil
      ::Thread.current[@key_count] = nil
    else
      ::Thread.current[@key_count] -= 1
    end
  else
    raise Bundler::ConnectionPool::Error, "no connections are checked out"
  end

  nil
end

#checkout(options = {}) ⇒ Object



72
73
74
75
76
77
78
79
80
# File 'lib/bundler/vendor/connection_pool/lib/connection_pool.rb', line 72

def checkout(options = {})
  if ::Thread.current[@key]
    ::Thread.current[@key_count] += 1
    ::Thread.current[@key]
  else
    ::Thread.current[@key_count] = 1
    ::Thread.current[@key] = @available.pop(options[:timeout] || @timeout)
  end
end

#reload(&block) ⇒ Object

Reloads the Bundler::ConnectionPool by passing each connection to block and then removing it the pool. Subsequent checkouts will create new connections as needed.



112
113
114
# File 'lib/bundler/vendor/connection_pool/lib/connection_pool.rb', line 112

def reload(&block)
  @available.shutdown(reload: true, &block)
end

#shutdown(&block) ⇒ Object

Shuts down the Bundler::ConnectionPool by passing each connection to block and then removing it from the pool. Attempting to checkout a connection after shutdown will raise Bundler::ConnectionPool::PoolShuttingDownError.



103
104
105
# File 'lib/bundler/vendor/connection_pool/lib/connection_pool.rb', line 103

def shutdown(&block)
  @available.shutdown(&block)
end

#with(options = {}) ⇒ Object Also known as: then



58
59
60
61
62
63
64
65
66
67
68
69
# File 'lib/bundler/vendor/connection_pool/lib/connection_pool.rb', line 58

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