jstests/sharding/jumbo1.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
46

(function() {
'use strict';

load("jstests/sharding/libs/find_chunks_util.js");

var s = new ShardingTest({shards: 2, other: {chunkSize: 1}});

assert.commandWorked(s.s.adminCommand({enablesharding: "test", primaryShard: s.shard1.shardName}));
assert.commandWorked(
    s.s.adminCommand({addShardToZone: s.shard0.shardName, zone: 'finalDestination'}));

// Set the chunk range with a zone that will cause the chunk to be in the wrong place so the
// balancer will be forced to attempt to move it out.
assert.commandWorked(s.s.adminCommand({shardcollection: "test.foo", key: {x: 1}}));
assert.commandWorked(s.s.adminCommand(
    {updateZoneKeyRange: 'test.foo', min: {x: 0}, max: {x: MaxKey}, zone: 'finalDestination'}));

var db = s.getDB("test");

const big = 'X'.repeat(1024 * 1024);  // 1MB

// Insert 3MB of documents to create a jumbo chunk, and use the same shard key in all of
// them so that the chunk cannot be split.
var bulk = db.foo.initializeUnorderedBulkOp();
for (var i = 0; i < 3; i++) {
    bulk.insert({x: 0, big: big});
}

assert.commandWorked(bulk.execute());

s.startBalancer();

// Wait for the balancer to try to move the chunk and mark it as jumbo.
assert.soon(() => {
    let chunk = findChunksUtil.findOneChunkByNs(s.getDB('config'), 'test.foo', {min: {x: 0}});
    if (chunk == null) {
        // Balancer hasn't run and enforce the zone boundaries yet.
        return false;
    }

    assert.eq(s.shard1.shardName, chunk.shard, `${tojson(chunk)} was moved by the balancer`);
    return chunk.jumbo;
});

s.stop();
})();