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
|
// @tags: [requires_sharding]
(function() {
'use strict';
var s = new ShardingTest({shards: 2, mongos: 1, other: {chunkSize: 1, enableBalancer: true}});
assert.commandWorked(s.s0.adminCommand({enablesharding: "test"}));
s.ensurePrimaryShard('test', s.shard1.shardName);
var S = "";
while (S.length < 500) {
S += "123123312312";
}
var N = 100000;
var db = s.getDB("test");
var bulk = db.foo.initializeUnorderedBulkOp();
var bulk2 = db.bar.initializeUnorderedBulkOp();
for (i = 0; i < N; i++) {
bulk.insert({_id: i, s: S});
bulk2.insert({_id: i, s: S, s2: S});
}
assert.writeOK(bulk.execute());
assert.writeOK(bulk2.execute());
s.printShardingStatus();
function mytest(coll, i, loopNumber) {
try {
var x = coll.find({_id: i}).explain();
} catch (e) {
// Ignore stale shard version exceptions, since there are many migrations happening, and
// mongos may not be able to complete the find within the stale version retries limit.
if (e.message.contains("stale config")) {
return;
}
throw e;
}
if (x)
return;
throw Error("can't find " + i + " in " + coll.getName() + " on loopNumber: " + loopNumber +
" explain: " + tojson(x));
}
for (var loopNumber = 0;; loopNumber++) {
for (var i = 0; i < N; i++) {
mytest(db.foo, i, loopNumber);
mytest(db.bar, i, loopNumber);
if (i % 1000 == 0)
print(i);
}
s.printShardingStatus();
if (loopNumber == 0) {
assert.commandWorked(s.s0.adminCommand({shardcollection: "test.foo", key: {_id: 1}}));
assert.commandWorked(s.s0.adminCommand({shardcollection: "test.bar", key: {_id: 1}}));
}
assert(loopNumber < 1000, "taking too long");
if (s.chunkDiff("foo") < 12 && s.chunkDiff("bar") < 12) {
break;
}
}
s.stop();
})();
|
