summaryrefslogtreecommitdiff
path: root/jstests/concurrency/fsm_workloads/agg_graph_lookup.js
blob: b10fc88a99dda384d8a01eb6e7a1f07b56cc19fc (plain)
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
'use strict';

/**
 * agg_graph_lookup.js
 *
 * Runs a $graphLookup aggregation simultaneously with updates.
 */
var $config = (function() {

    var data = {numDocs: 1000};

    var states = {
        query: function query(db, collName) {
            var limitAmount = 20;
            var startingId = Random.randInt(this.numDocs - limitAmount);
            var res = db[collName]
                          .aggregate([
                              {$match: {_id: {$gt: startingId}}},
                              {
                                $graphLookup: {
                                    from: collName,
                                    startWith: "$to",
                                    connectToField: "_id",
                                    connectFromField: "to",
                                    maxDepth: 10,
                                    as: "out",
                                }
                              },
                              {$limit: limitAmount}
                          ])
                          .toArray();

            assertWhenOwnColl.eq(res.length, limitAmount);
        },
        update: function update(db, collName) {
            var index = Random.randInt(this.numDocs + 1);
            var update = Random.randInt(this.numDocs + 1);
            var res = db[collName].update({_id: index}, {$set: {to: update}});
            assertWhenOwnColl.writeOK(res);
        }
    };

    var transitions = {query: {query: 0.5, update: 0.5}, update: {query: 0.5, update: 0.5}};

    function setup(db, collName, cluster) {
        // Load example data.
        var bulk = db[collName].initializeUnorderedBulkOp();
        for (var i = 0; i < this.numDocs; ++i) {
            bulk.insert({_id: i, to: i + 1});
        }
        var res = bulk.execute();
        assertWhenOwnColl.writeOK(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,
    };
})();