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
|
// Tests that profiled $lookups contain the correct namespace and that Top is updated accordingly.
// @tags: [
// does_not_support_stepdowns,
// requires_profiling,
// ]
(function() {
"use strict";
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) We expect the count to be at least one. 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 = 1;
const eqLookupNodes = getAggPlanStages(localColl.explain().aggregate(pipeline), "EQ_LOOKUP");
if (eqLookupNodes.length === 0) {
expectedCount += 3;
}
assert.eq(expectedCount, actualCount);
}());
|
