SERVER-20134 Restore the authSchemaUpgrade command

1 files changed, 14 insertions, 0 deletions

diff --git a/src/mongo/db/auth/user_management_commands_parser.h b/src/mongo/db/auth/user_management_commands_parser.h
index c55210e8978..2bc278feecb 100644
--- a/src/mongo/db/auth/user_management_commands_parser.h
+++ b/src/mongo/db/auth/user_management_commands_parser.h
@@ -223,5 +223,19 @@ struct MergeAuthzCollectionsArgs {
 Status parseMergeAuthzCollectionsCommand(const BSONObj& cmdObj,
                                          MergeAuthzCollectionsArgs* parsedArgs);
 
+struct AuthSchemaUpgradeArgs {
+    int maxSteps = 3;
+    bool shouldUpgradeShards = true;
+    BSONObj writeConcern;
+};
+
+/**
+ * Takes a command object describing an invocation of the "authSchemaUpgrade" command and
+ * parses out the write concern, maximum steps to take and whether or not shard servers should
+ * also be upgraded, in the sharded deployment case.
+ */
+Status parseAuthSchemaUpgradeCommand(const BSONObj& cmdObj,
+                                     const std::string& dbname,
+                                     AuthSchemaUpgradeArgs* parsedArgs);
 }  // namespace auth
 }  // namespace mongo