SERVER-44689 Add serverStatus counter for each use of an aggregation stage in a user's request

1 files changed, 14 insertions, 0 deletions

diff --git a/src/mongo/db/pipeline/lite_parsed_pipeline.cpp b/src/mongo/db/pipeline/lite_parsed_pipeline.cpp
index 954239fa928..153912ef448 100644
--- a/src/mongo/db/pipeline/lite_parsed_pipeline.cpp
+++ b/src/mongo/db/pipeline/lite_parsed_pipeline.cpp
@@ -32,6 +32,7 @@
 #include "mongo/db/pipeline/lite_parsed_pipeline.h"
 
 #include "mongo/db/operation_context.h"
+#include "mongo/db/stats/counters.h"
 
 namespace mongo {
 
@@ -110,4 +111,17 @@ void LiteParsedPipeline::verifyIsSupported(
     }
 }
 
+void LiteParsedPipeline::tickGlobalStageCounters() const {
+    for (auto&& stage : _stageSpecs) {
+        // Tick counter corresponding to current stage.
+        aggStageCounters.stageCounterMap.find(stage->getParseTimeName())
+            ->second->counter.increment(1);
+
+        // Recursively step through any sub-pipelines.
+        for (auto&& subPipeline : stage->getSubPipelines()) {
+            subPipeline.tickGlobalStageCounters();
+        }
+    }
+}
+
 }  // namespace mongo