Add QueryTrackerStage to GetQueryTrackerInfo response. Pass stage through rpc calls

Issue: #823

---
d01cae68878432db1730f7ad8e4b2f063e15ff1a

Pull Request resolved: ytsaurus/ytsaurus#825

Co-authored-by: mpereskokova <[email protected]>

Author: 2 people

yt/yt/client/api/query_tracker_client.h

@@ -152,6 +152,7 @@ struct TGetQueryTrackerInfoOptions
 
 struct TGetQueryTrackerInfoResult
 {
+    TString QueryTrackerStage;
     TString ClusterName;
     NYson::TYsonString SupportedFeatures;
     std::vector<TString> AccessControlObjects;

yt/yt/client/api/rpc_proxy/client_impl.cpp

@@ -2406,6 +2406,7 @@ TFuture<TGetQueryTrackerInfoResult> TClient::GetQueryTrackerInfo(
 
     return req->Invoke().Apply(BIND([] (const TApiServiceProxy::TRspGetQueryTrackerInfoPtr& rsp) {
         return TGetQueryTrackerInfoResult{
+            .QueryTrackerStage = rsp->query_tracker_stage(),
             .ClusterName = rsp->cluster_name(),
             .SupportedFeatures = TYsonString(rsp->supported_features()),
             .AccessControlObjects = FromProto<std::vector<TString>>(rsp->access_control_objects()),

yt/yt/client/driver/query_commands.cpp

@@ -386,6 +386,7 @@ void TGetQueryTrackerInfoCommand::DoExecute(ICommandContextPtr context)
 
     context->ProduceOutputValue(BuildYsonStringFluently()
         .BeginMap()
+            .Item("query_tracker_stage").Value(result.QueryTrackerStage)
             .Item("cluster_name").Value(result.ClusterName)
             .Item("supported_features").Value(result.SupportedFeatures)
             .Item("access_control_objects").Value(result.AccessControlObjects)

yt/yt_proto/yt/client/api/rpc_proxy/proto/api_service.proto

@@ -3368,6 +3368,7 @@ message TRspGetQueryTrackerInfo
     required string cluster_name = 1;
     required bytes supported_features = 2; // YSON
     repeated string access_control_objects = 3;
+    optional string query_tracker_stage = 4;
 }
 
 ////////////////////////////////////////////////////////////////////////////////