summaryrefslogtreecommitdiff
path: root/jstests/sharding/migrateBig.js
blob: 81961d9c2f538b1657f979e0940db36fb1cb3691 (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
(function() {

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

s.config.settings.update( { _id: "balancer" }, { $set : { _waitForDelete : true } } , true);
s.adminCommand( { enablesharding : "test" } );
s.ensurePrimaryShard('test', 'shard0001');
s.adminCommand( { shardcollection : "test.foo" , key : { x : 1 } } );

db = s.getDB( "test" );
coll = db.foo;

big = "";
while ( big.length < 10000 )
    big += "eliot";

var bulk = coll.initializeUnorderedBulkOp();
for ( x=0; x<100; x++ ) {
    bulk.insert( { x : x , big : big } );
}
assert.writeOK(bulk.execute());

db.printShardingStatus();

s.adminCommand( { split : "test.foo" , middle : { x : 30 } } );
s.adminCommand( { split : "test.foo" , middle : { x : 66 } } );
s.adminCommand( { movechunk : "test.foo" , find : { x : 90 } , to : s.getOther( s.getServer( "test" ) ).name } );

db.printShardingStatus();

print( "YO : "  + s.getServer( "test" ).host );
direct = new Mongo( s.getServer( "test" ).host );
print( "direct : " + direct );

directDB = direct.getDB( "test" );

for ( done=0; done<2*1024*1024; done+=big.length ){
    assert.writeOK(directDB.foo.insert( { x : 50 + Math.random() , big : big } ));
}

db.printShardingStatus();

assert.throws( function(){  s.adminCommand( { movechunk : "test.foo" , find : { x : 50 } , to : s.getOther( s.getServer( "test" ) ).name } ); } , [] , "move should fail" );

for ( i=0; i<20; i+= 2 ) {
    try {
        s.adminCommand( { split : "test.foo" , middle : { x : i } } );
    }
    catch ( e ) {
        // we may have auto split on some of these
        // which is ok
        print(e);
    }
}

db.printShardingStatus();

s.config.settings.update( { _id: "balancer" }, { $set : { stopped: false } } , true );

assert.soon( function(){ var x = s.chunkDiff( "foo" , "test" ); print( "chunk diff: " + x ); return x < 2; } , "no balance happened" , 8 * 60 * 1000 , 2000 ); 

assert.soon( function(){ return !s.isAnyBalanceInFlight(); } );

assert.eq( coll.count() , coll.find().itcount() );

s.stop();

})();