summaryrefslogtreecommitdiff
path: root/jstests/noPassthrough/start_up_with_custom_cost_model.js
blob: 4158eee4c94b18b9b2786c08245c0b2bd9ea7bec (plain)
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
/**
 * Tests that 'internalCostModelCoefficients' can be set on startup.
 */
(function() {
'use strict';

load("jstests/libs/optimizer_utils.js");  // For checkCascadesOptimizerEnabled.

function getScanCostWith(customScanCost) {
    const costStr = `{"scanIncrementalCost": ${customScanCost}}`;
    const conn = MongoRunner.runMongod({setParameter: {'internalCostModelCoefficients': costStr}});

    const db = conn.getDB(jsTestName());
    const coll = db.start_up_with_custom_cost_model;

    const qfc = db.adminCommand({getParameter: 1, internalQueryFrameworkControl: 1});

    if (!checkCascadesOptimizerEnabled(db) || qfc === "forceClassicEngine") {
        jsTestLog("Skipping test because the Cascades optimizer is not enabled");
        MongoRunner.stopMongod(conn);
        return;
    }

    assert.commandWorked(
        db.adminCommand({setParameter: 1, internalQueryFrameworkControl: "forceBonsai"}));

    const nDocuments = 100;
    assert.commandWorked(coll.insert(Array.from({length: nDocuments}, (_, i) => {
        return {a: 3, b: 3, c: i};
    })));

    // Get scan cost.
    const explain = coll.explain("executionStats").aggregate([]);
    assert.eq(nDocuments, explain.executionStats.nReturned);
    const internalCost =
        assert.commandWorked(db.adminCommand({getParameter: 1, internalCostModelCoefficients: 1}));

    assert(internalCost.hasOwnProperty("internalCostModelCoefficients"));
    assert.eq(internalCost['internalCostModelCoefficients'], costStr);

    const scanNode = navigateToPlanPath(explain, "child");
    assertValueOnPath("PhysicalScan", scanNode, "nodeType");

    const scanCost = scanNode.properties.cost;

    MongoRunner.stopMongod(conn);

    return scanCost;
}

const scanCost1 = getScanCostWith(0.2);
const scanCost2 = getScanCostWith(0.4);
if (scanCost1 === undefined) {
    return;
}

assert.lt(scanCost1, scanCost2);
}());