Adding connectionStatus command. (#145)

Author: apkar

src/ExtCmd.actor.cpp

@@ -1425,3 +1425,23 @@ struct GetDistinctCmd {
 	}
 };
 REGISTER_CMD(GetDistinctCmd, "distinct");
+
+struct ConnectionStatusCmd {
+	static const char* name;
+	static Future<Reference<ExtMsgReply>> call(Reference<ExtConnection> ec,
+	                                           Reference<ExtMsgQuery> query,
+	                                           Reference<ExtMsgReply> reply) {
+		const bool showPrivileges = query->query.getBoolField("showPrivileges");
+
+		bson::BSONObjBuilder authInfo;
+		authInfo.append("authenticatedUsers", std::vector<std::string>());
+		authInfo.append("authenticatedUserRoles", std::vector<std::string>());
+		if (showPrivileges) {
+			authInfo.append("authenticatedUserPrivileges", std::vector<std::string>());
+		}
+		reply->addDocument(BSON("authInfo" << authInfo.obj() << "ok" << 1));
+
+		return reply;
+	}
+};
+REGISTER_CMD(ConnectionStatusCmd, "connectionstatus");

test/correctness/smoke/test_basic_flow.py

@@ -60,3 +60,18 @@ def test_simple_coll_index(fixture_collection):
     returned = collection.find({'a': {'$gt': 10, '$lt': 21}}).count()
     assert returned == 10, "Expected: 10, Received: {}".format(returned)
 
+
+def test_connection_status(fixture_db):
+    db = fixture_db
+    status = db.command("connectionStatus", showPrivileges=True)
+    assert status['authInfo']['authenticatedUsers'] == list()
+    assert status['authInfo']['authenticatedUserRoles'] == list()
+    assert status['authInfo']['authenticatedUserPrivileges'] == list()
+    assert status['ok'] == 1
+
+    status = db.command("connectionStatus", showPrivileges=False)
+    assert status['authInfo']['authenticatedUsers'] == list()
+    assert status['authInfo']['authenticatedUserRoles'] == list()
+    assert 'authenticatedUserPrivileges' not in status['authInfo']
+    assert status['ok'] == 1
+