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
|
// Test the killCursors command.
(function() {
'use strict';
var cmdRes;
var cursor;
var cursorId;
var coll = db.jstest_killcursors;
coll.drop();
for (var i = 0; i < 10; i++) {
assert.writeOK(coll.insert({_id: i}));
}
// killCursors command should fail if the collection name is not a string.
cmdRes = db.runCommand({
killCursors: {foo: "bad collection param"},
cursors: [NumberLong(123), NumberLong(456)]
});
assert.commandFailedWithCode(cmdRes, ErrorCodes.FailedToParse);
// killCursors command should fail if the cursors parameter is not an array.
cmdRes = db.runCommand({
killCursors: coll.getName(),
cursors: {a: NumberLong(123), b: NumberLong(456)}
});
assert.commandFailedWithCode(cmdRes, ErrorCodes.FailedToParse);
// killCursors command should fail if the cursors parameter is an empty array.
cmdRes = db.runCommand({
killCursors: coll.getName(),
cursors: []
});
assert.commandFailedWithCode(cmdRes, ErrorCodes.BadValue);
// killCursors command should report cursors as not found if the collection does not exist.
cmdRes = db.runCommand({
killCursors: "non-existent-collection",
cursors: [NumberLong(123), NumberLong(456)]
});
assert.commandWorked(cmdRes);
assert.eq(cmdRes.cursorsKilled, []);
assert.eq(cmdRes.cursorsNotFound, [NumberLong(123), NumberLong(456)]);
assert.eq(cmdRes.cursorsAlive, []);
assert.eq(cmdRes.cursorsUnknown, []);
// killCursors command should report non-existent cursors as "not found".
cmdRes = db.runCommand({
killCursors: coll.getName(),
cursors: [NumberLong(123), NumberLong(456)]
});
assert.commandWorked(cmdRes);
assert.eq(cmdRes.cursorsKilled, []);
assert.eq(cmdRes.cursorsNotFound, [NumberLong(123), NumberLong(456)]);
assert.eq(cmdRes.cursorsAlive, []);
assert.eq(cmdRes.cursorsUnknown, []);
// Test a case where one cursors exists and is killed but the other does not exist.
cmdRes = db.runCommand({find: coll.getName(), batchSize: 2});
assert.commandWorked(cmdRes);
cursorId = cmdRes.cursor.id;
assert.neq(cursorId, NumberLong(0));
cmdRes = db.runCommand({
killCursors: coll.getName(),
cursors: [NumberLong(123), cursorId]
});
assert.commandWorked(cmdRes);
assert.eq(cmdRes.cursorsKilled, [cursorId]);
assert.eq(cmdRes.cursorsNotFound, [NumberLong(123)]);
assert.eq(cmdRes.cursorsAlive, []);
assert.eq(cmdRes.cursorsUnknown, []);
// Test killing a noTimeout cursor.
cmdRes = db.runCommand({find: coll.getName(), batchSize: 2, noCursorTimeout: true});
assert.commandWorked(cmdRes);
cursorId = cmdRes.cursor.id;
assert.neq(cursorId, NumberLong(0));
cmdRes = db.runCommand({
killCursors: coll.getName(),
cursors: [NumberLong(123), cursorId]
});
assert.commandWorked(cmdRes);
assert.eq(cmdRes.cursorsKilled, [cursorId]);
assert.eq(cmdRes.cursorsNotFound, [NumberLong(123)]);
assert.eq(cmdRes.cursorsAlive, []);
assert.eq(cmdRes.cursorsUnknown, []);
// Test killing a pinned cursor. Since cursors are generally pinned for short periods of time
// while result batches are generated, this requires some special machinery to keep a cursor
// permanently pinned.
var failpointName = "keepCursorPinnedDuringGetMore";
var cleanup;
try {
// Enable a failpoint to ensure that the cursor remains pinned.
assert.commandWorked(db.adminCommand({
configureFailPoint: failpointName,
mode: "alwaysOn"
}));
cmdRes = db.runCommand({find: coll.getName(), batchSize: 2});
assert.commandWorked(cmdRes);
cursorId = cmdRes.cursor.id;
assert.neq(cursorId, NumberLong(0));
cmdRes = db.runCommand({isMaster: 1});
assert.commandWorked(cmdRes);
var isMongos = (cmdRes.msg === "isdbgrid");
// Pin the cursor during a getMore.
var code = 'db.runCommand({getMore: ' + cursorId.toString() +
', collection: "' + coll.getName() + '"});';
cleanup = startParallelShell(code);
// Sleep to make it more likely that the cursor will be pinned.
sleep(2000);
// Attempt to kill the cursor. In order to avoid flakiness, we do not assume that the cursor
// is already pinned (although generally it will be).
//
// Currently, pinned cursors that are targeted by a killCursors operation are kept alive on
// mongod but are killed on mongos (see SERVER-21710).
cmdRes = db.runCommand({
killCursors: coll.getName(),
cursors: [NumberLong(123), cursorId]
});
assert.commandWorked(cmdRes);
assert.eq(cmdRes.cursorsNotFound, [NumberLong(123)]);
assert.eq(cmdRes.cursorsUnknown, []);
if (isMongos) {
assert.eq(cmdRes.cursorsKilled, [cursorId]);
assert.eq(cmdRes.cursorsAlive, []);
}
else {
// If the cursor has already been pinned it will be left alive; otherwise it will be
// killed.
if (cmdRes.cursorsAlive.length === 1) {
assert.eq(cmdRes.cursorsKilled, []);
assert.eq(cmdRes.cursorsAlive, [cursorId]);
}
else {
assert.eq(cmdRes.cursorsKilled, [cursorId]);
assert.eq(cmdRes.cursorsAlive, []);
}
}
} finally {
assert.commandWorked(db.adminCommand({configureFailPoint: failpointName, mode: "off"}));
if (cleanup) {
cleanup();
}
}
})();
|
