/
persistence.rb
80 lines (68 loc) · 1.92 KB
/
persistence.rb
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
# encoding: UTF-8
module MongoMapper
module Plugins
module Persistence
extend ActiveSupport::Concern
module ClassMethods
def inherited(subclass)
unless subclass.embeddable?
subclass.connection(connection)
subclass.set_database_name(database_name)
subclass.set_collection_name(collection_name) unless subclass.explicit_collection_defined?
end
super
end
def connection(mongo_connection=nil)
assert_supported
if mongo_connection.nil? && MongoMapper.connection?
@connection ||= MongoMapper.connection
else
@connection = mongo_connection
end
@connection
end
def set_database_name(name)
assert_supported
@database_name = name
end
def database_name
assert_supported
@database_name ||= nil # ||= idiom prevents uninitialized ivar warnings.
end
def database
assert_supported
if database_name.nil?
MongoMapper.database
else
connection.db(database_name)
end
end
def set_collection_name(name)
assert_supported
@collection_name = name
end
def collection_name
assert_supported
@collection_name ||= self.to_s.tableize.gsub(/\//, '.')
end
def collection
assert_supported
database.collection(collection_name)
end
private
def assert_supported
@embeddable ||= embeddable?
if @embeddable
raise MongoMapper::NotSupported.new('This is not supported for embeddable documents at this time.')
end
end
end
def collection
_root_document.class.collection
end
def database
_root_document.class.database
end
end
end
end