SERVER-23765 Update config string of shardIdentity document

1 files changed, 14 insertions, 0 deletions

diff --git a/src/mongo/db/s/type_shard_identity.cpp b/src/mongo/db/s/type_shard_identity.cpp
index f776e8cd573..7ef675ad334 100644
--- a/src/mongo/db/s/type_shard_identity.cpp
+++ b/src/mongo/db/s/type_shard_identity.cpp
@@ -30,16 +30,22 @@
 
 #include "mongo/db/s/type_shard_identity.h"
 
+#include "mongo/bson/bsonobjbuilder.h"
 #include "mongo/bson/util/bson_extract.h"
 #include "mongo/util/assert_util.h"
 
 namespace mongo {
 
 const std::string ShardIdentityType::IdName("shardIdentity");
+
+namespace {
+
 const BSONField<std::string> configsvrConnString("configsvrConnectionString");
 const BSONField<std::string> shardName("shardName");
 const BSONField<OID> clusterId("clusterId");
 
+}  // unnamed namespace
+
 StatusWith<ShardIdentityType> ShardIdentityType::fromBSON(const BSONObj& source) {
     if (!source.hasField("_id")) {
         return {ErrorCodes::NoSuchKey,
@@ -198,4 +204,12 @@ void ShardIdentityType::setClusterId(OID clusterId) {
     _clusterId = std::move(clusterId);
 }
 
+BSONObj ShardIdentityType::createConfigServerUpdateObject(const std::string& newConnString) {
+    BSONObjBuilder builder;
+    BSONObjBuilder setConfigBuilder(builder.subobjStart("$set"));
+    setConfigBuilder.append(configsvrConnString(), newConnString);
+    setConfigBuilder.doneFast();
+    return builder.obj();
+}
+
 }  // namespace mongo