Sha256: 15775c620090eeb8ed0599ed41b614a984aff95e5e2a4a5ab1c94066ed35bb47
Contents?: true
Size: 1.86 KB
Versions: 30
Compression:
Stored size: 1.86 KB
Contents
require "ftw/namespace"
require "thread"
# A simple thread-safe resource pool.
#
# Resources in this pool must respond to 'available?'.
# For best results, your resources should just 'include FTW::Poolable'
#
# The primary use case was as a way to pool FTW::Connection instances.
class FTW::Pool
def initialize
# Pool is a hash of arrays.
@pool = Hash.new { |h,k| h[k] = Array.new }
@lock = Mutex.new
end # def initialize
# Add an object to the pool with a given identifier. For example:
#
# pool.add("www.google.com:80", connection1)
# pool.add("www.google.com:80", connection2)
# pool.add("github.com:443", connection3)
def add(identifier, object)
@lock.synchronize do
@pool[identifier] << object
end
return object
end # def add
# Fetch a resource from this pool. If no available resources
# are found, the 'default_block' is invoked and expected to
# return a new resource to add to the pool that satisfies
# the fetch..
#
# Example:
#
# pool.fetch("github.com:443") do
# conn = FTW::Connection.new("github.com:443")
# conn.secure
# conn
# end
def fetch(identifier, &default_block)
@lock.synchronize do
object = @pool[identifier].find { |o| o.available? }
return object if !object.nil?
end
# Otherwise put the return value of default_block in the
# pool and return it, but don't put nil values in the pool.
obj = default_block.call
if obj.nil?
return nil
else
return add(identifier, obj)
end
end # def fetch
# Iterate over all pool members.
#
# This holds the pool lock during this method, so you should not call 'fetch'
# or 'add'.
def each(&block)
@lock.synchronize do
@pool.each do |identifier, object|
block.call(identifier, object)
end
end
end # def each
end # class FTW::Pool
Version data entries
30 entries across 30 versions & 1 rubygems