[dq] Separate statistics/counters for HandlePull/HandleResult/HandlePrecompute

1 files changed, 15 insertions, 0 deletions

diff --git a/ydb/library/yql/providers/dq/provider/exec/yql_dq_exectransformer.cpp b/ydb/library/yql/providers/dq/provider/exec/yql_dq_exectransformer.cpp
index 6e5b852c0ce..f1c3797d8d5 100644
--- a/ydb/library/yql/providers/dq/provider/exec/yql_dq_exectransformer.cpp
+++ b/ydb/library/yql/providers/dq/provider/exec/yql_dq_exectransformer.cpp
@@ -789,6 +789,11 @@ private:
                         future = NThreading::MakeFuture<IDqGateway::TResult>(std::move(result));
                     }
                 } else {
+                    if (State->Metrics) {
+                        State->Metrics->IncCounter("dq", "HandleResult");
+                    }
+                    State->Statistics[State->MetricId++].Entries.push_back(TOperationStatistics::TEntry("HandleResult", 0, 0, 0, 0, 1));
+
                     graphParams["Evaluation"] = ToString(!ctx.Step.IsDone(TExprStep::ExprEval));
                     future = State->ExecutePlan(
                         State->SessionId, executionPlanner->GetPlan(), columns, secureParams, graphParams,
@@ -1150,6 +1155,11 @@ private:
             settings->_RowsLimitPerWrite = 0;
         }
 
+        if (State->Metrics) {
+            State->Metrics->IncCounter("dq", "HandlePull");
+        }
+        State->Statistics[State->MetricId++].Entries.push_back(TOperationStatistics::TEntry("HandlePull", 0, 0, 0, 0, 1));
+
         IDqGateway::TDqProgressWriter progressWriter = MakeDqProgressWriter(publicIds);
 
         auto future = State->ExecutePlan(State->SessionId, executionPlanner->GetPlan(), columns, secureParams, graphParams,
@@ -1595,6 +1605,11 @@ private:
                 return FallbackWithMessage(*input, "Too big attachment", ctx, false);
             }
 
+            if (State->Metrics) {
+                State->Metrics->IncCounter("dq", "Precompute");
+            }
+            State->Statistics[State->MetricId++].Entries.push_back(TOperationStatistics::TEntry("Precompute", 0, 0, 0, 0, 1));
+
             auto graphParams = GatherGraphParams(optimizedInput);
             graphParams["Precompute"] = ToString(true);
             MarkProgressStarted(publicIds->AllPublicIds, State->ProgressWriter);