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
|
// Tests that pipeline optimization works properly when the failpoint isn't triggered, and is
// disabled properly when it is triggered.
(function() {
"use strict";
load("jstests/libs/analyze_plan.js"); // For aggPlan functions.
Random.setRandomSeed();
const conn = MongoRunner.runMongod({});
assert.neq(conn, null, "Mongod failed to start up.");
const testDb = conn.getDB("test");
const coll = testDb.agg_opt;
const pops = new Set();
for (let i = 0; i < 25; ++i) {
let pop;
do {
pop = Random.randInt(100000);
} while (pops.has(pop));
pops.add(pop);
assert.commandWorked(coll.insert({_id: i, city: "Cleveland", pop: pop, state: "OH"}));
}
const pipeline = [
{$match: {state: "OH"}},
// The test-only '$_internalInhibitOptimization' operator prevents the $sort and $limit from
// being pushed down into the PlanStage layer, thereby ensuring that these two stages remain
// inside the pipeline layer. We need to make sure that the pipeline does get optimized away,
// since the "disablePipelineOptimization" failpoint does nothing in the
// "optimizedPipeline:true" case.
{$_internalInhibitOptimization: {}},
{$sort: {pop: -1}},
{$limit: 10}
];
const enabledPlan = coll.explain().aggregate(pipeline);
// Test that sort and the limit were combined.
assert.eq(aggPlanHasStage(enabledPlan, "$limit"), false);
assert.eq(aggPlanHasStage(enabledPlan, "$cursor"), true);
assert.eq(aggPlanHasStage(enabledPlan, "$_internalInhibitOptimization"), true);
assert.eq(aggPlanHasStage(enabledPlan, "$sort"), true);
assert.eq(enabledPlan.stages.length, 3);
const enabledResult = coll.aggregate(pipeline).toArray();
// Enable a failpoint that will cause pipeline optimizations to be skipped.
assert.commandWorked(
testDb.adminCommand({configureFailPoint: "disablePipelineOptimization", mode: "alwaysOn"}));
const disabledPlan = coll.explain().aggregate(pipeline);
// Test that the $limit still exists and hasn't been optimized away.
assert.eq(aggPlanHasStage(enabledPlan, "$cursor"), true);
assert.eq(aggPlanHasStage(enabledPlan, "$_internalInhibitOptimization"), true);
assert.eq(aggPlanHasStage(disabledPlan, "$sort"), true);
assert.eq(aggPlanHasStage(disabledPlan, "$limit"), true);
assert.eq(disabledPlan.stages.length, 4);
const disabledResult = coll.aggregate(pipeline).toArray();
// Test that the result is the same with and without optimizations enabled.
assert.eq(enabledResult, disabledResult);
MongoRunner.stopMongod(conn);
}());
|
