jstests/sharding/bouncing_count.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
47
48
49
50
51
52
53
54
55

// Tests whether new sharding is detected on insert by mongos
(function() {

var st = new ShardingTest({ name: "test",
                            shards: 10,
                            mongos: 3 });

var mongosA = st.s0;
var mongosB = st.s1;
var mongosC = st.s2;

var admin = mongosA.getDB("admin");
var config = mongosA.getDB("config");

var collA = mongosA.getCollection( "foo.bar" );
var collB = mongosB.getCollection( "" + collA );
var collC = mongosB.getCollection( "" + collA );

admin.runCommand({ enableSharding : "" + collA.getDB() });
st.ensurePrimaryShard(collA.getDB().getName(), 'shard0001');
admin.runCommand({ shardCollection : "" + collA, key : { _id : 1 } });

var shards = config.shards.find().sort({ _id : 1 }).toArray();

jsTestLog( "Splitting up the collection..." );

// Split up the collection
for( var i = 0; i < shards.length; i++ ){
    printjson( admin.runCommand({ split : "" + collA, middle : { _id : i } }) );
    printjson( admin.runCommand({ moveChunk : "" + collA, find : { _id : i }, to : shards[i]._id }) );
}

mongosB.getDB("admin").runCommand({ flushRouterConfig : 1 });
mongosC.getDB("admin").runCommand({ flushRouterConfig : 1 });
printjson( collB.count() );
printjson( collC.count() );

// Change up all the versions...
for( var i = 0; i < shards.length; i++ ){
    printjson( admin.runCommand({ moveChunk : "" + collA, find : { _id : i }, to : shards[ (i + 1) % shards.length ]._id }) );
}

// Make sure mongos A is up-to-date
mongosA.getDB("admin").runCommand({ flushRouterConfig : 1 });

st.printShardingStatus(true);

jsTestLog( "Running count!" );

printjson( collB.count() );
printjson( collC.find().toArray() );

st.stop();

})();