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 the _configsvrRemoveChunks internal command.
* @tags: [
* ]
*/
(function() {
'use strict';
load("jstests/libs/retryable_writes_util.js");
load("jstests/sharding/libs/find_chunks_util.js");
function runConfigsvrRemoveChunksWithRetries(conn, uuid, lsid, txnNumber) {
var res;
assert.soon(() => {
res = st.configRS.getPrimary().adminCommand({
_configsvrRemoveChunks: 1,
collectionUUID: uuid,
lsid: lsid,
txnNumber: txnNumber,
writeConcern: {w: "majority"}
});
if (RetryableWritesUtil.isRetryableCode(res.code) ||
RetryableWritesUtil.errmsgContainsRetryableCodeName(res.errmsg) ||
(res.writeConcernError &&
RetryableWritesUtil.isRetryableCode(res.writeConcernError.code))) {
return false; // Retry
}
return true;
});
return res;
}
function insertLeftoverChunks(configDB, uuid) {
const chunkDocsForNs = findChunksUtil.findChunksByNs(configDB, ns).toArray();
var chunksToInsert = [];
chunkDocsForNs.forEach(originalChunk => {
var newChunk = originalChunk;
newChunk._id = ObjectId();
newChunk.uuid = otherCollectionUUID;
chunksToInsert.push(newChunk);
});
assert.commandWorked(
configDB.getCollection("chunks").insertMany(chunksToInsert, {ordered: false}));
}
let st = new ShardingTest({mongos: 1, shards: 1});
const configDB = st.s.getDB('config');
const dbName = "test";
const collName = "foo";
const ns = dbName + "." + collName;
let lsid = assert.commandWorked(st.s.getDB("admin").runCommand({startSession: 1})).id;
assert.commandWorked(st.s.adminCommand({enableSharding: dbName}));
assert.commandWorked(st.s.adminCommand({shardcollection: ns, key: {_id: 1}}));
// Insert some chunks not associated to any collection to simulate leftovers from a failed
// shardCollection
const otherCollectionUUID = UUID();
insertLeftoverChunks(configDB, otherCollectionUUID);
assert.eq(1, configDB.getCollection("chunks").find({uuid: otherCollectionUUID}).itcount());
// Remove the leftover chunks matching 'otherCollectionUUID'
assert.commandWorked(runConfigsvrRemoveChunksWithRetries(
st.configRS.getPrimary(), otherCollectionUUID, lsid, NumberLong(1)));
assert.eq(0, configDB.getCollection("chunks").find({uuid: otherCollectionUUID}).itcount());
// Insert new leftover chunks
insertLeftoverChunks(configDB, otherCollectionUUID);
assert.eq(1, configDB.getCollection("chunks").find({uuid: otherCollectionUUID}).itcount());
// Check that _configsvrRemoveChunks with a txnNumber lesser than the previous one won't remove the
// chunk documents
assert.commandFailedWithCode(
runConfigsvrRemoveChunksWithRetries(
st.configRS.getPrimary(), otherCollectionUUID, lsid, NumberLong(0)),
ErrorCodes.TransactionTooOld);
assert.eq(1, configDB.getCollection("chunks").find({uuid: otherCollectionUUID}).itcount());
// Cleanup the leftover chunks before finishing
configDB.getCollection("chunks").remove({uuid: otherCollectionUUID});
st.stop();
})();
|
