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
|
require 'mongo_mapper'
module MongoMapper
module Document
module ClassMethods
include OrmAdapter::ToAdapter
end
class OrmAdapter < ::OrmAdapter::Base
# get a list of column names for a given class
def column_names
klass.column_names
end
# @see OrmAdapter::Base#get!
def get!(id)
klass.find!(wrap_key(id))
end
# @see OrmAdapter::Base#get
def get(id)
klass.first({ :id => wrap_key(id) })
end
# @see OrmAdapter::Base#find_first
def find_first(conditions = {})
conditions, order = extract_conditions!(conditions)
conditions = conditions.merge(:sort => order) unless order.nil?
klass.first(conditions_to_fields(conditions))
end
# @see OrmAdapter::Base#find_all
def find_all(conditions = {})
conditions, order, limit, offset = extract_conditions!(conditions)
conditions = conditions.merge(:sort => order) unless order.nil?
conditions = conditions.merge(:limit => limit) unless limit.nil?
conditions = conditions.merge(:offset => offset) unless limit.nil? || offset.nil?
klass.all(conditions_to_fields(conditions))
end
# @see OrmAdapter::Base#create!
def create!(attributes = {})
klass.create!(attributes)
end
# @see OrmAdapter::Base#destroy
def destroy(object)
object.destroy if valid_object?(object)
end
protected
# converts and documents to ids
def conditions_to_fields(conditions)
conditions.inject({}) do |fields, (key, value)|
if value.is_a?(MongoMapper::Document) && klass.key?("#{key}_id")
fields.merge("#{key}_id" => value.id)
else
fields.merge(key => value)
end
end
end
end
end
end
|
