summaryrefslogtreecommitdiff
path: root/jstests/concurrency/fsm_workloads/compact.js
blob: 855cd6e73fab7312f837a5271c86e6e9f0974679 (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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
'use strict';

/**
 * compact.js
 *
 * Bulk inserts 1000 documents and builds indexes. Then alternates between compacting the
 * collection and verifying the number of documents and indexes. Operates on a separate collection
 * for each thread.
 */

load('jstests/concurrency/fsm_workload_helpers/drop_utils.js'); // for dropCollections
load('jstests/concurrency/fsm_workload_helpers/server_types.js'); // for isEphemeral

var $config = (function() {
    var data = {
        nDocumentsToInsert: 1000,
        nIndexes: 3 + 1, // The number of indexes created in createIndexes + 1 for { _id: 1 }
        prefix: 'compact' // Use filename for prefix because filename is assumed unique
    };

    var states = (function() {
        function insertDocuments(db, collName) {
            var bulk = db[collName].initializeUnorderedBulkOp();
            for (var i = 0; i < this.nDocumentsToInsert; ++i) {
                bulk.insert({ 
                    a: Random.randInt(10),
                    b: Random.randInt(10),
                    c: Random.randInt(10)
                });
            }
            var res = bulk.execute();
            assertAlways.writeOK(res);
            assertAlways.eq(this.nDocumentsToInsert, res.nInserted);
        }

        function createIndexes(db, collName) {
            // The number of indexes created here is also stored in data.nIndexes
            var aResult = db[collName].ensureIndex({ a: 1 });
            assertAlways.commandWorked(aResult);

            var bResult = db[collName].ensureIndex({ b: 1 });
            assertAlways.commandWorked(bResult);

            var cResult = db[collName].ensureIndex({ c: 1 });
            assertAlways.commandWorked(cResult);
        }

        // This method is independent of collectionSetup to allow it to be overridden in
        // workloads that extend this one
        function init(db, collName) {
            this.threadCollName = this.prefix + '_' + this.tid; 
        }

        function collectionSetup(db, collName) {
            insertDocuments.call(this, db, this.threadCollName);
            createIndexes.call(this, db, this.threadCollName);
        }

        function compact(db, collName) {
            var res = db.runCommand({
                compact: this.threadCollName,
                paddingFactor: 1.0,
                force: true
            });
            if (!isEphemeral(db)) {
                assertAlways.commandWorked(res);
            } else {
                assertAlways.commandFailedWithCode(res, ErrorCodes.CommandNotSupported);
            }
        }

        function query(db, collName) {
            var count = db[this.threadCollName].find().itcount();
            assertWhenOwnColl.eq(count, this.nDocumentsToInsert, 'number of documents in ' +
                                 'collection should not change following a compact');
            var indexesCount = db[this.threadCollName].getIndexes().length;
            assertWhenOwnColl.eq(indexesCount, this.nIndexes);
        }

        return {
            init: init,
            collectionSetup: collectionSetup,
            compact: compact,
            query: query
        };
    })();

    var transitions = {
        init:            { collectionSetup: 1 },
        collectionSetup: { compact: 0.5, query: 0.5 },
        compact:         { compact: 0.5, query: 0.5 },
        query:           { compact: 0.5, query: 0.5 }
    };

    var teardown = function teardown(db, collName, cluster) {
        var pattern = new RegExp('^' + this.prefix + '_\\d+$');
        dropCollections(db, pattern);
    };

    return {
        threadCount: 15,
        iterations: 10,
        states: states,
        transitions: transitions,
        teardown: teardown,
        data: data
    };
})();