jstests/concurrency/fsm_workloads/indexed_insert_heterogeneous.js


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

'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;
                             });