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
|
/**
* Utilities for turning on/off and waiting for fail points.
*/
var configureFailPoint;
var configureFailPointForRS;
var kDefaultWaitForFailPointTimeout;
(function() {
"use strict";
if (configureFailPoint) {
return; // Protect against this file being double-loaded.
}
kDefaultWaitForFailPointTimeout = 5 * 60 * 1000;
configureFailPoint = function(conn, failPointName, data = {}, failPointMode = "alwaysOn") {
const res = sh.assertRetryableCommandWorkedOrFailedWithCodes(() => {
return conn.adminCommand(
{configureFailPoint: failPointName, mode: failPointMode, data: data});
}, "Timed out enabling fail point " + failPointName);
return {
conn: conn,
failPointName: failPointName,
timesEntered: res.count,
wait: function({maxTimeMS = kDefaultWaitForFailPointTimeout, timesEntered = 1} = {}) {
// Can only be called once because this function does not keep track of the
// number of times the fail point is entered between the time it returns
// and the next time it gets called.
sh.assertRetryableCommandWorkedOrFailedWithCodes(() => {
return conn.adminCommand({
waitForFailPoint: failPointName,
timesEntered: this.timesEntered + timesEntered,
maxTimeMS: maxTimeMS
});
}, "Timed out waiting for failpoint " + failPointName);
},
waitWithTimeout: function(timeoutMS) {
// This function has three possible outcomes:
//
// 1) Returns true when the failpoint was hit.
// 2) Returns false when the command returned a `MaxTimeMSExpired` response.
// 3) Otherwise, this throws for an unexpected error.
let res = sh.assertRetryableCommandWorkedOrFailedWithCodes(() => {
return conn.adminCommand({
waitForFailPoint: failPointName,
timesEntered: this.timesEntered + 1,
maxTimeMS: timeoutMS
});
}, "Timed out waiting for failpoint " + failPointName, [ErrorCodes.MaxTimeMSExpired]);
return res !== undefined && res["ok"] === 1;
},
off: function() {
sh.assertRetryableCommandWorkedOrFailedWithCodes(() => {
return conn.adminCommand({configureFailPoint: failPointName, mode: "off"});
}, "Timed out disabling fail point " + failPointName);
}
};
};
configureFailPointForRS = function(conns, failPointName, data = {}, failPointMode = "alwaysOn") {
conns.forEach((conn) => {
sh.assertRetryableCommandWorkedOrFailedWithCodes(() => {
return conn.adminCommand(
{configureFailPoint: failPointName, mode: failPointMode, data: data});
}, "Timed out setting failpoint " + failPointName);
});
return {
conns: conns,
failPointName: failPointName,
off: function() {
conns.forEach((conn) => {
sh.assertRetryableCommandWorkedOrFailedWithCodes(() => {
return conn.adminCommand({configureFailPoint: failPointName, mode: "off"});
}, "Timed out disabling fail point " + failPointName);
});
}
};
};
})();
|
