diff options
| author | Aleksandr Gaev <aleksandr.gaev@tracto.ai> | 2024-09-05 10:34:41 +0300 |
|---|---|---|
| committer | robot-piglet <robot-piglet@yandex-team.com> | 2024-09-05 11:06:58 +0300 |
| commit | 0c708029742f34234612e3ff3ffd36fa18aad0f8 (patch) | |
| tree | 805c0be8c9baf2cc33322cb064a9e2f00b997cad | |
| parent | 7797fb8f1275e6c9bc031e86d4075df929cd5295 (diff) | |
| download | ydb-0c708029742f34234612e3ff3ffd36fa18aad0f8.tar.gz | |
Add QueryTrackerStage to GetQueryTrackerInfo response. Pass stage through rpc calls
Issue: #823
---
d01cae68878432db1730f7ad8e4b2f063e15ff1a
Pull Request resolved: https://github.com/ytsaurus/ytsaurus/pull/825
Co-authored-by: mpereskokova <mpereskokova@yandex-team.com>
| -rw-r--r-- | yt/yt/client/api/query_tracker_client.h | 1 | ||||
| -rw-r--r-- | yt/yt/client/api/rpc_proxy/client_impl.cpp | 1 | ||||
| -rw-r--r-- | yt/yt/client/driver/query_commands.cpp | 1 | ||||
| -rw-r--r-- | yt/yt_proto/yt/client/api/rpc_proxy/proto/api_service.proto | 1 |
4 files changed, 4 insertions, 0 deletions
diff --git a/yt/yt/client/api/query_tracker_client.h b/yt/yt/client/api/query_tracker_client.h index f4b8d48c4c..4e198ed4b3 100644 --- a/yt/yt/client/api/query_tracker_client.h +++ b/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; diff --git a/yt/yt/client/api/rpc_proxy/client_impl.cpp b/yt/yt/client/api/rpc_proxy/client_impl.cpp index 454d6fd16b..bd432d3d9e 100644 --- a/yt/yt/client/api/rpc_proxy/client_impl.cpp +++ b/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()), diff --git a/yt/yt/client/driver/query_commands.cpp b/yt/yt/client/driver/query_commands.cpp index d73dce2825..e3903d004b 100644 --- a/yt/yt/client/driver/query_commands.cpp +++ b/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) 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 f9606ee6bd..467bb6a78a 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 @@ -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; } //////////////////////////////////////////////////////////////////////////////// |