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
123
124
|
// 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();
|