summaryrefslogtreecommitdiff
path: root/jstests/sharding/balance_repl.js
blob: 02e2c54bbc0615093786e93a16796f86c2f8a2f6 (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
59
60
61
62
63
64
65
66
67
68
69
/**
 * Testing migrations are successful and immediately visible on the secondaries, when
 * secondaryThrottle is used.
 */
(function() {
    'use strict';

    // The mongod secondaries are set to priority 0 and votes 0 to prevent the primaries
    // from stepping down during migrations on slow evergreen builders.
    var s = new ShardingTest({
        shards: 2,
        other: {
            chunkSize: 1,
            rs0: {
                nodes: [
                    {rsConfig: {votes: 1}},
                    {rsConfig: {priority: 0, votes: 0}},
                ],
            },
            rs1: {
                nodes: [
                    {rsConfig: {votes: 1}},
                    {rsConfig: {priority: 0, votes: 0}},
                ],
            }
        }
    });

    var bulk = s.s0.getDB('TestDB').TestColl.initializeUnorderedBulkOp();
    for (var i = 0; i < 2100; i++) {
        bulk.insert({_id: i, x: i});
    }
    assert.writeOK(bulk.execute());

    assert.commandWorked(s.s0.adminCommand({enablesharding: 'TestDB'}));
    s.ensurePrimaryShard('TestDB', s.shard0.shardName);
    assert.commandWorked(s.s0.adminCommand({shardcollection: 'TestDB.TestColl', key: {_id: 1}}));

    for (i = 0; i < 20; i++) {
        assert.commandWorked(s.s0.adminCommand({split: 'TestDB.TestColl', middle: {_id: i * 100}}));
    }

    var collPrimary = (new Mongo(s.s0.host)).getDB('TestDB').TestColl;
    assert.eq(2100, collPrimary.find().itcount());

    var collSlaveOk = (new Mongo(s.s0.host)).getDB('TestDB').TestColl;
    collSlaveOk.setSlaveOk();
    assert.eq(2100, collSlaveOk.find().itcount());

    for (i = 0; i < 20; i++) {
        // Needs to waitForDelete because we'll be performing a slaveOk query, and secondaries don't
        // have a chunk manager so it doesn't know how to filter out docs it doesn't own.
        assert.commandWorked(s.s0.adminCommand({
            moveChunk: 'TestDB.TestColl',
            find: {_id: i * 100},
            to: s.shard1.shardName,
            _secondaryThrottle: true,
            writeConcern: {w: 2},
            _waitForDelete: true
        }));

        assert.eq(2100,
                  collSlaveOk.find().itcount(),
                  'Incorrect count when reading from secondary. Count from primary is ' +
                      collPrimary.find().itcount());
    }

    s.stop();
}());