Sha256: 5a6e56289de28d8431c5516b56b6c370e14552e0e14eb681d6c311a30a99f8af
Contents?: true
Size: 1.95 KB
Versions: 7
Compression:
Stored size: 1.95 KB
Contents
require 'rack/session/abstract/id'
require 'dalli'
module Rack
module Session
class Dalli < Abstract::ID
attr_reader :pool, :mutex
DEFAULT_OPTIONS = Abstract::ID::DEFAULT_OPTIONS.merge \
:namespace => 'rack:session',
:memcache_server => 'localhost:11211'
def initialize(app, options={})
super
@mutex = Mutex.new
mserv = @default_options[:memcache_server]
mopts = @default_options.reject{|k,v| !DEFAULT_OPTIONS.include? k }
@pool = options[:cache] || ::Dalli::Client.new(mserv, mopts)
@pool.alive!
end
def generate_sid
loop do
sid = super
break sid unless @pool.get(sid)
end
end
def get_session(env, sid)
with_lock(env, [nil, {}]) do
unless sid and !sid.empty? and session = @pool.get(sid)
sid, session = generate_sid, {}
unless @pool.add(sid, session)
raise "Session collision on '#{sid.inspect}'"
end
end
[sid, session]
end
end
def set_session(env, session_id, new_session, options)
return false unless session_id
expiry = options[:expire_after]
expiry = expiry.nil? ? 0 : expiry + 1
with_lock(env, false) do
@pool.set session_id, new_session, expiry
session_id
end
end
def destroy_session(env, session_id, options)
with_lock(env) do
@pool.delete(session_id)
generate_sid unless options[:drop]
end
end
def with_lock(env, default=nil)
@mutex.lock if env['rack.multithread']
yield
rescue ::Dalli::DalliError, Errno::ECONNREFUSED
raise if $!.message =~ /undefined class/
if $VERBOSE
warn "#{self} is unable to find memcached server."
warn $!.inspect
end
default
ensure
@mutex.unlock if @mutex.locked?
end
end
end
end
Version data entries
7 entries across 6 versions & 2 rubygems