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
93
94
95
96
97
98
99
|
/**
* Tests the collectionUUID parameter of the reshardCollection command.
*
* @tags: [
* featureFlagCommandsAcceptCollectionUUID,
* ]
*/
(function() {
'use strict';
const st = new ShardingTest({shards: 1});
const mongos = st.s0;
const db = mongos.getDB(jsTestName());
const coll = db['coll'];
assert.commandWorked(mongos.adminCommand({enableSharding: db.getName()}));
const oldKeyDoc = {
a: 1
};
const newKeyDoc = {
b: 1
};
const resetColl = function(shardedColl) {
shardedColl.drop();
assert.commandWorked(shardedColl.insert({a: 1, b: 2}));
assert.commandWorked(mongos.getCollection(shardedColl.getFullName()).createIndex(oldKeyDoc));
assert.commandWorked(mongos.getCollection(shardedColl.getFullName()).createIndex(newKeyDoc));
assert.commandWorked(
mongos.adminCommand({shardCollection: shardedColl.getFullName(), key: oldKeyDoc}));
};
const uuid = function() {
return assert.commandWorked(db.runCommand({listCollections: 1}))
.cursor.firstBatch.find(c => c.name === coll.getName())
.info.uuid;
};
resetColl(coll);
// The command succeeds when provided with the correct collection UUID.
assert.commandWorked(mongos.adminCommand(
{reshardCollection: coll.getFullName(), key: newKeyDoc, collectionUUID: uuid()}));
// The command fails when provided with a UUID with no corresponding collection.
resetColl(coll);
const nonexistentUUID = UUID();
let res = assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: coll.getFullName(),
key: newKeyDoc,
collectionUUID: nonexistentUUID,
}),
ErrorCodes.CollectionUUIDMismatch);
assert.eq(res.collectionUUID, nonexistentUUID);
assert.eq(res.expectedNamespace, coll.getFullName());
assert.eq(res.actualNamespace, "");
// The command fails when provided with a different collection's UUID.
const coll2 = db['coll_2'];
resetColl(coll2);
res = assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: coll2.getFullName(),
key: newKeyDoc,
collectionUUID: uuid(),
}),
ErrorCodes.CollectionUUIDMismatch);
assert.eq(res.collectionUUID, uuid());
assert.eq(res.expectedNamespace, coll2.getFullName());
assert.eq(res.actualNamespace, coll.getFullName());
// The command fails when provided with a different collection's UUID, even if the provided
// namespace does not exist.
coll2.drop();
res = assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: coll2.getFullName(),
key: newKeyDoc,
collectionUUID: uuid(),
}),
ErrorCodes.CollectionUUIDMismatch);
assert.eq(res.collectionUUID, uuid());
assert.eq(res.expectedNamespace, coll2.getFullName());
assert.eq(res.actualNamespace, coll.getFullName());
// The command fails when the provided UUID corresponds to a different collection, even if the
// provided namespace is a view.
const view = db['view'];
assert.commandWorked(db.createView(view.getName(), coll.getName(), []));
res = assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: view.getFullName(),
key: newKeyDoc,
collectionUUID: uuid(),
}),
ErrorCodes.CollectionUUIDMismatch);
assert.eq(res.collectionUUID, uuid());
assert.eq(res.expectedNamespace, view.getFullName());
assert.eq(res.actualNamespace, coll.getFullName());
st.stop();
})();
|
