summaryrefslogtreecommitdiff
path: root/src/mongo/s/write_ops/batch_write_op.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/mongo/s/write_ops/batch_write_op.cpp')
-rw-r--r--src/mongo/s/write_ops/batch_write_op.cpp7
1 files changed, 7 insertions, 0 deletions
diff --git a/src/mongo/s/write_ops/batch_write_op.cpp b/src/mongo/s/write_ops/batch_write_op.cpp
index 14e3fce88c1..7a4b0f88efe 100644
--- a/src/mongo/s/write_ops/batch_write_op.cpp
+++ b/src/mongo/s/write_ops/batch_write_op.cpp
@@ -36,6 +36,7 @@
#include "mongo/base/error_codes.h"
#include "mongo/db/operation_context.h"
+#include "mongo/s/client/num_hosts_targeted_metrics.h"
#include "mongo/stdx/memory.h"
#include "mongo/util/transitional_tools_do_not_use/vector_spooling.h"
@@ -378,6 +379,8 @@ Status BatchWriteOp::targetBatch(const NSTargeter& targeter,
targetedBatches->emplace(batch->getEndpoint().shardName, batch);
}
+ _nShardsOwningChunks = targeter.getNShardsOwningChunks();
+
return Status::OK();
}
@@ -750,6 +753,10 @@ int BatchWriteOp::numWriteOpsIn(WriteOpState opState) const {
});
}
+boost::optional<int> BatchWriteOp::getNShardsOwningChunks() {
+ return _nShardsOwningChunks;
+}
+
void BatchWriteOp::_incBatchStats(const BatchedCommandResponse& response) {
const auto batchType = _clientRequest.getBatchType();