1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
class CouchRest::Session
end
require 'couchrest/session/utility'
require 'couchrest/session/document'
module CouchRest
class StorageMissing < Exception
attr_reader :db
def initialize(request, db)
super(request)
@db = db.name
@message = "The database '#{db}' does not exist."
end
end
class Session
include CouchRest::Session::Utility
def self.fetch(sid)
self.allocate.tap do |session_doc|
session_doc.fetch(sid) || raise(CouchRest::NotFound)
end
end
def self.build(sid, session, options = {})
session_doc = CouchRest::Session::Document.new "_id" => sid
self.new(session_doc).
update session, options
end
def self.build_or_update(sid, session, options = {})
options[:marshal_data] = true if options[:marshal_data].nil?
self.fetch(sid).
update session, options
rescue CouchRest::NotFound
self.build sid, session, options
end
def initialize(doc)
@doc = doc
end
def fetch(sid = nil)
@doc = CouchRest::Session::Document.fetch(sid || doc['_id'])
end
def to_session
if doc["marshalled"]
session = unmarshal(doc["data"])
else
session = doc["data"]
end
return session
end
def delete
doc.destroy
end
def update(session, options)
# clean up old data but leave id and revision intact
doc.reject! { |k,_v| k[0] != '_' }
doc.merge! data_for_doc(session, options)
self
end
def save
doc.save
rescue CouchRest::Conflict
fetch
retry
rescue CouchRest::NotFound => exc
if exc.http_body =~ /no_db_file/
exc = CouchRest::StorageMissing.new(exc.response, doc.database)
end
raise exc
end
def expired?
expires && expires < Time.now
end
protected
attr_reader :doc
def data_for_doc(session, options)
{ "data" => options[:marshal_data] ? marshal(session) : session,
"marshalled" => options[:marshal_data],
"expires" => expiry_from_options(options) }
end
def expiry_from_options(options)
expire_after = options[:expire_after]
expire_after && (Time.now + expire_after).utc
end
def expires
doc["expires"] && Time.iso8601(doc["expires"])
end
end
end
|