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
|
(function() {
'use strict';
var s = new ShardingTest({shards: 2});
// Make sure that findAndModify with upsert against a non-existent database and collection will
// implicitly create them both
assert.eq(undefined,
assert.commandWorked(s.s0.adminCommand({listDatabases: 1, nameOnly: 1}))
.databases.find((dbInfo) => {
return (dbInfo.name === 'NewUnshardedDB');
}));
var newlyCreatedDb = s.getDB('NewUnshardedDB');
assert.eq(0, newlyCreatedDb.unsharded_coll.find({}).itcount());
newlyCreatedDb.unsharded_coll.findAndModify(
{query: {_id: 1}, update: {$set: {Value: 'Value'}}, upsert: true});
assert.eq(1, newlyCreatedDb.unsharded_coll.find({}).itcount());
// Tests with sharded database
assert.commandWorked(s.s0.adminCommand({enablesharding: "test"}));
s.ensurePrimaryShard('test', s.shard1.shardName);
assert.commandWorked(s.s0.adminCommand({shardcollection: "test.sharded_coll", key: {_id: 1}}));
var db = s.getDB('test');
var numObjs = 20;
// Pre-split the collection so to avoid interference from auto-split
assert.commandWorked(
s.s0.adminCommand({split: "test.sharded_coll", middle: {_id: numObjs / 2}}));
assert.commandWorked(s.s0.adminCommand(
{movechunk: "test.sharded_coll", find: {_id: numObjs / 2}, to: s.shard0.shardName}));
var bulk = db.sharded_coll.initializeUnorderedBulkOp();
for (var i = 0; i < numObjs; i++) {
bulk.insert({_id: i});
}
assert.writeOK(bulk.execute());
// Put two docs in each chunk (avoid the split in 0, since there are no docs less than 0)
for (var i = 2; i < numObjs; i += 2) {
if (i == numObjs / 2)
continue;
assert.commandWorked(s.s0.adminCommand({split: "test.sharded_coll", middle: {_id: i}}));
}
s.printChunks();
assert.eq(
numObjs / 2, s.config.chunks.count({"ns": "test.sharded_coll"}), 'Split was incorrect');
assert.eq(numObjs / 4,
s.config.chunks.count({shard: s.shard0.shardName, "ns": "test.sharded_coll"}));
assert.eq(numObjs / 4,
s.config.chunks.count({shard: s.shard1.shardName, "ns": "test.sharded_coll"}));
// update
for (var i = 0; i < numObjs; i++) {
assert.eq(db.sharded_coll.count({b: 1}), i, "2 A");
var out = db.sharded_coll.findAndModify({query: {_id: i, b: null}, update: {$set: {b: 1}}});
assert.eq(out._id, i, "2 E");
assert.eq(db.sharded_coll.count({b: 1}), i + 1, "2 B");
}
// remove
for (var i = 0; i < numObjs; i++) {
assert.eq(db.sharded_coll.count(), numObjs - i, "3 A");
assert.eq(db.sharded_coll.count({_id: i}), 1, "3 B");
var out = db.sharded_coll.findAndModify({remove: true, query: {_id: i}});
assert.eq(db.sharded_coll.count(), numObjs - i - 1, "3 C");
assert.eq(db.sharded_coll.count({_id: i}), 0, "3 D");
assert.eq(out._id, i, "3 E");
}
s.stop();
})();
|
