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
|
/**
* Tests that resumable index builds in the drain writes phase write their state to disk upon clean
* shutdown and are resumed from the same phase to completion when the node is started back up.
*
* @tags: [
* requires_majority_read_concern,
* requires_persistence,
* requires_replication,
* ]
*/
(function() {
"use strict";
load("jstests/noPassthrough/libs/index_build.js");
load("jstests/libs/sbe_util.js"); // For checkSBEEnabled.
const dbName = "test";
const rst = new ReplSetTest({nodes: 1});
rst.startSet();
rst.initiate();
const columnstoreEnabled =
checkSBEEnabled(rst.getPrimary().getDB(dbName), ["featureFlagColumnstoreIndexes"], true);
const runTests = function(docs, indexSpecsFlat, sideWrites, collNameSuffix) {
const coll = rst.getPrimary().getDB(dbName).getCollection(jsTestName() + collNameSuffix);
assert.commandWorked(coll.insert(docs));
const runTest = function(indexSpecs, iteration) {
ResumableIndexBuildTest.run(
rst,
dbName,
coll.getName(),
indexSpecs,
[{name: "hangIndexBuildDuringDrainWritesPhase", logIdWithIndexName: 4841800}],
iteration,
["drain writes"],
[{skippedPhaseLogID: 20392}],
sideWrites,
[{a: 4}, {a: 5}]);
};
runTest([[indexSpecsFlat[0]]], 0);
runTest([[indexSpecsFlat[0]]], 1);
runTest([[indexSpecsFlat[0]], [indexSpecsFlat[1]]], 0);
runTest([[indexSpecsFlat[0]], [indexSpecsFlat[1]]], 1);
runTest([indexSpecsFlat], 0);
runTest([indexSpecsFlat], 1);
};
runTests({a: 1}, [{a: 1}, {b: 1}], [{a: 2, b: 2}, {a: 3, b: 3}], "");
runTests({a: 1}, [{a: 1}, {b: 1}], [{a: [2, 3], b: [2, 3]}, {a: 3, b: 3}], "_multikey_first");
runTests({a: 1}, [{a: 1}, {b: 1}], [{a: 2, b: 2}, {a: [3, 4], b: [3, 4]}], "_multikey_last");
runTests({a: 1}, [{a: 1}, {b: 1}], [{a: [2, 3], b: 2}, {a: 3, b: [3, 4]}], "_multikey_mixed");
runTests({a: 1},
[{"$**": 1}, {h: 1}],
[{a: [1, 2], b: {c: [3, 4]}, d: ""}, {e: "", f: [[]], g: null, h: 8}],
"_wildcard");
if (columnstoreEnabled) {
runTests({a: 1},
[{"$**": "columnstore"}, {h: 1}],
(function(collection) {
assert.commandWorked(collection.insert([{a: [{c: 2}], b: 2}, {a: 3, b: 3}]));
assert.commandWorked(collection.update({a: 3}, {a: 4, b: 3}));
assert.commandWorked(collection.remove({"a.c": 2}));
assert.commandWorked(collection.insert({a: 4, b: 4}));
assert.commandWorked(collection.remove({b: 3}));
assert.commandWorked(collection.update({a: 4}, {a: 2}));
}),
"_columnstore");
}
rst.stopSet();
})();
|
