summaryrefslogtreecommitdiff
path: root/jstests/core/query/or/orp.js
blob: 738e2330f69429a157bf2c5b15586b270a2afbce (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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
// $or clause deduping with result set sizes > 101 (smaller result sets are now also deduped by the
// query optimizer cursor).

(function() {
'use strict';

const collNamePrefix = 'jstests_orp_';
let collCount = 0;
let docId = 0;

let t = db.getCollection(collNamePrefix + collCount++);
t.drop();

assert.commandWorked(t.createIndexes([{a: 1}, {b: 1}, {c: 1}]));

let docs = [];
for (let i = 0; i < 110; ++i) {
    docs.push({_id: docId++, a: 1, b: 1});
}
assert.commandWorked(t.insert(docs));

// Deduping results from the previous clause.
assert.eq(docs.length, t.countDocuments({$or: [{a: 1}, {b: 1}]}));

// Deduping results from a prior clause.
assert.eq(docs.length, t.countDocuments({$or: [{a: 1}, {c: 1}, {b: 1}]}));
assert.commandWorked(t.insert({_id: docId++, c: 1}));
assert.eq(docs.length + 1, t.countDocuments({$or: [{a: 1}, {c: 1}, {b: 1}]}));

// Deduping results that would normally be index only matches on overlapping and double scanned $or
// field regions.
t = db.getCollection(collNamePrefix + collCount++);
t.drop();
assert.commandWorked(t.createIndex({a: 1, b: 1}));
docs = [];
let k = 11;
for (let i = 0; i < k; ++i) {
    for (let j = 0; j < k; ++j) {
        docs.push({_id: docId++, a: i, b: j});
    }
}
assert.commandWorked(t.insert(docs));
assert.eq(k * k,
          t.countDocuments({$or: [{a: {$gte: 0}, b: {$gte: 0}}, {a: {$lte: k}, b: {$lte: k}}]}));

// Deduping results from a clause that completed before the multi cursor takeover.
t = db.getCollection(collNamePrefix + collCount++);
t.drop();
assert.commandWorked(t.createIndexes([{a: 1}, {b: 1}]));
docs = [];
k = 120;
docs.push({_id: docId++, a: 1, b: k});
for (let i = 0; i < k; ++i) {
    docs.push({_id: docId++, b: i});
}
assert.commandWorked(t.insert(docs));
assert.eq(k + 1, t.countDocuments({$or: [{a: 1}, {b: {$gte: 0}}]}));
})();