Class: Rack::Session::Dalli

Inherits:
Object
  • Object
show all
Defined in:
lib/rack/session/dalli.rb

Constant Summary collapse

DEFAULT_OPTIONS = 
Abstract::ID::DEFAULT_OPTIONS.merge \
:namespace => 'rack:session',
:memcache_server => 'localhost:11211'

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(app, options = {}) ⇒ Dalli

Returns a new instance of Dalli.



13
14
15
16
17
18
19
20
 
# File 'lib/rack/session/dalli.rb', line 13

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

Instance Attribute Details

#mutex ⇒ Object (readonly)

Returns the value of attribute mutex.



7
8
9
 
# File 'lib/rack/session/dalli.rb', line 7

def mutex
  @mutex
end

#pool ⇒ Object (readonly)

Returns the value of attribute pool.



7
8
9
 
# File 'lib/rack/session/dalli.rb', line 7

def pool
  @pool
end

Instance Method Details

#delete_session(req, sid, options) ⇒ Object 



31
32
33
 
# File 'lib/rack/session/dalli.rb', line 31

def delete_session(req, sid, options)
  destroy_session req.env, sid, options
end

#destroy_session(env, session_id, options) ⇒ Object 



66
67
68
69
70
71
 
# File 'lib/rack/session/dalli.rb', line 66

def destroy_session(env, session_id, options)
  with_lock(env) do
    @pool.delete(session_id)
    generate_sid unless options[:drop]
  end
end

#find_session(req, sid) ⇒ Object 



23
24
25
 
# File 'lib/rack/session/dalli.rb', line 23

def find_session(req, sid)
  get_session req.env, sid
end

#generate_sid ⇒ Object 



36
37
38
39
40
41
 
# File 'lib/rack/session/dalli.rb', line 36

def generate_sid
  while true
    sid = super
    break sid unless @pool.get(sid)
  end
end

#get_session(env, sid) ⇒ Object 



43
44
45
46
47
48
49
50
51
52
53
 
# File 'lib/rack/session/dalli.rb', line 43

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

#set_session(env, session_id, new_session, options) ⇒ Object 



55
56
57
58
59
60
61
62
63
64
 
# File 'lib/rack/session/dalli.rb', line 55

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

#with_lock(env, default = nil) ⇒ Object 



73
74
75
76
77
78
79
80
81
82
83
84
85
 
# File 'lib/rack/session/dalli.rb', line 73

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

#write_session(req, sid, session, options) ⇒ Object 



27
28
29
 
# File 'lib/rack/session/dalli.rb', line 27

def write_session(req, sid, session, options)
  set_session req.env, sid, session, options
end