summaryrefslogtreecommitdiff
path: root/jstests/concurrency/fsm_workloads/indexed_insert_heterogeneous.js
blob: 5306facb834b2f5bf21abee4b46aa452b7ccd2e9 (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
'use strict';

/**
 * indexed_insert_heterogeneous.js
 *
 * Inserts multiple documents into an indexed collection. Asserts that all
 * documents appear in both a collection scan and an index scan. The indexed
 * value is a different BSON type, depending on the thread's id.
 */
load('jstests/concurrency/fsm_libs/extend_workload.js');           // for extendWorkload
load('jstests/concurrency/fsm_workloads/indexed_insert_base.js');  // for $config

var $config = extendWorkload($config, function($config, $super) {
    $config.data.indexedField = 'indexed_insert_heterogeneous';
    $config.data.shardKey = {};
    $config.data.shardKey[$config.data.indexedField] = 1;

    $config.states.init = function init(db, collName) {
        $super.states.init.apply(this, arguments);

        // prefix str with zeroes to make it have length len
        function pad(len, str) {
            var padding = new Array(len + 1).join('0');
            return (padding + str).slice(-len);
        }

        function makeOID(tid) {
            var str = pad(24, tid.toString(16));
            return new ObjectId(str);
        }

        function makeDate(tid) {
            var d = new ISODate('2000-01-01T00:00:00.000Z');
            // setSeconds(n) where n >= 60 will just cause the minutes,
            // hours, etc to increase,
            // so this produces a unique date for each tid
            d.setSeconds(tid);
            return d;
        }

        var choices = [
            this.tid,                                     // int
            this.tid.toString(),                          // string
            this.tid * 0.0001,                            // float
            {tid: this.tid},                              // subdocument
            makeOID(this.tid),                            // objectid
            makeDate(this.tid),                           // date
            new Function('', 'return ' + this.tid + ';')  // function
        ];

        this.indexedValue = choices[this.tid % choices.length];
    };

    return $config;
});