SERVER-24126 Add step to _cfgsvrAddShard command where it inserts the shardIdentity document to the new shard

1 files changed, 10 insertions, 0 deletions

diff --git a/src/mongo/db/s/type_shard_identity.h b/src/mongo/db/s/type_shard_identity.h
index 30e2c832dff..9c51913fe0f 100644
--- a/src/mongo/db/s/type_shard_identity.h
+++ b/src/mongo/db/s/type_shard_identity.h
@@ -32,6 +32,7 @@
 
 #include "mongo/client/connection_string.h"
 #include "mongo/db/jsobj.h"
+#include "mongo/s/write_ops/batched_update_request.h"
 
 namespace mongo {
 
@@ -40,6 +41,8 @@ namespace mongo {
  */
 class ShardIdentityType {
 public:
+    ShardIdentityType() = default;
+
     // The _id value for this document type.
     static const std::string IdName;
 
@@ -78,6 +81,13 @@ public:
     void setClusterId(OID clusterId);
 
     /**
+     * Returns an update object that can be used to insert a shardIdentity doc on a shard or update
+     * the existing shardIdentity doc's configsvrConnString (if the _id, shardName, and clusterId
+     * match those in this ShardIdentityType instance).
+     */
+    std::unique_ptr<BatchedUpdateRequest> createUpsertForAddShard() const;
+
+    /**
      * Returns an update object that can be used to update the config server field of the
      * shardIdentity document with the new connection string.
      */