summaryrefslogtreecommitdiff
path: root/jstests/sharding/merge_chunks_test.js
blob: 22d0e8fc0fa238d74ec9f9e8f865c43f2d471aac (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
//
// Tests that merging chunks via mongos works/doesn't work with different chunk configurations
//

var options = { separateConfig : true, shardOptions : { verbose : 0 } };

var st = new ShardingTest({ shards : 2, mongos : 2, other : options });
st.stopBalancer();

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
coll.insert({ _id : 0 });
coll.insert({ _id : 40 });
coll.insert({ _id : 110 });
assert.eq( null, coll.getDB().getLastError() );

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

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

// Make sure merging three chunks is invalid (for now)

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

// 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 : 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.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() );

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

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

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

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

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

st.printShardingStatus();

st.stop();