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
|
// Test transactions including multi-updates.
// @tags: [uses_transactions]
(function() {
"use strict";
const dbName = "test";
const collName = "multi_update_in_transaction";
const testDB = db.getSiblingDB(dbName);
const testColl = testDB[collName];
testDB.runCommand({drop: collName, writeConcern: {w: "majority"}});
assert.commandWorked(testDB.createCollection(testColl.getName(), {writeConcern: {w: "majority"}}));
const sessionOptions = {
causalConsistency: false
};
const session = db.getMongo().startSession(sessionOptions);
const sessionDb = session.getDatabase(dbName);
const sessionColl = sessionDb[collName];
jsTest.log("Prepopulate the collection.");
assert.commandWorked(testColl.insert([{_id: 0, a: 0}, {_id: 1, a: 0}, {_id: 2, a: 1}],
{writeConcern: {w: "majority"}}));
jsTest.log("Do an empty multi-update.");
session.startTransaction({writeConcern: {w: "majority"}});
// Update 0 docs.
let res = sessionColl.update({a: 99}, {$set: {b: 1}}, {multi: true});
assert.eq(0, res.nModified);
res = sessionColl.find({});
assert.sameMembers(res.toArray(), [{_id: 0, a: 0}, {_id: 1, a: 0}, {_id: 2, a: 1}]);
assert.commandWorked(session.commitTransaction_forTesting());
jsTest.log("Do a single-result multi-update.");
session.startTransaction({writeConcern: {w: "majority"}});
// Update 1 doc.
res = sessionColl.update({a: 1}, {$set: {b: 1}}, {multi: true});
assert.eq(1, res.nModified);
res = sessionColl.find({});
assert.sameMembers(res.toArray(), [{_id: 0, a: 0}, {_id: 1, a: 0}, {_id: 2, a: 1, b: 1}]);
assert.commandWorked(session.commitTransaction_forTesting());
jsTest.log("Do a multiple-result multi-update.");
session.startTransaction({writeConcern: {w: "majority"}});
// Update 2 docs.
res = sessionColl.update({a: 0}, {$set: {b: 2}}, {multi: true});
assert.eq(2, res.nModified);
res = sessionColl.find({});
assert.sameMembers(res.toArray(),
[{_id: 0, a: 0, b: 2}, {_id: 1, a: 0, b: 2}, {_id: 2, a: 1, b: 1}]);
assert.commandWorked(session.commitTransaction_forTesting());
jsTest.log("Do a multiple-query multi-update.");
session.startTransaction({writeConcern: {w: "majority"}});
// Bulk update 3 docs.
let bulk = sessionColl.initializeUnorderedBulkOp();
bulk.find({a: 0}).update({$set: {c: 1}});
bulk.find({_id: 2}).update({$set: {c: 2}});
res = assert.commandWorked(bulk.execute());
assert.eq(3, res.nModified);
res = sessionColl.find({});
assert.sameMembers(
res.toArray(),
[{_id: 0, a: 0, b: 2, c: 1}, {_id: 1, a: 0, b: 2, c: 1}, {_id: 2, a: 1, b: 1, c: 2}]);
assert.commandWorked(session.commitTransaction_forTesting());
jsTest.log("Do a multi-update with upsert.");
session.startTransaction({writeConcern: {w: "majority"}});
// Upsert 1 doc.
res = sessionColl.update({_id: 3}, {$set: {d: 1}}, {multi: true, upsert: true});
assert.eq(1, res.nUpserted);
res = sessionColl.find({});
assert.sameMembers(res.toArray(), [
{_id: 0, a: 0, b: 2, c: 1},
{_id: 1, a: 0, b: 2, c: 1},
{_id: 2, a: 1, b: 1, c: 2},
{_id: 3, d: 1}
]);
assert.commandWorked(session.commitTransaction_forTesting());
}());
|
