summaryrefslogtreecommitdiff
path: root/jstests/concurrency/fsm_workloads/count.js
blob: 626ac49a4c80d128a7c193c37ab6edd1ed304141 (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
67
68
69
70
71
72
73
74
75
76
'use strict';

/**
 * count.js
 *
 * Base workload for count.
 * Runs count on a non-indexed field and verifies that the count
 * is correct.
 * Each thread picks a random 'modulus' in range [5, 10]
 * and a random 'countPerNum' in range [50, 100]
 * and then inserts 'modulus * countPerNum' documents. [250, 1000]
 * All threads insert into the same collection.
 */
var $config = (function() {

    var data = {
        randRange: function randRange(low, high) {
            // return random number in range [low, high]
            assert.gt(high, low);
            return low + Random.randInt(high - low + 1);
        },
        getNumDocs: function getNumDocs() {
            return this.modulus * this.countPerNum;
        },
        getCount: function getCount(db, predicate) {
            var query = Object.extend({ tid: this.tid }, predicate);
            return db[this.threadCollName].count(query);
        }
    };

    var states = (function() {

        function init(db, collName) {
            this.modulus = this.randRange(5, 10);
            this.countPerNum = this.randRange(50, 100);

            // workloads that extend this one might have set 'this.threadCollName'
            this.threadCollName = this.threadCollName || collName;

            var bulk = db[this.threadCollName].initializeUnorderedBulkOp();
            for (var i = 0; i < this.getNumDocs(); ++i) {
                bulk.insert({ i: i % this.modulus, tid: this.tid });
            }
            var res = bulk.execute();
            assertAlways.writeOK(res);
            assertAlways.eq(this.getNumDocs(), res.nInserted);
        }

        function count(db, collName) {
            assertWhenOwnColl.eq(this.getCount(db), this.getNumDocs());

            var num = Random.randInt(this.modulus);
            assertWhenOwnColl.eq(this.getCount(db, { i: num }), this.countPerNum);
        }

        return {
            init: init,
            count: count
        };

    })();

    var transitions = {
        init: { count: 1 },
        count: { count: 1 }
    };

    return {
        data: data,
        threadCount: 10,
        iterations: 20,
        states: states,
        transitions: transitions
    };

})();