blob: 8ab372d87f0963907e67017014227d2ebde53ff0 (
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
|
'use strict';
/**
* agg_sort.js
*
* Runs an aggregation with a $match that returns half the documents followed
* by a $sort on a field containing a random float.
*/
load('jstests/concurrency/fsm_libs/extend_workload.js'); // for extendWorkload
load('jstests/concurrency/fsm_workloads/agg_base.js'); // for $config
load('jstests/concurrency/fsm_workload_helpers/drop_utils.js'); // for dropCollections
var $config = extendWorkload($config, function($config, $super) {
$config.data.getOutputCollPrefix = function getOutputCollPrefix(collName) {
return collName + '_out_agg_sort_';
};
$config.states.query = function query(db, collName) {
var otherCollName = this.getOutputCollPrefix(collName) + this.tid;
var cursor = db[collName].aggregate(
[{$match: {flag: true}}, {$sort: {rand: 1}}, {$out: otherCollName}]);
assertAlways.eq(0, cursor.itcount());
assertWhenOwnColl.eq(db[collName].find().itcount() / 2, db[otherCollName].find().itcount());
};
$config.teardown = function teardown(db, collName, cluster) {
$super.teardown.apply(this, arguments);
// drop all collections with this workload's assumed-to-be-unique prefix
// NOTE: assumes the prefix contains no special regex chars
dropCollections(db, new RegExp('^' + this.getOutputCollPrefix(collName)));
};
return $config;
});
|