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
|
(function() {
"use strict";
load("jstests/libs/optimizer_utils.js"); // For checkCascadesOptimizerEnabled.
if (!checkCascadesOptimizerEnabled(db)) {
jsTestLog("Skipping test because the optimizer is not enabled");
return;
}
const coll = db.cqf_chess;
Random.srand(0);
const players = [
"penguingim1", "aladdin65", "aleksey472", "azuaga", "benpig",
"blackboarder", "bockosrb555", "bogdan_low_player", "charlytb", "chchbbuur",
"chessexplained", "cmcookiemonster", "crptone", "cselhu3", "darkzam",
"dmitri31", "dorado99", "ericrosen", "fast-tsunami", "flaneur"
];
const sources = [1, 2, 3, 4, 5, 6, 7, 8];
const variants = [1, 2, 3, 4, 5, 6, 7, 8];
const results = [1, 2, 3, 4, 5, 6, 7, 8];
const winColor = [true, false, null];
const nbGames = 1000; // 1000 * 1000;
function intRandom(max) {
return Random.randInt(max);
}
function anyOf(as) {
return as [intRandom(as.length)];
}
coll.drop();
print(`Adding ${nbGames} games`);
const bulk = coll.initializeUnorderedBulkOp();
for (let i = 0; i < nbGames; i++) {
const users = [anyOf(players), anyOf(players)];
const winnerIndex = intRandom(2);
bulk.insert({
users: users,
winner: users[winnerIndex],
loser: users[1 - winnerIndex],
winColor: anyOf(winColor),
avgRating: NumberInt(600 + intRandom(2400)),
source: NumberInt(anyOf(sources)),
variants: NumberInt(anyOf(variants)),
mode: !!intRandom(2),
turns: NumberInt(1 + intRandom(300)),
minutes: NumberInt(30 + intRandom(3600 * 3)),
clock: {init: NumberInt(0 + intRandom(10800)), inc: NumberInt(0 + intRandom(180))},
result: anyOf(results),
date: new Date(Date.now() - intRandom(118719488)),
analysed: !!intRandom(2)
});
if (i % 1000 == 0) {
print(`${i} / ${nbGames}`);
}
}
assert.commandWorked(bulk.execute());
const indexes = [
{users: 1},
{winner: 1},
{loser: 1},
{winColor: 1},
{avgRating: 1},
{source: 1},
{variants: 1},
{mode: 1},
{turns: 1},
{minutes: 1},
{'clock.init': 1},
{'clock.inc': 1},
{result: 1},
{date: 1},
{analysed: 1}
];
print("Adding indexes");
indexes.forEach(index => {
printjson(index);
coll.createIndex(index);
});
print("Searching");
const res = coll.explain("executionStats").aggregate([
{
$match: {
avgRating: {$gt: 1000},
turns: {$lt: 250},
'clock.init': {$gt: 1},
minutes: {$gt: 2, $lt: 150}
}
},
{$sort: {date: -1}},
{$limit: 20}
]);
// TODO: verify expected results.
// Verify we are using the index on "minutes".
const indexNode = res.queryPlanner.winningPlan.optimizerPlan.child.child.leftChild;
assert.eq("IndexScan", indexNode.nodeType);
assert.eq("minutes_1", indexNode.indexDefName);
}());
|
