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
|
/**
* Tests the collectionUUID parameter of the aggregate command for $indexStats and $collStats
* pipelines on sharded collection.
* @tags: [featureFlagCommandsAcceptCollectionUUID]
*/
(function() {
'use strict';
load("jstests/libs/write_concern_util.js"); // For 'shardCollectionWithChunks'
const validateErrorResponse = function(res, collectionUUID, actualNamespace) {
assert.eq(res.collectionUUID, collectionUUID);
assert.eq(res.actualNamespace, actualNamespace);
};
const getUUID = function(database, collName) {
return assert.commandWorked(database.runCommand({listCollections: 1}))
.cursor.firstBatch.find(c => c.name === collName)
.info.uuid;
};
const st = new ShardingTest({
name: jsTestName(),
shards: 2,
rs: {setParameter: {logComponentVerbosity: tojson({command: {verbosity: 1}})}}
});
const testDB = st.s.getDB(jsTestName());
assert.commandWorked(testDB.dropDatabase());
const shardedCollection = testDB["shardedCollection"];
// Set up sharded collection. Put 5 documents on each shard, with keys {x: 0...9}.
shardCollectionWithChunks(st, shardedCollection, 10 /* numDocs */);
const sameShardColl = testDB["sameShardCollection"];
assert.commandWorked(sameShardColl.insert({x: 1, y: 1}));
const otherDB = testDB.getSiblingDB("otherDB");
assert.commandWorked(otherDB.dropDatabase());
const otherShardColl = otherDB["otherShardColl"];
assert.commandWorked(otherShardColl.insert({x: 1, y: 1}));
// Make sure that the primary shard is different for the shardedCollection and the otherShardColl.
st.ensurePrimaryShard(testDB.getName(), st.shard0.shardName);
st.ensurePrimaryShard(otherDB.getName(), st.shard1.shardName);
const otherShardUUID = getUUID(otherDB, otherShardColl.getName());
const shardedUUID = getUUID(testDB, shardedCollection.getName());
const sameShardUUID = getUUID(testDB, sameShardColl.getName());
const testCommand = function(cmd, cmdObj) {
jsTestLog("The command '" + cmd +
"' succeeds on a sharded collection when the correct UUID is provided.");
cmdObj[cmd] = shardedCollection.getName();
cmdObj["collectionUUID"] = shardedUUID;
assert.commandWorked(testDB.runCommand(cmdObj));
jsTestLog("The command '" + cmd +
"' fails on sharded collection when the provided UUID corresponds to a different " +
"collection:");
jsTestLog("If the aggregation command hits all shards, then it should return the " +
"actual namespace of the unsharded collection that has same primary shard.");
cmdObj["collectionUUID"] = sameShardUUID;
let res =
assert.commandFailedWithCode(testDB.runCommand(cmdObj), ErrorCodes.CollectionUUIDMismatch);
validateErrorResponse(res, sameShardUUID, sameShardColl.getFullName());
jsTestLog("If the aggregation command hits all shards, then it should return the " +
"actual namespace of the unsharded collection that has different primary shard.");
cmdObj["collectionUUID"] = otherShardUUID;
res =
assert.commandFailedWithCode(testDB.runCommand(cmdObj), ErrorCodes.CollectionUUIDMismatch);
validateErrorResponse(res, otherShardUUID, otherShardColl.getFullName());
jsTestLog(
"If the aggregation command hits only one shards, then it can't find the actual namespace" +
" when the provided UUID corresponds to an unsharded collection that has different " +
"primary shard.");
cmdObj[cmd] = sameShardColl.getName();
cmdObj["collectionUUID"] = otherShardUUID;
res =
assert.commandFailedWithCode(testDB.runCommand(cmdObj), ErrorCodes.CollectionUUIDMismatch);
validateErrorResponse(res, otherShardUUID, "");
};
testCommand("aggregate", {aggregate: "", pipeline: [{$collStats: {latencyStats: {}}}], cursor: {}});
testCommand("aggregate", {aggregate: "", pipeline: [{$indexStats: {}}], cursor: {}});
st.stop();
})();
|
