Sha256: f2b480ae1981d5731a0425aa84edde1accfda8798865119a2f8dd652c9353019
Contents?: true
Size: 1.56 KB
Versions: 8
Compression:
Stored size: 1.56 KB
Contents
require 'eventmachine'
module Swift
class Pool
module Handler
def initialize request, pool
@request, @pool = request, pool
end
def socket
@request.socket
end
def notify_readable
if @request.process
detach
@pool.detach self
end
end
def notify_writable
notify_readable
end
end # Handler
def initialize size, options
@pool = Swift::DB::Pool.new size, options
# used to be used for db2
@writable = false
@pending = {}
@queue = []
end
def attach c
@pending[c] = true
end
def detach c
@pending.delete(c)
unless @queue.empty?
sql, bind, callback = @queue.shift
execute(sql, *bind, &callback)
end
end
def attached? fd
@pending.keys.select{|c| c.socket == fd}.length > 0
end
def execute sql, *bind, &callback
request = @pool.execute sql, *bind, &callback
# NOTE EM segfaults if we try to attach same fd twice.
if request && !attached?(request.socket)
EM.watch(request.socket, Handler, request, self) do |c|
attach c
c.notify_writable = @writable
c.notify_readable = true
end
else
@queue << [ sql, bind, callback ]
end
end
def run &block
EM.run{ yield self }
end
end # Pool
def self.pool size, name = :default, &block
pool = Pool.new(size, Swift.db(name).options)
pool.run(&block) if block_given?
pool
end
end # Swift
Version data entries
8 entries across 8 versions & 1 rubygems