YT-22938: Total grouped rows statistic in QL

* Changelog entry
  Type: feature
  Component: dynamic-tables

Added total_qrouped_row_count to QL statistics.
commit_hash:44a2bde635bf51ad758b55e2a7affb13aca8ebee

4 files changed, 5 insertions, 0 deletions

diff --git a/yt/yt/client/query_client/query_statistics.cpp b/yt/yt/client/query_client/query_statistics.cpp
index 1ae041a9952..3db6b343ac5 100644
--- a/yt/yt/client/query_client/query_statistics.cpp
+++ b/yt/yt/client/query_client/query_statistics.cpp
@@ -38,6 +38,7 @@ void ToProto(NProto::TQueryStatistics* serialized, const TQueryStatistics& origi
     serialized->set_incomplete_input(original.IncompleteInput);
     serialized->set_incomplete_output(original.IncompleteOutput);
     serialized->set_memory_usage(original.MemoryUsage);
+    serialized->set_total_grouped_row_count(original.TotalGroupedRowCount);
     ToProto(serialized->mutable_inner_statistics(), original.InnerStatistics);
 }
 
@@ -56,6 +57,7 @@ void FromProto(TQueryStatistics* original, const NProto::TQueryStatistics& seria
     original->IncompleteInput = serialized.incomplete_input();
     original->IncompleteOutput = serialized.incomplete_output();
     original->MemoryUsage = serialized.memory_usage();
+    original->TotalGroupedRowCount = serialized.total_grouped_row_count();
     FromProto(&original->InnerStatistics, serialized.inner_statistics());
 }
 
diff --git a/yt/yt/client/query_client/query_statistics.h b/yt/yt/client/query_client/query_statistics.h
index a759a09372e..06063b7d494 100644
--- a/yt/yt/client/query_client/query_statistics.h
+++ b/yt/yt/client/query_client/query_statistics.h
@@ -23,6 +23,7 @@ struct TQueryStatistics
     bool IncompleteInput = false;
     bool IncompleteOutput = false;
     size_t MemoryUsage = 0;
+    i64 TotalGroupedRowCount = 0;
 
     std::vector<TQueryStatistics> InnerStatistics;
 
diff --git a/yt/yt_proto/yt/client/api/rpc_proxy/proto/api_service.proto b/yt/yt_proto/yt/client/api/rpc_proxy/proto/api_service.proto
index 9c9e018be9d..37ceb255230 100644
--- a/yt/yt_proto/yt/client/api/rpc_proxy/proto/api_service.proto
+++ b/yt/yt_proto/yt/client/api/rpc_proxy/proto/api_service.proto
@@ -605,6 +605,7 @@ message TQueryStatistics
     optional bool incomplete_input = 11;
     optional bool incomplete_output = 12;
     optional uint64 memory_usage = 15;
+    optional int64 total_grouped_row_count = 16;
 
     repeated TQueryStatistics inner_statistics = 14;
 }
diff --git a/yt/yt_proto/yt/client/query_client/proto/query_statistics.proto b/yt/yt_proto/yt/client/query_client/proto/query_statistics.proto
index acb02c2c8a1..0a44f858efb 100644
--- a/yt/yt_proto/yt/client/query_client/proto/query_statistics.proto
+++ b/yt/yt_proto/yt/client/query_client/proto/query_statistics.proto
@@ -14,6 +14,7 @@ message TQueryStatistics
     required uint64 write_time = 7;
     required uint64 codegen_time = 10;
     optional uint64 wait_on_ready_event_time = 14 [ default = 0 ];
+    optional int64 total_grouped_row_count = 15;
 
     required bool incomplete_input = 8;
     required bool incomplete_output = 9;