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
|
var count = 0;
var w = 0;
var wait = function(f,msg) {
w++;
var n = 0;
while (!f()) {
if( n % 4 == 0 )
print("waiting " + w);
if (++n == 4) {
print("" + f);
}
assert(n < 200, 'tried 200 times, giving up on ' + msg );
sleep(1000);
}
};
/**
* Use this to do something once every 4 iterations.
*
* <pre>
* for (i=0; i<1000; i++) {
* occasionally(function() { print("4 more iterations"); });
* }
* </pre>
*/
var occasionally = function(f, n) {
var interval = n || 4;
if (count % interval == 0) {
f();
}
count++;
};
var reconnect = function(a) {
wait(function() {
try {
// make this work with either dbs or connections
if (typeof(a.getDB) == "function") {
db = a.getDB('foo');
}
else {
db = a;
}
db.bar.stats();
if (jsTest.options().keyFile || jsTest.options().useX509) { // SERVER-4241: Shell connections don't re-authenticate on reconnect
return jsTest.authenticate(db.getMongo());
}
return true;
} catch(e) {
print(e);
return false;
}
});
};
var getLatestOp = function(server) {
server.getDB("admin").getMongo().setSlaveOk();
var log = server.getDB("local")['oplog.rs'];
var cursor = log.find({}).sort({'$natural': -1}).limit(1);
if (cursor.hasNext()) {
return cursor.next();
}
return null;
};
var waitForAllMembers = function(master, timeout) {
var failCount = 0;
assert.soon( function() {
var state = null
try {
state = master.getSisterDB("admin").runCommand({replSetGetStatus:1});
failCount = 0;
} catch ( e ) {
// Connection can get reset on replica set failover causing a socket exception
print( "Calling replSetGetStatus failed" );
print( e );
return false;
}
occasionally(function() { printjson(state); }, 10);
for (var m in state.members) {
if (state.members[m].state != 1 && // PRIMARY
state.members[m].state != 2 && // SECONDARY
state.members[m].state != 7) { // ARBITER
return false;
}
}
printjson( state );
return true;
}, "not all members ready", timeout || 60000);
print( "All members are now in state PRIMARY, SECONDARY, or ARBITER" );
};
var reconfig = function(rs, config) {
var admin = rs.getMaster().getDB("admin");
try {
var ok = admin.runCommand({replSetReconfig : config});
assert.eq(ok.ok,1);
}
catch(e) {
print(e);
}
master = rs.getMaster().getDB("admin");
waitForAllMembers(master);
return master;
};
|