summaryrefslogtreecommitdiff
path: root/jstests/sharding/merge_chunks_test.js
blob: 6b00a6532eae6a4d89e19a8cfde3286f63ac5181 (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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
//
// Tests that merging chunks via mongos works/doesn't work with different chunk configurations
//
(function() {
    'use strict';

    var st = new ShardingTest({shards: 2, mongos: 2});

    var mongos = st.s0;
    var staleMongos = st.s1;
    var admin = mongos.getDB("admin");
    var shards = mongos.getCollection("config.shards").find().toArray();
    var coll = mongos.getCollection("foo.bar");

    assert(admin.runCommand({enableSharding: coll.getDB() + ""}).ok);
    printjson(admin.runCommand({movePrimary: coll.getDB() + "", to: shards[0]._id}));
    assert(admin.runCommand({shardCollection: coll + "", key: {_id: 1}}).ok);

    // Create ranges MIN->0,0->10,(hole),20->40,40->50,50->90,(hole),100->110,110->MAX on first
    // shard
    jsTest.log("Creating ranges...");

    assert(admin.runCommand({split: coll + "", middle: {_id: 0}}).ok);
    assert(admin.runCommand({split: coll + "", middle: {_id: 10}}).ok);
    assert(admin.runCommand({split: coll + "", middle: {_id: 20}}).ok);
    assert(admin.runCommand({split: coll + "", middle: {_id: 40}}).ok);
    assert(admin.runCommand({split: coll + "", middle: {_id: 50}}).ok);
    assert(admin.runCommand({split: coll + "", middle: {_id: 90}}).ok);
    assert(admin.runCommand({split: coll + "", middle: {_id: 100}}).ok);
    assert(admin.runCommand({split: coll + "", middle: {_id: 110}}).ok);

    assert(admin.runCommand({moveChunk: coll + "", find: {_id: 10}, to: shards[1]._id}).ok);
    assert(admin.runCommand({moveChunk: coll + "", find: {_id: 90}, to: shards[1]._id}).ok);

    st.printShardingStatus();

    // Insert some data into each of the consolidated ranges
    assert.writeOK(coll.insert({_id: 0}));
    assert.writeOK(coll.insert({_id: 40}));
    assert.writeOK(coll.insert({_id: 110}));

    var staleCollection = staleMongos.getCollection(coll + "");

    jsTest.log("Trying merges that should fail...");

    // S0: min->0, 0->10, 20->40, 40->50, 50->90, 100->110, 110->max
    // S1: 10->20, 90->100

    // Make sure merging non-exact chunks is invalid

    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: MinKey}, {_id: 5}]}).ok);
    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 5}, {_id: 10}]}).ok);
    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 15}, {_id: 50}]}).ok);
    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 20}, {_id: 55}]}).ok);
    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 105}, {_id: MaxKey}]}).ok);

    // Make sure merging single chunks is invalid

    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: MinKey}, {_id: 0}]}).ok);
    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 20}, {_id: 40}]}).ok);
    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 110}, {_id: MaxKey}]}).ok);

    // Make sure merging over holes is invalid

    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 0}, {_id: 40}]}).ok);
    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 40}, {_id: 110}]}).ok);
    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 50}, {_id: 110}]}).ok);

    // Make sure merging between shards is invalid

    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 0}, {_id: 20}]}).ok);
    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 10}, {_id: 40}]}).ok);
    assert(!admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 40}, {_id: 100}]}).ok);

    assert.eq(3, staleCollection.find().itcount());

    jsTest.log("Trying merges that should succeed...");

    assert(admin.runCommand({mergeChunks: coll + "", bounds: [{_id: MinKey}, {_id: 10}]}).ok);

    assert.eq(3, staleCollection.find().itcount());

    // S0: min->10, 20->40, 40->50, 50->90, 100->110, 110->max
    // S1: 10->20, 90->100

    // Make sure merging three chunks is valid.

    jsTest.log(tojson(admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 20}, {_id: 90}]})));

    // S0: min->10, 20->90, 100->110, 110->max
    // S1: 10->20, 90->100

    assert.eq(3, staleCollection.find().itcount());

    assert(admin.runCommand({mergeChunks: coll + "", bounds: [{_id: 100}, {_id: MaxKey}]}).ok);

    assert.eq(3, staleCollection.find().itcount());

    // S0: min->10, 20->90, 100->max
    // S1: 10->20, 90->100

    st.printShardingStatus();

    st.stop();

})();