Collect metrics from isolated cluster (#25681)

Author: Evgenik2

.github/last_commit.txt

@@ -1 +1 @@
-51b66527943329176e027929f58092c77e94c2f8
+265bf81d86234be2a1f2d2834ef72e36b4f43e73

include/ydb-cpp-sdk/client/monitoring/monitoring.h

@@ -5,6 +5,7 @@
 namespace Ydb {
 namespace Monitoring {
     class SelfCheckResult;
+    class ClusterStateResult;
 }
 }
 
@@ -34,6 +35,12 @@ struct TSelfCheckSettings : public TOperationRequestSettings<TSelfCheckSettings>
     FLUENT_SETTING_OPTIONAL(uint32_t, MaximumLevel);
 };
 
+struct TClusterStateSettings : public TOperationRequestSettings<TClusterStateSettings> {
+    FLUENT_SETTING_OPTIONAL(uint32_t, DurationSeconds);
+    FLUENT_SETTING_OPTIONAL(uint32_t, PeriodSeconds);
+};
+
+
 class TSelfCheckResult : public TStatus {
     friend class NYdb::TProtoAccessor;
 public:
@@ -45,13 +52,26 @@ class TSelfCheckResult : public TStatus {
 
 using TAsyncSelfCheckResult = NThreading::TFuture<TSelfCheckResult>;
 
+class TClusterStateResult : public TStatus {
+    friend class NYdb::TProtoAccessor;
+public:
+    TClusterStateResult(TStatus&& status, Ydb::Monitoring::ClusterStateResult&& result);
+private:
+    class TImpl;
+    std::shared_ptr<TImpl> Impl_;
+};
+
+using TAsyncClusterStateResult = NThreading::TFuture<TClusterStateResult>;
+
 class TMonitoringClient {
     class TImpl;
 
 public:
     TMonitoringClient(const TDriver& driver, const TCommonClientSettings& settings = TCommonClientSettings());
 
     TAsyncSelfCheckResult SelfCheck(const TSelfCheckSettings& settings = TSelfCheckSettings());
+
+    TAsyncClusterStateResult ClusterState(const TClusterStateSettings& settings = TClusterStateSettings());
 private:
     std::shared_ptr<TImpl> Impl_;
 };

include/ydb-cpp-sdk/client/proto/accessor.h

@@ -49,6 +49,7 @@ class TProtoAccessor {
     static const Ydb::Topic::DescribeTopicResult& GetProto(const NYdb::NTopic::TTopicDescription& topicDescription);
     static const Ydb::Topic::DescribeConsumerResult& GetProto(const NYdb::NTopic::TConsumerDescription& consumerDescription);
     static const Ydb::Monitoring::SelfCheckResult& GetProto(const NYdb::NMonitoring::TSelfCheckResult& selfCheckResult);
+    static const Ydb::Monitoring::ClusterStateResult& GetProto(const NYdb::NMonitoring::TClusterStateResult& clusterStateResult);
     static const Ydb::Coordination::DescribeNodeResult& GetProto(const NYdb::NCoordination::TNodeDescription& describeNodeResult);
     static const Ydb::Import::ListObjectsInS3ExportResult& GetProto(const NYdb::NImport::TListObjectsInS3ExportResult& result);
 

src/api/grpc/ydb_monitoring_v1.proto

@@ -10,4 +10,6 @@ service MonitoringService {
     rpc SelfCheck(Monitoring.SelfCheckRequest) returns (Monitoring.SelfCheckResponse);
     // Checks current node health
     rpc NodeCheck(Monitoring.NodeCheckRequest) returns (Monitoring.NodeCheckResponse);
+    // Get cluster state
+    rpc ClusterState(Monitoring.ClusterStateRequest) returns (Monitoring.ClusterStateResponse);
 }

src/api/protos/ydb_monitoring.proto

@@ -234,3 +234,18 @@ message SelfCheckResult {
     repeated DatabaseStatus database_status = 3;
     LocationNode location = 4;
 }
+
+message ClusterStateRequest {
+    Ydb.Operations.OperationParams operation_params = 1; // basic operation params, including timeout
+    uint32 duration_seconds = 2;
+    uint32 period_seconds = 3;
+}
+
+message ClusterStateResponse {
+    // After successfull completion must contain ClusterStateResult.
+    Ydb.Operations.Operation operation = 1;
+}
+
+message ClusterStateResult {
+    string result = 1;
+}

src/client/monitoring/monitoring.cpp

@@ -22,11 +22,24 @@ class TSelfCheckResult::TImpl {
     Ydb::Monitoring::SelfCheckResult Result;
 };
 
+class TClusterStateResult::TImpl {
+public:
+    TImpl(Ydb::Monitoring::ClusterStateResult&& result)
+        : Result(std::move(result))
+    {}
+    Ydb::Monitoring::ClusterStateResult Result;
+};
+
 TSelfCheckResult::TSelfCheckResult(TStatus&& status, Ydb::Monitoring::SelfCheckResult&& result)
     : TStatus(std::move(status))
     , Impl_(std::make_shared<TSelfCheckResult::TImpl>(std::move(result)))
 {}
 
+TClusterStateResult::TClusterStateResult(TStatus&& status, Ydb::Monitoring::ClusterStateResult&& result)
+    : TStatus(std::move(status))
+    , Impl_(std::make_shared<TClusterStateResult::TImpl>(std::move(result)))
+{}
+
 class TMonitoringClient::TImpl : public TClientImplCommon<TMonitoringClient::TImpl> {
 public:
     TImpl(std::shared_ptr<TGRpcConnectionsImpl>&& connections, const TCommonClientSettings& settings)
@@ -87,6 +100,45 @@ class TMonitoringClient::TImpl : public TClientImplCommon<TMonitoringClient::TIm
 
         return promise.GetFuture();
     }
+
+    TAsyncClusterStateResult ClusterState(const TClusterStateSettings& settings) {
+        auto request = MakeOperationRequest<Ydb::Monitoring::ClusterStateRequest>(settings);
+
+        if (settings.DurationSeconds_) {
+            request.set_duration_seconds(settings.DurationSeconds_.value());
+        }
+
+        if (settings.PeriodSeconds_) {
+            request.set_period_seconds(settings.PeriodSeconds_.value());
+        }
+        auto promise = NThreading::NewPromise<TClusterStateResult>();
+
+        auto extractor = [promise]
+            (google::protobuf::Any* any, TPlainStatus status) mutable {
+                Ydb::Monitoring::ClusterStateResult result;
+                if (any) {
+                    any->UnpackTo(&result);
+                }
+                TClusterStateResult val(
+                    TStatus(std::move(status)),
+                    std::move(result));
+
+                promise.SetValue(std::move(val));
+        };
+
+        using Ydb::Monitoring::ClusterStateRequest;
+        using Ydb::Monitoring::ClusterStateResponse;
+
+        Connections_->RunDeferred<Ydb::Monitoring::V1::MonitoringService, ClusterStateRequest, ClusterStateResponse>(
+            std::move(request),
+            extractor,
+            &Ydb::Monitoring::V1::MonitoringService::Stub::AsyncClusterState,
+            DbDriverState_,
+            INITIAL_DEFERRED_CALL_DELAY,
+            TRpcRequestSettings::Make(settings));
+
+        return promise.GetFuture();
+    }
 };
 
 TMonitoringClient::TMonitoringClient(const TDriver& driver, const TCommonClientSettings& settings)
@@ -97,10 +149,17 @@ TAsyncSelfCheckResult TMonitoringClient::SelfCheck(const TSelfCheckSettings& set
     return Impl_->SelfCheck(settings);
 }
 
+TAsyncClusterStateResult TMonitoringClient::ClusterState(const TClusterStateSettings& settings) {
+    return Impl_->ClusterState(settings);
+}
+
 }
 
 const Ydb::Monitoring::SelfCheckResult& TProtoAccessor::GetProto(const NYdb::NMonitoring::TSelfCheckResult& selfCheckResult) {
     return selfCheckResult.Impl_->Result;
 }
 
+const Ydb::Monitoring::ClusterStateResult& TProtoAccessor::GetProto(const NYdb::NMonitoring::TClusterStateResult& clusterStateResult) {
+    return clusterStateResult.Impl_->Result;
+}
 }