jstests/hooks/drop_sharded_collections.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

/**
 * Drops all sharded collections (except for collections used internally,
 * like config.system.sessions).
 */
(function() {
    "use strict";

    load("jstests/libs/fixture_helpers.js");  // For isMongos.

    assert.neq(typeof db, 'undefined', 'No `db` object, is the shell connected to a mongod?');
    assert(FixtureHelpers.isMongos(db), "not connected to mongos");

    let balSettingResult = assert.commandWorked(db.adminCommand({balancerStatus: 1}));
    if (balSettingResult.mode !== 'off') {
        assert.commandWorked(db.adminCommand({balancerStop: 1}));
    }

    db.getSiblingDB('config').collections.find().forEach(collEntry => {
        if (collEntry._id !== 'config.system.sessions') {
            let nsSplit = collEntry._id.split('.');
            const dbName = nsSplit.shift();
            const collName = nsSplit.join('.');

            // Note: drop also cleans up tags and chunks associated with ns.
            assert.commandWorked(db.getSiblingDB(dbName).runCommand({drop: collName}));
        }
    });

    // Turn balancer back on if it was not off earlier.
    if (balSettingResult.mode !== 'off') {
        assert.commandWorked(db.adminCommand({balancerStart: 1}));
    }
})();