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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
// Check that doing updates done during a migrate all go to the right place
(function() {
var s = new ShardingTest({shards: 2, mongos: 1, other: {chunkSize: 1, enableAutoSplit: true}});
assert.commandWorked(s.s0.adminCommand({enablesharding: "test"}));
s.ensurePrimaryShard('test', s.shard1.shardName);
assert.commandWorked(s.s0.adminCommand({shardcollection: "test.foo", key: {_id: 1}}));
s.config.settings.find().forEach(printjson);
db = s.getDB("test");
bigString = "";
while (bigString.length < 10000)
bigString += "asdasdasdasdadasdasdasdasdasdasdasdasda";
N = 3000;
num = 0;
var counts = {};
//
// TODO: Rewrite to make much clearer.
//
// The core behavior of this test is to add a bunch of documents to a sharded collection, then
// incrementally update each document and make sure the counts in the document match our update
// counts while balancing occurs (doUpdate()). Every once in a while we also check (check())
// our counts via a query.
//
// If during a chunk migration an update is missed, we trigger an assertion and fail.
//
function doUpdate(bulk, includeString, optionalId) {
var up = {$inc: {x: 1}};
if (includeString) {
up["$set"] = {s: bigString};
}
var myid = optionalId == undefined ? Random.randInt(N) : optionalId;
bulk.find({_id: myid}).upsert().update(up);
counts[myid] = (counts[myid] ? counts[myid] : 0) + 1;
return myid;
}
Random.setRandomSeed();
// Initially update all documents from 1 to N, otherwise later checks can fail because no
// document previously existed
var bulk = db.foo.initializeUnorderedBulkOp();
for (i = 0; i < N; i++) {
doUpdate(bulk, true, i);
}
for (i = 0; i < N * 9; i++) {
doUpdate(bulk, false);
}
assert.writeOK(bulk.execute());
for (var i = 0; i < 50; i++) {
s.printChunks("test.foo");
if (check("initial:" + i, true))
break;
sleep(5000);
}
check("initial at end");
assert.lt(20, s.config.chunks.count(), "setup2");
function check(msg, dontAssert) {
for (var x in counts) {
var e = counts[x];
var z = db.foo.findOne({_id: parseInt(x)});
if (z && z.x == e)
continue;
if (dontAssert) {
if (z)
delete z.s;
print("not asserting for key failure: " + x + " want: " + e + " got: " + tojson(z));
return false;
}
s.s.getDB("admin").runCommand({setParameter: 1, logLevel: 2});
printjson(db.foo.findOne({_id: parseInt(x)}));
var y = db.foo.findOne({_id: parseInt(x)});
if (y) {
delete y.s;
}
s.printChunks("test.foo");
assert(z, "couldn't find : " + x + " y:" + tojson(y) + " e: " + e + " " + msg);
assert.eq(e, z.x, "count for : " + x + " y:" + tojson(y) + " " + msg);
}
return true;
}
var consecutiveNoProgressMadeErrors = 0;
function diff1() {
jsTest.log("Running diff1...");
var bulk = db.foo.initializeUnorderedBulkOp();
var myid = doUpdate(bulk, false);
var res = bulk.execute();
assert(res instanceof BulkWriteResult,
'Result from bulk.execute should be of type BulkWriteResult');
if (res.hasWriteErrors()) {
res.writeErrors.forEach(function(err) {
// Ignore up to 3 consecutive NoProgressMade errors for the cases where migration
// might be going faster than the writes are executing
if (err.code == ErrorCodes.NoProgressMade) {
consecutiveNoProgressMadeErrors++;
if (consecutiveNoProgressMadeErrors < 3) {
return;
}
}
assert.writeOK(res);
});
} else {
consecutiveNoProgressMadeErrors = 0;
assert.eq(1,
res.nModified,
"diff myid: " + myid + " 2: " + res.toString() + "\n" +
" correct count is: " + counts[myid] + " db says count is: " +
tojson(db.foo.findOne({_id: myid})));
}
var x = s.chunkCounts("foo");
if (Math.random() > .999)
printjson(x);
return Math.max(x.shard0000, x.shard0001) - Math.min(x.shard0000, x.shard0001);
}
assert.lt(20, diff1(), "initial load");
print(diff1());
s.startBalancer();
assert.soon(function() {
var d = diff1();
return d < 5;
}, "balance didn't happen", 1000 * 60 * 20, 1);
s.stop();
})();
|
