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
|
// Tests various cases of dropping and recreating collections in the same namespace with multiple mongoses
var st = new ShardingTest({ shards : 3, mongos : 3, verbose : 1 });
// Balancer is by default stopped, thus it will not interfere
// Use separate mongoses for admin, inserting data, and validating results, so no
// single-mongos tricks will work
var insertMongos = st.s2;
var staleMongos = st.s1;
var config = st.s.getDB( "config" );
var admin = st.s.getDB( "admin" );
var coll = st.s.getCollection( "foo.bar" );
insertMongos.getDB( "admin" ).runCommand({ setParameter : 1, traceExceptions : true });
var shards = {};
config.shards.find().forEach( function( doc ){
shards[ doc._id ] = new Mongo( doc.host );
});
//
// Test that inserts and queries go to the correct shard even when the collection has been sharded
// in the background
//
jsTest.log( "Enabling sharding for the first time..." );
admin.runCommand({ enableSharding : coll.getDB() + "" });
st.ensurePrimaryShard(coll.getDB().getName(), 'shard0001');
admin.runCommand({ shardCollection : coll + "", key : { _id : 1 } });
var bulk = insertMongos.getCollection( coll + "" ).initializeUnorderedBulkOp();
for( var i = 0; i < 100; i++ ) {
bulk.insert({ _id : i, test : "a" });
}
assert.writeOK( bulk.execute() );
assert.eq( 100, staleMongos.getCollection( coll + "" ).find({ test : "a" }).itcount() );
coll.drop();
//
// Test that inserts and queries go to the correct shard even when the collection has been
// re-sharded in the background
//
jsTest.log( "Re-enabling sharding with a different key..." );
admin.runCommand({ enableSharding : coll.getDB() + "" });
st.ensurePrimaryShard(coll.getDB().getName(), 'shard0001');
coll.ensureIndex({ notId : 1 });
admin.runCommand({ shardCollection : coll + "", key : { notId : 1 } });
bulk = insertMongos.getCollection( coll + "" ).initializeUnorderedBulkOp();
for( var i = 0; i < 100; i++ ) {
bulk.insert({ notId : i, test : "b" });
}
assert.writeOK( bulk.execute() );
assert.eq( 100, staleMongos.getCollection( coll + "" ).find({ test : "b" }).itcount() );
assert.eq( 0, staleMongos.getCollection( coll + "" ).find({ test : { $in : [ "a" ] } }).itcount() );
coll.drop();
//
// Test that inserts and queries go to the correct shard even when the collection has been
// unsharded and moved to a different primary
//
jsTest.log( "Re-creating unsharded collection from a sharded collection on different primary..." );
var getOtherShard = function( shard ){
for( id in shards ){
if( id != shard ) return id;
}
};
var otherShard = getOtherShard(config.databases.findOne({_id: coll.getDB() + ""}).primary);
assert.commandWorked(admin.runCommand({movePrimary: coll.getDB() + "", to: otherShard}));
if (st.configRS) {
// If we are in CSRS mode need to make sure that staleMongos will actually get
// the most recent config data.
st.configRS.awaitLastOpCommitted();
}
jsTest.log( "moved primary..." );
bulk = insertMongos.getCollection( coll + "" ).initializeUnorderedBulkOp();
for( var i = 0; i < 100; i++ )
bulk.insert({ test : "c" });
assert.writeOK( bulk.execute() );
assert.eq( 100, staleMongos.getCollection( coll + "" ).find({ test : "c" }).itcount() );
assert.eq( 0, staleMongos.getCollection( coll + "" ).find({ test : { $in : [ "a", "b" ] } }).itcount() );
coll.drop();
//
// Test that inserts and queries go to correct shard even when the collection has been unsharded,
// resharded, and moved to a different primary
//
jsTest.log( "Re-creating sharded collection with different primary..." );
admin.runCommand({ enableSharding : coll.getDB() + "" });
admin.runCommand({ movePrimary : coll.getDB() + "",
to : getOtherShard( config.databases.findOne({ _id : coll.getDB() + "" }).primary ) });
admin.runCommand({ shardCollection : coll + "", key : { _id : 1 } });
bulk = insertMongos.getCollection( coll + "" ).initializeUnorderedBulkOp();
for( var i = 0; i < 100; i++ )
bulk.insert({ test : "d" });
assert.writeOK( bulk.execute() );
assert.eq( 100, staleMongos.getCollection( coll + "" ).find({ test : "d" }).itcount() );
assert.eq( 0, staleMongos.getCollection( coll + "" ).find({ test : { $in : [ "a", "b", "c" ] } }).itcount() );
coll.drop();
jsTest.log( "Done!" );
st.stop();
|
