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
|
'use strict';
/**
* agg_graph_lookup.js
*
* Runs a $graphLookup aggregation simultaneously with updates.
*/
var $config = (function() {
const data = {numDocs: 1000};
const states = (function() {
function query(db, collName) {
const limitAmount = 20;
const startingId = Random.randInt(this.numDocs - limitAmount);
function getQueryResults() {
let arr = null;
try {
const cursor = db[collName]
.aggregate([
{$match: {_id: {$gt: startingId}}},
{
$graphLookup: {
from: collName,
startWith: "$to",
connectToField: "_id",
connectFromField: "to",
maxDepth: 10,
as: "out",
}
},
{$limit: limitAmount}
]);
arr = cursor.toArray();
} catch (e) {
if (TestData.runningWithShardStepdowns) {
// When running with stepdowns, we expect to sometimes see the query
// killed.
assert.eq(e.code, ErrorCodes.QueryPlanKilled);
} else {
throw e;
}
}
return arr;
}
const res = getQueryResults();
if (res) {
assertWhenOwnColl.eq(res.length, limitAmount);
}
}
function update(db, collName) {
const index = Random.randInt(this.numDocs + 1);
const update = Random.randInt(this.numDocs + 1);
const res = db[collName].update({_id: index}, {$set: {to: update}});
assertWhenOwnColl.commandWorked(res);
}
return {query, update};
})();
const transitions = {query: {query: 0.5, update: 0.5}, update: {query: 0.5, update: 0.5}};
function setup(db, collName, cluster) {
// Load example data.
const bulk = db[collName].initializeUnorderedBulkOp();
for (let i = 0; i < this.numDocs; ++i) {
bulk.insert({_id: i, to: i + 1});
}
const res = bulk.execute();
assertWhenOwnColl.commandWorked(res);
assertWhenOwnColl.eq(this.numDocs, res.nInserted);
assertWhenOwnColl.eq(this.numDocs, db[collName].find().itcount());
}
return {
threadCount: 10,
iterations: 100,
states: states,
startState: 'query',
transitions: transitions,
data: data,
setup: setup,
};
})();
|
