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
108
109
110
111
112
113
114
|
// This test makes assertions about how many keys are examined during query execution, which can
// change depending on whether/how many documents are filtered out by the SHARDING_FILTER stage.
// @tags: [
// assumes_unsharded_collection,
// ]
(function() {
'use strict';
const collNamePrefix = 'jstests_index_check6_';
let collCount = 0;
let t = db.getCollection(collNamePrefix + collCount++);
t.drop();
function keysExamined(query, hint) {
let explain = t.find(query).hint(hint).explain("executionStats");
return explain.executionStats.totalKeysExamined;
}
assert.commandWorked(t.createIndex({age: 1, rating: 1}));
for (let age = 10; age < 50; age++) {
for (let rating = 0; rating < 10; rating++) {
assert.commandWorked(t.insert({age: age, rating: rating}));
}
}
assert.eq(10, keysExamined({age: 30}, {}), "A");
assert.eq(20, keysExamined({age: {$gte: 29, $lte: 30}}, {}), "B");
assert.eq(19,
keysExamined({age: {$gte: 25, $lte: 30}, rating: {$in: [0, 9]}}, {age: 1, rating: 1}),
"C1");
assert.eq(24,
keysExamined({age: {$gte: 25, $lte: 30}, rating: {$in: [0, 8]}}, {age: 1, rating: 1}),
"C2");
assert.eq(29,
keysExamined({age: {$gte: 25, $lte: 30}, rating: {$in: [1, 8]}}, {age: 1, rating: 1}),
"C3");
assert.eq(5,
keysExamined({age: {$gte: 29, $lte: 30}, rating: 5}, {age: 1, rating: 1}),
"C"); // SERVER-371
assert.eq(
7,
keysExamined({age: {$gte: 29, $lte: 30}, rating: {$gte: 4, $lte: 5}}, {age: 1, rating: 1}),
"D"); // SERVER-371
assert.eq(2,
t.find({age: 30, rating: {$gte: 4, $lte: 5}})
.explain('executionStats')
.executionStats.totalKeysExamined);
function doPopulateData() {
for (let a = 1; a < 10; a++) {
for (let b = 0; b < 10; b++) {
for (let c = 0; c < 10; c++) {
assert.commandWorked(t.insert({a: a, b: b, c: c}));
}
}
}
}
function doQuery(count, query, sort, index) {
let explain = t.find(query).hint(index).sort(sort).explain("executionStats");
let nscanned = explain.executionStats.totalKeysExamined;
assert(Math.abs(count - nscanned) <= 2);
}
function doTest(sort, index) {
doQuery(1, {a: 5, b: 5, c: 5}, sort, index);
doQuery(2, {a: 5, b: 5, c: {$gte: 5, $lte: 6}}, sort, index);
doQuery(1, {a: 5, b: 5, c: {$gte: 5.5, $lte: 6}}, sort, index);
doQuery(1, {a: 5, b: 5, c: {$gte: 5, $lte: 5.5}}, sort, index);
doQuery(3, {a: 5, b: 5, c: {$gte: 5, $lte: 7}}, sort, index);
doQuery(4, {a: 5, b: {$gte: 5, $lte: 6}, c: 5}, sort, index);
if (sort.b > 0) {
doQuery(3, {a: 5, b: {$gte: 5.5, $lte: 6}, c: 5}, sort, index);
doQuery(3, {a: 5, b: {$gte: 5, $lte: 5.5}, c: 5}, sort, index);
} else {
doQuery(3, {a: 5, b: {$gte: 5.5, $lte: 6}, c: 5}, sort, index);
doQuery(3, {a: 5, b: {$gte: 5, $lte: 5.5}, c: 5}, sort, index);
}
doQuery(8, {a: 5, b: {$gte: 5, $lte: 7}, c: 5}, sort, index);
doQuery(5, {a: {$gte: 5, $lte: 6}, b: 5, c: 5}, sort, index);
if (sort.a > 0) {
doQuery(3, {a: {$gte: 5.5, $lte: 6}, b: 5, c: 5}, sort, index);
doQuery(3, {a: {$gte: 5, $lte: 5.5}, b: 5, c: 5}, sort, index);
doQuery(3, {a: {$gte: 5.5, $lte: 6}, b: 5, c: {$gte: 5, $lte: 6}}, sort, index);
} else {
doQuery(3, {a: {$gte: 5.5, $lte: 6}, b: 5, c: 5}, sort, index);
doQuery(3, {a: {$gte: 5, $lte: 5.5}, b: 5, c: 5}, sort, index);
doQuery(4, {a: {$gte: 5.5, $lte: 6}, b: 5, c: {$gte: 5, $lte: 6}}, sort, index);
}
doQuery(8, {a: {$gte: 5, $lte: 7}, b: 5, c: 5}, sort, index);
doQuery(7, {a: {$gte: 5, $lte: 6}, b: 5, c: {$gte: 5, $lte: 6}}, sort, index);
doQuery(7, {a: 5, b: {$gte: 5, $lte: 6}, c: {$gte: 5, $lte: 6}}, sort, index);
doQuery(11, {a: {$gte: 5, $lte: 6}, b: {$gte: 5, $lte: 6}, c: 5}, sort, index);
doQuery(15, {a: {$gte: 5, $lte: 6}, b: {$gte: 5, $lte: 6}, c: {$gte: 5, $lte: 6}}, sort, index);
}
for (let a = -1; a <= 1; a += 2) {
for (let b = -1; b <= 1; b += 2) {
for (let c = -1; c <= 1; c += 2) {
// Retain collection from previous iteration for validation purposes
// and to be consistent with other jsCore tests.
t = db.getCollection(collNamePrefix + collCount++);
t.drop();
doPopulateData();
let spec = {a: a, b: b, c: c};
t.createIndex(spec);
doTest(spec, spec);
doTest({a: -a, b: -b, c: -c}, spec);
}
}
}
})();
|