diff options
Diffstat (limited to 'src/mongo/shell/bench.cpp')
| -rw-r--r-- | src/mongo/shell/bench.cpp | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/src/mongo/shell/bench.cpp b/src/mongo/shell/bench.cpp index 5dbcd7ff782..5e5a71c5d42 100644 --- a/src/mongo/shell/bench.cpp +++ b/src/mongo/shell/bench.cpp @@ -542,6 +542,10 @@ void BenchRunWorker::generateLoadOnConnection(DBClientBase* conn) { docBuilder.append(BSON("q" << query << "u" << update << "multi" << multi << "upsert" << upsert)); docBuilder.done(); + auto wcElem = e["writeConcern"]; + if (wcElem.ok()) { + builder.append("writeConcern", wcElem.Obj()); + } conn->runCommand( nsToDatabaseSubstring(ns).toString(), builder.done(), result); } else { @@ -583,6 +587,7 @@ void BenchRunWorker::generateLoadOnConnection(DBClientBase* conn) { BSONObj insertDoc; if (useWriteCmd) { + // TODO: Replace after SERVER-11774. BSONObjBuilder builder; builder.append("insert", nsToCollectionSubstring(ns)); BSONArrayBuilder docBuilder(builder.subarrayStart("documents")); @@ -596,7 +601,10 @@ void BenchRunWorker::generateLoadOnConnection(DBClientBase* conn) { docBuilder.append(insertDoc); } docBuilder.done(); - // TODO: Replace after SERVER-11774. + auto wcElem = e["writeConcern"]; + if (wcElem.ok()) { + builder.append("writeConcern", wcElem.Obj()); + } conn->runCommand( nsToDatabaseSubstring(ns).toString(), builder.done(), result); } else { @@ -655,6 +663,10 @@ void BenchRunWorker::generateLoadOnConnection(DBClientBase* conn) { int limit = (multi == true) ? 0 : 1; docBuilder.append(BSON("q" << predicate << "limit" << limit)); docBuilder.done(); + auto wcElem = e["writeConcern"]; + if (wcElem.ok()) { + builder.append("writeConcern", wcElem.Obj()); + } conn->runCommand( nsToDatabaseSubstring(ns).toString(), builder.done(), result); } else { |