Module: MongoMapper::Plugins::Modifiers::ClassMethods

Defined in:
lib/mongo_mapper/plugins/modifiers.rb

Instance Method Summary collapse

Instance Method Details

#add_to_set(*args) ⇒ Object Also known as: push_uniq 



59
60
61
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 59

def add_to_set(*args)
  modifier_update('$addToSet', args)
end

#decrement(*args) ⇒ Object 



12
13
14
15
16
17
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 12

def decrement(*args)
  criteria, keys, options = criteria_and_keys_from_args(args)
  values, to_decrement = keys.values, {}
  keys.keys.each_with_index { |k, i| to_decrement[k] = -values[i].abs }
  collection.update_many(criteria, {'$inc' => to_decrement}, options || {})
end

#find_one_and_update(args) ⇒ Object Also known as: find_and_modify 



76
77
78
79
80
81
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 76

def find_one_and_update(args)
  args = args.dup
  args[:query]  = dealias_keys(args.delete :query)  if args.key? :query
  args[:update] = dealias_keys(args.delete :update) if args.key? :update
  collection.find_one_and_update(args[:query], args[:update], args)
end

#increment(*args) ⇒ Object 



8
9
10
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 8

def increment(*args)
  modifier_update('$inc', args)
end

#pop(*args) ⇒ Object 



72
73
74
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 72

def pop(*args)
  modifier_update('$pop', args)
end

#pull(*args) ⇒ Object 



64
65
66
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 64

def pull(*args)
  modifier_update('$pull', args)
end

#pull_all(*args) ⇒ Object 



68
69
70
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 68

def pull_all(*args)
  modifier_update('$pullAll', args)
end

#push(*args) ⇒ Object 



42
43
44
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 42

def push(*args)
  modifier_update('$push', args)
end

#push_all(*args) ⇒ Object 



46
47
48
49
50
51
52
53
54
55
56
57
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 46

def push_all(*args)
  Kernel.warn "push_all no longer supported. use $push with $each"

  hash = args.pop
  ids = args

  push_values = hash.inject({}) do |hsh, (key, values)|
    { key => { '$each' => values } }
  end

  modifier_update('$addToSet', [ids, push_values].flatten)
end

#set(*args) ⇒ Object 



19
20
21
22
23
24
25
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 19

def set(*args)
  criteria, updates, options = criteria_and_keys_from_args(args)
  updates.each do |key, value|
    updates[key] = keys[key.to_s].set(value) if key?(key)
  end
  modifier_update('$set', [criteria, updates, options])
end

#unset(*args) ⇒ Object 



27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 27

def unset(*args)
  if args[0].is_a?(Hash)
    criteria, keys = args.shift, args
    options = keys.last.is_a?(Hash) ? keys.pop : {}
  else
    keys, ids = args.partition { |arg| arg.is_a?(Symbol) }
    options = ids.last.is_a?(Hash) ? ids.pop : {}
    criteria = {:id => ids}
  end

  criteria = criteria_hash(criteria).to_hash
  updates = keys.inject({}) { |hash, key| hash[key] = 1; hash }
  modifier_update('$unset', [criteria, updates, options])
end

#upsert(selector, updates, args = {}) ⇒ Object 



84
85
86
87
88
 
# File 'lib/mongo_mapper/plugins/modifiers.rb', line 84

def upsert(selector, updates, args = {})
  criteria = dealias_keys(selector)
  updates  = dealias_keys(updates)
  collection.update_one(criteria, updates, args.merge(upsert: true))
end