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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
/**
* Tests for basic functionality of the resharding improvements feature.
*
* @tags: [
* require_fcv_71,
* featureFlagReshardingImprovements
* ]
*/
(function() {
'use strict';
load("jstests/libs/feature_flag_util.js");
load("jstests/sharding/libs/reshard_collection_util.js");
const st = new ShardingTest({mongos: 1, shards: 2});
const kDbName = 'db';
const collName = 'foo';
const ns = kDbName + '.' + collName;
const mongos = st.s0;
const kNumInitialDocs = 500;
const reshardCmdTest =
new ReshardCollectionCmdTest({st, dbName: kDbName, collName, numInitialDocs: kNumInitialDocs});
const testShardDistribution = (mongos) => {
if (!FeatureFlagUtil.isEnabled(mongos, "ReshardingImprovements")) {
jsTestLog("Skipping test since featureFlagReshardingImprovements is not enabled");
return;
}
assert.commandWorked(mongos.adminCommand({enableSharding: kDbName}));
assert.commandWorked(mongos.adminCommand({shardCollection: ns, key: {oldKey: 1}}));
jsTest.log("reshardCollection cmd should fail when shardDistribution is missing min or max.");
assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: ns,
key: {newKey: 1},
shardDistribution: [
{shard: st.shard0.shardName, min: {newKey: MinKey}},
{shard: st.shard1.shardName, max: {newKey: MaxKey}}
]
}),
ErrorCodes.InvalidOptions);
jsTest.log(
"reshardCollection cmd should fail when shardDistribution is not specified using the shard key.");
assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: ns,
key: {newKey: 1},
shardDistribution: [
{shard: st.shard0.shardName, min: {oldKey: MinKey}, max: {oldKey: 0}},
{shard: st.shard1.shardName, min: {oldKey: 0}, max: {oldKey: MaxKey}}
]
}),
ErrorCodes.InvalidOptions);
jsTest.log(
"reshardCollection cmd should fail when one shard specifies min/max and the other does not.");
assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: ns,
key: {newKey: 1},
shardDistribution: [
{shard: st.shard0.shardName},
{shard: st.shard1.shardName, min: {newKey: MinKey}, max: {newKey: MaxKey}}
]
}),
ErrorCodes.InvalidOptions);
jsTest.log(
"reshardCollection cmd should fail when shardDistribution is not starting with globalMin.");
assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: ns,
key: {newKey: 1},
shardDistribution: [
{shard: st.shard0.shardName, min: {newKey: -1}, max: {newKey: 0}},
{shard: st.shard1.shardName, min: {newKey: 0}, max: {newKey: MaxKey}}
]
}),
ErrorCodes.InvalidOptions);
jsTest.log("reshardCollection cmd should fail when shardDistribution is not continuous.");
assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: ns,
key: {newKey: 1},
shardDistribution: [
{shard: st.shard0.shardName, min: {newKey: MinKey}, max: {newKey: -1}},
{shard: st.shard1.shardName, min: {newKey: 0}, max: {newKey: MaxKey}}
]
}),
ErrorCodes.InvalidOptions);
jsTest.log(
"reshardCollection cmd should fail when the shardId in shardDistribution is not recognized.");
assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: ns,
key: {newKey: 1},
shardDistribution: [
{shard: "s1", min: {newKey: MinKey}, max: {newKey: 0}},
{shard: "s2", min: {newKey: 0}, max: {newKey: MaxKey}}
]
}),
ErrorCodes.ShardNotFound);
jsTest.log("reshardCollection cmd should succeed with shardDistribution parameter.");
// TODO(SERVER-76791): This should work after supporting non-explicit form of shardDistribution.
assert.commandFailedWithCode(mongos.adminCommand({
reshardCollection: ns,
key: {newKey: 1},
shardDistribution: [{shard: st.shard0.shardName}, {shard: st.shard1.shardName}]
}),
ErrorCodes.InvalidOptions);
reshardCmdTest.assertReshardCollOk(
{
reshardCollection: ns,
key: {newKey: 1},
shardDistribution: [
{shard: st.shard0.shardName, min: {newKey: MinKey}, max: {newKey: 0}},
{shard: st.shard1.shardName, min: {newKey: 0}, max: {newKey: MaxKey}}
]
},
2,
[
{recipientShardId: st.shard0.shardName, min: {newKey: MinKey}, max: {newKey: 0}},
{recipientShardId: st.shard1.shardName, min: {newKey: 0}, max: {newKey: MaxKey}}
]);
};
testShardDistribution(mongos);
st.stop();
})();
|
