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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
/*
* Utilities for checking whether SBE is enabled.
*/
load("jstests/libs/discover_topology.js"); // For findNonConfigNodes.
load("jstests/libs/fixture_helpers.js"); // For 'isMongos'
/**
* Returns whether or not SBE is enabled for the given connection. Assumes that for repl sets and
* sharded clusters, SBE is either enabled on each node, or disabled on each node.
* If 'featureFlags' is non-empty, checks if SBE and all the feature flags are enabled.
* If 'checkAllNodes` is true, explicitly checks if feature flags are enabled for all
* nodes.
*/
function checkSBEEnabled(theDB, featureFlags = [], checkAllNodes = false) {
// By default, we find that SBE is enabled. If, for any node, we find that the classic engine is
// on, `checkResult` will be set to false. This is done intentionally so that in the event that
// we check all nodes, the effects from previously visited nodes will carry over into the rest.
// If we are not checking all nodes, `checkResult` is reset to true before each iteration.
let checkResult = true;
assert.soon(() => {
// Some test suites kill the primary, potentially resulting in networking errors. We use:
// 1. try..catch below to retry the whole check if we failed to discover topology
// 2. try..catch in the loop to try the next node if the current is killed (if we aren't
// checking to ensure that all feature flags are enabled on all nodes).
let nodes;
checkResult = true;
try {
nodes = DiscoverTopology.findNonConfigNodes(theDB.getMongo());
} catch (e) {
return false;
}
// Find a non-mongos node and check whether its forceClassicEngine flag is on. We assume
// either all nodes in the cluster have SBE disabled or none.
for (const node of nodes) {
try {
const conn = new Mongo(node);
if (FixtureHelpers.isMongos(conn.getDB("admin"))) {
continue;
}
const getParam = conn.adminCommand({
getParameter: 1,
internalQueryForceClassicEngine: 1,
internalQueryFrameworkControl: 1,
});
// v6.0 does not include the new internalQueryFrameworkControl server parameter.
// Here, we are accounting for both the old and new frameworks (where enabling a
// certain engine differs semantically).
if (getParam.hasOwnProperty("internalQueryForceClassicEngine") &&
getParam.internalQueryForceClassicEngine) {
checkResult = false;
}
if (getParam.hasOwnProperty("internalQueryFrameworkControl") &&
getParam.internalQueryFrameworkControl === "forceClassicEngine") {
checkResult = false;
}
if (!getParam.hasOwnProperty("internalQueryForceClassicEngine") &&
!getParam.hasOwnProperty("internalQueryFrameworkControl")) {
checkResult = false;
}
featureFlags.forEach(function(featureFlag) {
const featureFlagParam = conn.adminCommand({getParameter: 1, [featureFlag]: 1});
checkResult = checkResult && featureFlagParam.hasOwnProperty(featureFlag) &&
featureFlagParam[featureFlag]["value"];
});
// Exit `assert.soon` if we are only analyzing one node in the cluster.
if (!checkAllNodes) {
return true;
}
} catch (e) {
// Unable to verify that all feature flags are enabled on the current node. Return
// early from `assert.soon` if we're checking all nodes; otherwise, try the next
// node.
if (checkAllNodes) {
return false;
} else {
checkResult = true;
continue;
}
}
}
// If we are not checking feature flags on all nodes, this result only occurs when we catch
// an exception for each node. In this case, the output of `assert.soon` should be false.
// If we are checking feature flags on all nodes, this result only occurs when we
// successfully iterate over each node and update the local boolean flags. In this case, the
// output of `assert.soon` should be true.
return checkAllNodes;
});
return checkResult;
}
|