Object
Generic connection pool class for e.g. sharing a limited number of network connections among many threads. Note: Connections are eager created.
Example usage with block (faster):
@pool = 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 = 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
# File lib/connection_pool.rb, line 43 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
# File lib/connection_pool.rb, line 78 def checkin stack = ::Thread.current[@key] raise ConnectionPool::Error, 'no connections are checked out' if !stack || stack.empty? conn = stack.pop if stack.empty? @available << conn end nil end
# File lib/connection_pool.rb, line 64 def checkout(options = {}) stack = ::Thread.current[@key] ||= [] if stack.empty? timeout = options[:timeout] || @timeout conn = @available.pop(timeout) else conn = stack.last end stack.push conn conn end
Generated with the Darkfish Rdoc Generator 2.