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
|
'use strict';
/**
* count_indexed.js
*
* Runs count on an indexed field (using hint), which results in an index scan,
* and verifies the result.
* 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]
* Each thread inserts docs into a unique collection.
*/
load('jstests/concurrency/fsm_libs/extend_workload.js'); // for extendWorkload
load('jstests/concurrency/fsm_workloads/count.js'); // for $config
load('jstests/concurrency/fsm_workload_helpers/drop_utils.js'); // for dropCollections
var $config = extendWorkload($config, function($config, $super) {
$config.data.prefix = 'count_fsm';
$config.data.shardKey = {tid: 1, i: 1};
$config.data.getCount = function getCount(db, predicate) {
var query = Object.extend({tid: this.tid}, predicate);
return db[this.threadCollName].find(query).hint({tid: 1, i: 1}).count();
};
$config.states.init = function init(db, collName) {
this.threadCollName = this.prefix + '_' + this.tid;
$super.states.init.apply(this, arguments);
assertAlways.commandWorked(db[this.threadCollName].ensureIndex({tid: 1, i: 1}));
};
$config.teardown = function teardown(db, collName) {
var pattern = new RegExp('^' + this.prefix + '_\\d+$');
dropCollections(db, pattern);
$super.teardown.apply(this, arguments);
};
return $config;
});
|
