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
|
/**
* Testing of just the query layer's integration for columnar indexes that encode large arrays.
* @tags: [
* # Columnstore tests set server parameters to disable columnstore query planning heuristics -
* # 1) server parameters are stored in-memory only so are not transferred onto the recipient,
* # 2) server parameters may not be set in stepdown passthroughs because it is a command that may
* # return different values after a failover
* tenant_migration_incompatible,
* does_not_support_stepdowns,
* not_allowed_with_security_token,
* requires_fcv_63,
* ]
*/
(function() {
"use strict";
load("jstests/libs/analyze_plan.js"); // For "planHasStage."
load("jstests/libs/columnstore_util.js"); // For "setUpServerForColumnStoreIndexTest."
if (!setUpServerForColumnStoreIndexTest(db)) {
return;
}
const coll = db.columnstore_large_array_index_correctness;
coll.drop();
const uint8 = {
num: 0,
a: Array.from({length: 50}, (_, i) => ({b: [2 * i, 2 * i + 1]}))
};
const uint16 = {
num: 1,
a: Array.from({length: 150}, (_, i) => ({b: [2 * i, 2 * i + 1]}))
};
const uint32 = {
num: 2,
a: Array.from({length: 15000}, (_, i) => ({b: [2 * i, 2 * i + 1]}))
};
const docs = [uint8, uint16, uint32];
for (let doc of docs) {
coll.insert(doc);
}
assert.commandWorked(coll.createIndex({"$**": "columnstore"}));
const kProjection = {
_id: 0,
"a.b": 1,
num: 1,
};
// Ensure this test is exercising the column scan.
let explain = coll.find({}, kProjection).sort({num: 1}).explain();
assert(planHasStage(db, explain, "COLUMN_SCAN"), explain);
// Run a query getting all of the results using the column index.
let results = coll.find({}, kProjection).sort({num: 1}).toArray();
assert.gt(results.length, 0);
// Run a query getting all results without column index
let trueResults = coll.find({}, kProjection).hint({$natural: 1}).sort({num: 1}).toArray();
assert.eq(results.length, trueResults.length);
for (let i = 0; i < results.length; i++) {
const originalDoc = coll.findOne({num: results[i].num});
assert.eq(results[i], trueResults[i], () =>
`column store index output number: ${results[i].num}, collection scan output number: ${trueResults[i].num},
original document number was: ${originalDoc.num}`);
}
})();
|
