summaryrefslogtreecommitdiff
path: root/jstests/sharding/merge_chunks_test.js
blob: 2a853bbd1d9da296f515e65b5cb3c2d5b4acbe21 (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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
//
// 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();

})();