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
|
// @tags: [does_not_support_stepdowns, requires_profiling]
//
// Tests that profiled $lookups contain the correct namespace and that Top is updated accordingly.
(function() {
"use strict";
load("jstests/libs/sbe_util.js"); // For checkSBEEnabled.
load("jstests/libs/analyze_plan.js"); // For getAggPlanStages.
const localColl = db.local;
const foreignColl = db.foreign;
localColl.drop();
foreignColl.drop();
assert.commandWorked(localColl.insert([{a: 1}, {b: 1}, {a: 2}]));
assert.commandWorked(foreignColl.insert({a: 1}));
db.system.profile.drop();
db.setProfilingLevel(2);
let oldTop = db.adminCommand("top");
const pipeline =
[{$lookup: {from: foreignColl.getName(), as: "res", localField: "a", foreignField: "a"}}];
localColl.aggregate(pipeline);
db.setProfilingLevel(0);
// Confirm that namespace is the local rather than foreign collection.
let profileDoc = db.system.profile.findOne();
assert.eq("test.local", profileDoc.ns);
// Confirm that the local collection had one command added to Top.
let newTop = db.adminCommand("top");
assert.eq(1,
newTop.totals[localColl.getFullName()].commands.count -
oldTop.totals[localColl.getFullName()].commands.count);
const actualCount = newTop.totals[foreignColl.getFullName()].commands.count -
oldTop.totals[foreignColl.getFullName()].commands.count;
// Compute the expected count as follows:
// 1) If the feature flag is enabled, add one to the count. This is because we will take a lock
// over 'foreignColl' and, even if we don't push down $lookup into SBE, this will still
// increment the top counter for 'foreignColl' by one.
// 2) If $lookup is NOT pushed down into SBE, then we increment the count by three. This is
// because when executing $lookup in the classic engine, we will add one entry to top for the
// foreign collection for each document in the local collection (of which there are three).
let expectedCount = 0;
const getFeatureFlagSBELookupPushdown =
db.adminCommand({getParameter: 1, featureFlagSBELookupPushdown: 1});
const isSBELookupPushdownEnabled =
getFeatureFlagSBELookupPushdown.hasOwnProperty("featureFlagSBELookupPushdown") &&
getFeatureFlagSBELookupPushdown["featureFlagSBELookupPushdown"]["value"];
if (isSBELookupPushdownEnabled) {
expectedCount++;
}
const eqLookupNodes = getAggPlanStages(localColl.explain().aggregate(pipeline), "EQ_LOOKUP");
if (eqLookupNodes.length === 0) {
expectedCount += 3;
}
assert.eq(expectedCount, actualCount);
}());
|
