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
|
// When a $sort is followed by a $match, the pipeline optimizer should be able to move the match in
// front of the $sort, so as to reduce the number of documents that need to be sorted.
//
// @tags: [
// # Wrapping the pipeline in a $facet will prevent its $match from getting absorbed by the query
// # system, which is what we are testing for.
// do_not_wrap_aggregations_in_facets,
// # Likewise, splitting up the pipeline for sharding may also prevent the $match from getting
// # absorbed.
// assumes_unsharded_collection,
// # Don't disable the thing we are specifically testing for!
// requires_pipeline_optimization,
// ]
load('jstests/libs/analyze_plan.js');
(function() {
"use strict";
const coll = db.getSiblingDB("split_match_and_swap_with_sort")["test"];
coll.drop();
assert.commandWorked(
coll.insert([{_id: 1, a: 1, b: 3}, {_id: 2, a: 2, b: 2}, {_id: 3, a: 3, b: 1}]));
{
const pipeline = [{$sort: {b: 1}}, {$match: {a: {$ne: 2}}}];
assert.eq([{_id: 3, a: 3, b: 1}, {_id: 1, a: 1, b: 3}], coll.aggregate(pipeline).toArray());
const pipelineExplained = coll.explain().aggregate(pipeline);
const collScanStage = getPlanStage(pipelineExplained, "COLLSCAN");
// After moving the $match to the front of the pipeline, we expect the pipeline optimizer to
// push it down into the PlanExecutor, so that it becomes a filter on the COLLSCAN.
assert.neq(null, collScanStage, pipelineExplained);
assert.eq({a: {"$not": {"$eq": 2}}}, collScanStage.filter, collScanStage);
}
{
const pipeline =
[{$sort: {b: 1}}, {$match: {$and: [{a: {$ne: 2}}, {$expr: {$ne: ["$a", "$b"]}}]}}];
assert.eq([{_id: 3, a: 3, b: 1}, {_id: 1, a: 1, b: 3}], coll.aggregate(pipeline).toArray());
const pipelineExplained = coll.explain().aggregate(pipeline);
const collScanStage = getAggPlanStage(pipelineExplained, "COLLSCAN");
assert.neq(null, collScanStage, pipelineExplained);
assert.eq({$and: [{a: {"$not": {"$eq": 2}}}, {$expr: {$ne: ["$a", "$b"]}}]},
collScanStage.filter,
collScanStage);
}
{
// SERVER-46233: Normally a $or at the root of a $match expression prevents it from getting
// split. However, When the $or has only one child, the optimizer should be able to eliminate
// the $or and then recognize that the simplified result _is_ eligible to be split.
const pipeline =
[{$sort: {b: 1}}, {$match: {$or: [{$and: [{a: {$ne: 2}}, {$expr: {$ne: ["$a", "$b"]}}]}]}}];
assert.eq([{_id: 3, a: 3, b: 1}, {_id: 1, a: 1, b: 3}], coll.aggregate(pipeline).toArray());
const pipelineExplained = coll.explain().aggregate(pipeline);
const collScanStage = getAggPlanStage(pipelineExplained, "COLLSCAN");
assert.neq(null, collScanStage, pipelineExplained);
assert.eq({$and: [{a: {"$not": {"$eq": 2}}}, {$expr: {$ne: ["$a", "$b"]}}]},
collScanStage.filter,
collScanStage);
}
}());
|
