jstests/noPassthrough/update_yield1.js


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

// Ensure that multi-update operations yield regularly.
// Unless they are $isolated and they shouldn't yield at all.
(function() {
    'use strict';

    function countUpdateYields(coll) {
        var profileEntry = coll.getDB()
                               .system.profile.find({ns: coll.getFullName()})
                               .sort({$natural: -1})
                               .limit(1)
                               .next();
        printjson(profileEntry);
        assert.eq(profileEntry.op, 'update');
        return profileEntry.numYield;
    }

    var explain;
    var yieldCount;

    const nDocsToInsert = 300;
    const worksPerYield = 50;

    // Start a mongod that will yield every 50 work cycles.
    var mongod = MongoRunner.runMongod({
        setParameter: `internalQueryExecYieldIterations=${worksPerYield}`,
        profile: 2,
    });
    assert.neq(null, mongod, 'mongod was unable to start up');

    var coll = mongod.getDB('test').update_yield1;
    coll.drop();

    for (var i = 0; i < nDocsToInsert; i++) {
        assert.writeOK(coll.insert({_id: i}));
    }

    // A multi-update doing a collection scan should yield about nDocsToInsert / worksPerYield
    // times.
    assert.writeOK(coll.update({}, {$inc: {counter: 1}}, {multi: true}));
    assert.gt(countUpdateYields(coll), (nDocsToInsert / worksPerYield) - 2);

    // A multi-update shouldn't yield if it has $isolated.
    assert.writeOK(coll.update({$isolated: true}, {$inc: {counter: 1}}, {multi: true}));
    assert.eq(countUpdateYields(coll), 0, 'yielded during $isolated multi-update');
})();