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
|
'use strict';
/**
* Perform CRUD operations in parallel on a clustered collection. Disallows dropping the collection
* to prevent implicit creation of a non-clustered collection.
*
* @tags: [
* requires_fcv_53
* ]
*/
load('jstests/concurrency/fsm_libs/extend_workload.js'); // for extendWorkload
load('jstests/concurrency/fsm_workloads/CRUD_and_commands.js'); // for $config
var $config = extendWorkload($config, function($config, $super) {
// Exclude dropCollection to prevent implicit collection creation of a non-clustered
// collection.
const newStates = $super.states;
delete newStates.dropCollection;
$config.states = Object.extend({
createIndex: function createIndex(db, collName) {
db[collName].createIndex({value: 1});
},
dropIndex: function dropIndex(db, collName) {
db[collName].dropIndex({value: 1});
},
listCollections: function listCollections(db, collName) {
const listColls =
assert.commandWorked(db.runCommand({listCollections: 1, filter: {name: collName}}));
const listCollsOptions = listColls.cursor.firstBatch[0].options;
assert(listCollsOptions.clusteredIndex);
},
},
newStates);
$config.setup = function(db, coll, cluster) {
// As the default collection created by runner.js won't be clustered we need to recreate it.
db[coll].drop();
assertAlways.commandWorked(
db.runCommand({create: coll, clusteredIndex: {key: {_id: 1}, unique: true}}));
for (let i = 0; i < this.numIds; i++) {
const res = db[coll].insert({_id: i, value: this.docValue, num: 1});
assertAlways.commandWorked(res);
assert.eq(1, res.nInserted);
}
if (cluster.isSharded()) {
cluster.shardCollection(db[coll], {_id: 1}, true);
}
};
// Exclude dropCollection to prevent implicit collection creation of a non-clustered
// collection.
const newTransitions = Object.extend({}, $super.transitions);
delete newTransitions.dropCollection;
let defaultTransitionWeights = {};
Object.keys(newTransitions).forEach(function(stateName) {
if (newTransitions[stateName].dropCollection) {
delete newTransitions[stateName]["dropCollection"];
}
newTransitions[stateName]["createIndex"] = 0.10;
newTransitions[stateName]["dropIndex"] = 0.10;
newTransitions[stateName]["listCollections"] = 0.10;
if (stateName !== "init") {
defaultTransitionWeights = newTransitions[stateName];
}
});
newTransitions["createIndex"] = defaultTransitionWeights;
newTransitions["dropIndex"] = defaultTransitionWeights;
newTransitions["listCollections"] = defaultTransitionWeights;
$config.transitions = newTransitions;
return $config;
});
|
