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
|
// Regression test for SERVER-21673.
// A user can configure the shell to send commands via OP_QUERY or OP_MSG. This can be done at
// startup using the "--rpcProtocols" command line option, or at runtime using the
// "setClientRPCProtocols" method on the Mongo object.
// @tags: [requires_profiling]
var RPC_PROTOCOLS = {OP_QUERY: "opQueryOnly", OP_MSG: "opMsgOnly"};
(function() {
"use strict";
db.rpcProtocols.drop();
var oldProfilingLevel = db.getProfilingLevel();
assert.commandWorked(db.setProfilingLevel(2));
function runInShell(rpcProtocol, func) {
assert(0 ==
_runMongoProgram("mongo",
"--rpcProtocols=" + rpcProtocol,
"--readMode=commands", // ensure we use the find command.
"--eval",
"(" + func.toString() + ")();",
db.getMongo().host));
}
// Test that --rpcProtocols=opQueryOnly forces OP_QUERY commands.
runInShell(RPC_PROTOCOLS.OP_QUERY, function() {
assert(db.getMongo().getClientRPCProtocols() === "opQueryOnly");
db.getSiblingDB("test").rpcProtocols.find().comment("opQueryCommandLine").itcount();
});
var profileDoc = db.system.profile.findOne({"command.comment": "opQueryCommandLine"});
assert(profileDoc !== null);
assert.eq(profileDoc.protocol, "op_query");
// Test that --rpcProtocols=opMsgOnly forces OP_MSG commands.
runInShell(RPC_PROTOCOLS.OP_MSG, function() {
assert(db.getMongo().getClientRPCProtocols() === "opMsgOnly");
db.getSiblingDB("test").rpcProtocols.find().comment("opMsgCommandLine").itcount();
});
profileDoc = db.system.profile.findOne({"command.comment": "opMsgCommandLine"});
assert(profileDoc !== null);
assert.eq(profileDoc.protocol, "op_msg");
// Test that .setClientRPCProtocols("opQueryOnly") forces OP_QUERY commands. We start the shell
// in OP_MSG only mode, then switch it to OP_QUERY mode at runtime.
runInShell(RPC_PROTOCOLS.OP_MSG, function() {
assert(db.getMongo().getClientRPCProtocols() === "opMsgOnly");
db.getMongo().setClientRPCProtocols("opQueryOnly");
assert(db.getMongo().getClientRPCProtocols() === "opQueryOnly");
db.getSiblingDB("test").rpcProtocols.find().comment("opQueryRuntime").itcount();
});
profileDoc = db.system.profile.findOne({"command.comment": "opQueryRuntime"});
assert(profileDoc !== null);
assert.eq(profileDoc.protocol, "op_query");
// Test that .setClientRPCProtocols("opMsgOnly") forces OP_MSG commands. We start the
// shell in OP_QUERY only mode, then switch it to OP_MSG mode at runtime.
runInShell(RPC_PROTOCOLS.OP_QUERY, function() {
assert(db.getMongo().getClientRPCProtocols() === "opQueryOnly");
db.getMongo().setClientRPCProtocols("opMsgOnly");
assert(db.getMongo().getClientRPCProtocols() === "opMsgOnly");
db.getSiblingDB("test").rpcProtocols.find().comment("opMsgRuntime").itcount();
});
profileDoc = db.system.profile.findOne({"command.comment": "opMsgRuntime"});
assert(profileDoc !== null);
assert.eq(profileDoc.protocol, "op_msg");
// Reset profiling level.
assert.commandWorked(db.setProfilingLevel(oldProfilingLevel));
})();
|
