SERVER-35518: Support the failCommand fail point in mongoS

1 files changed, 14 insertions, 1 deletions

diff --git a/src/mongo/db/commands.h b/src/mongo/db/commands.h
index 5b47d9db106..d2445729ee8 100644
--- a/src/mongo/db/commands.h
+++ b/src/mongo/db/commands.h
@@ -50,10 +50,13 @@
 #include "mongo/rpc/op_msg.h"
 #include "mongo/rpc/reply_builder_interface.h"
 #include "mongo/stdx/functional.h"
+#include "mongo/util/fail_point_service.h"
 #include "mongo/util/string_map.h"
 
 namespace mongo {
 
+MONGO_FAIL_POINT_DECLARE(failCommand);
+
 class Command;
 class CommandInvocation;
 class OperationContext;
@@ -245,6 +248,16 @@ struct CommandHelpers {
     static Status canUseTransactions(StringData dbName, StringData cmdName);
 
     static constexpr StringData kHelpFieldName = "help"_sd;
+
+    /**
+     * Checks if the command passed in is in the list of failCommands defined in the fail point.
+     */
+    static bool shouldActivateFailCommandFailPoint(const BSONObj& data, StringData cmdName);
+
+    /**
+     * Possibly uasserts according to the "failCommand" fail point.
+     */
+    static void evaluateFailCommandFailPoint(OperationContext* opCtx, StringData commandName);
 };
 
 /**
@@ -333,7 +346,7 @@ public:
 
     /**
      * Return true if the command requires auth.
-    */
+     */
     virtual bool requiresAuth() const {
         return true;
     }