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
|
/**
* Tests that the analyzeShardKey command returns monotonicity metrics for compound shard keys that
* use range sharding.
*
* @tags: [requires_fcv_70, resource_intensive]
*/
(function() {
"use strict";
load("jstests/sharding/analyze_shard_key/libs/monotonicity_common.js");
// Construct test cases for testing the monotonicity of compound shard keys that use range sharding.
// For each test case:
// - 'shardKey' is the shard key being analyzed.
// - 'indexKey' is the supporting index for the shard key.
// - 'fieldOpts' specifies the name, type and order for each field inside the documents in the
// collection. The order refers to whether the value is constant, fluctuating, increasing or
// decreasing.
// - 'expected' is the expected monotonicity. Since the shard key is compound, its monotonicity is
// determined by the monotonicity of the first non-constant shard key field.
const testCases = [];
for (let orderType0 of kOrderTypes) {
const fieldName0 = AnalyzeShardKeyUtil.getRandomFieldName("a");
const fieldType0 = AnalyzeShardKeyUtil.getRandomElement(orderType0.supportedFieldTypes);
for (let orderType1 of kOrderTypes) {
const fieldName1 = AnalyzeShardKeyUtil.getRandomFieldName("b");
const fieldType1 = AnalyzeShardKeyUtil.getRandomElement(orderType1.supportedFieldTypes);
// Test compound shard key with a shard key index.
testCases.push({
shardKey: {[fieldName0]: 1, [fieldName1]: 1},
indexKey: {[fieldName0]: 1, [fieldName1]: 1},
fieldOpts: [
{name: fieldName0, type: fieldType0, order: orderType0.name},
{name: fieldName1, type: fieldType1, order: orderType1.name}
],
expected: orderType0.name == "constant" ? orderType1.monotonicity
: orderType0.monotonicity
});
for (let orderType2 of kOrderTypes) {
const fieldName2 = AnalyzeShardKeyUtil.getRandomFieldName("c");
const fieldType2 = AnalyzeShardKeyUtil.getRandomElement(orderType2.supportedFieldTypes);
// Test compound shard key with a compound shard key prefixed index.
testCases.push({
shardKey: {[fieldName0]: 1, [fieldName1]: 1},
indexKey: {[fieldName0]: 1, [fieldName1]: 1, [fieldName2]: 1},
fieldOpts: [
{name: fieldName0, type: fieldType0, order: orderType0.name},
{name: fieldName1, type: fieldType1, order: orderType1.name},
{name: fieldName2, type: fieldType2, order: orderType2.name}
],
expected: orderType0.name == "constant" ? orderType1.monotonicity
: orderType0.monotonicity
});
}
}
}
// This test requires the collection to contain at least a few thousands of documents to smooth out
// the noise in the insertion order caused by the oplog application batching on secondaries.
const numDocsRange = {
min: 7500,
max: 10000
};
const setParameterOpts = {
// Skip calculating the read and write distribution metrics since there are no sampled queries
// anyway.
"failpoint.analyzeShardKeySkipCalcalutingReadWriteDistributionMetrics":
tojson({mode: "alwaysOn"})
};
{
const st = new ShardingTest({shards: 2, rs: {nodes: 2, setParameter: setParameterOpts}});
testAnalyzeShardKeysUnshardedCollection(st.s, testCases, numDocsRange);
testAnalyzeShardKeysShardedCollection(st, testCases, numDocsRange);
st.stop();
}
{
const rst = new ReplSetTest({nodes: 2, nodeOptions: {setParameter: setParameterOpts}});
rst.startSet();
rst.initiate();
const primary = rst.getPrimary();
testAnalyzeShardKeysUnshardedCollection(primary, testCases, numDocsRange);
rst.stopSet();
}
})();
|
