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
|
# Copyright (C) 2014-2017 MongoDB, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
require 'mongo/operation/write/update/result'
module Mongo
module Operation
module Write
# A MongoDB update operation.
#
# @note If the server version is >= 2.5.5, a write command operation
# will be created and sent instead.
#
# @example Create the update operation.
# Write::Update.new({
# :update =>
# {
# :q => { :foo => 1 },
# :u => { :$set => { :bar => 1 }},
# :multi => true,
# :upsert => false
# :array_filters => []
# },
# :db_name => 'test',
# :coll_name => 'test_coll',
# :write_concern => write_concern
# })
# Initialization:
# param [ Hash ] spec The specifications for the update.
#
# option spec :update [ Hash ] The update document.
# option spec :db_name [ String ] The name of the database on which
# the query should be run.
# option spec :coll_name [ String ] The name of the collection on which
# the query should be run.
# option spec :write_concern [ Mongo::WriteConcern ] The write concern.
# option spec :options [ Hash ] Options for the command, if it ends up being a
# write command.
#
# @since 2.0.0
class Update
include GLE
include WriteCommandEnabled
include Specifiable
private
def write_command_op
s = spec.merge(:updates => [ update ])
s.delete(:update)
Command::Update.new(s)
end
def has_array_filters?
update[Operation::ARRAY_FILTERS]
end
def has_collation?
update[:collation] || update[Operation::COLLATION]
end
def message(server)
flags = []
flags << :multi_update if update[Operation::MULTI]
flags << :upsert if update[Operation::UPSERT]
Protocol::Update.new(
db_name,
coll_name,
update[Operation::Q],
update[Operation::U],
flags.empty? ? {} : { flags: flags }
)
end
end
end
end
end
|
