From 72e24ba9bbc1c3043d35d47e30cda75075ff3d95 Mon Sep 17 00:00:00 2001
From: Will Buerger <will.buerger@mongodb.com>
Date: Fri, 5 May 2023 13:36:36 +0000
Subject: SERVER-76557: Keep RequestShapifiers in telemetry store

---
 src/mongo/s/query/cluster_cursor_manager.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'src/mongo/s/query/cluster_cursor_manager.cpp')

diff --git a/src/mongo/s/query/cluster_cursor_manager.cpp b/src/mongo/s/query/cluster_cursor_manager.cpp
index 313f283a5b5..1cfbd7bf700 100644
--- a/src/mongo/s/query/cluster_cursor_manager.cpp
+++ b/src/mongo/s/query/cluster_cursor_manager.cpp
@@ -591,14 +591,15 @@ StatusWith<ClusterClientCursorGuard> ClusterCursorManager::_detachCursor(WithLoc
     return std::move(cursor);
 }
 
-void collectTelemetryMongos(OperationContext* opCtx, const BSONObj& originatingCommand) {
+void collectTelemetryMongos(OperationContext* opCtx,
+                            std::unique_ptr<telemetry::RequestShapifier> requestShapifier) {
     // If we haven't registered a cursor to prepare for getMore requests, we record
     // telemetry directly.
     auto&& opDebug = CurOp::get(opCtx)->debug();
     telemetry::writeTelemetry(
         opCtx,
         opDebug.telemetryStoreKey,
-        originatingCommand,
+        std::move(requestShapifier),
         opDebug.additiveMetrics.executionTime.value_or(Microseconds{0}).count(),
         opDebug.additiveMetrics.nreturned.value_or(0));
 }
-- 
cgit v1.2.1