Add ChainActor ping kind (#12917)

Author: eivanov89

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

@@ -41,21 +41,35 @@ class TTxProxyPingResult: public TStatus {
     {}
 };
 
+class TActorChainPingResult: public TStatus {
+public:
+    TActorChainPingResult(TStatus&& status)
+        : TStatus(std::move(status))
+    {}
+};
+
 ////////////////////////////////////////////////////////////////////////////////
 
 using TAsyncPlainGrpcPingResult = NThreading::TFuture<TPlainGrpcPingResult>;
 using TAsyncGrpcProxyPingResult = NThreading::TFuture<TGrpcProxyPingResult>;
 using TAsyncKqpProxyPingResult = NThreading::TFuture<TKqpProxyPingResult>;
 using TAsyncSchemeCachePingResult = NThreading::TFuture<TSchemeCachePingResult>;
 using TAsyncTxProxyPingResult = NThreading::TFuture<TTxProxyPingResult>;
+using TAsyncActorChainPingResult = NThreading::TFuture<TActorChainPingResult>;
 
 ////////////////////////////////////////////////////////////////////////////////
 
-struct TPlainGrpcPingSettings : public TOperationRequestSettings<TPlainGrpcPingSettings> {};
-struct TGrpcProxyPingSettings : public TOperationRequestSettings<TGrpcProxyPingSettings> {};
-struct TKqpProxyPingSettings : public TOperationRequestSettings<TKqpProxyPingSettings> {};
-struct TSchemeCachePingSettings : public TOperationRequestSettings<TSchemeCachePingSettings> {};
-struct TTxProxyPingSettings : public TOperationRequestSettings<TTxProxyPingSettings> {};
+struct TPlainGrpcPingSettings : public TSimpleRequestSettings<TPlainGrpcPingSettings> {};
+struct TGrpcProxyPingSettings : public TSimpleRequestSettings<TGrpcProxyPingSettings> {};
+struct TKqpProxyPingSettings : public TSimpleRequestSettings<TKqpProxyPingSettings> {};
+struct TSchemeCachePingSettings : public TSimpleRequestSettings<TSchemeCachePingSettings> {};
+struct TTxProxyPingSettings : public TSimpleRequestSettings<TTxProxyPingSettings> {};
+
+struct TActorChainPingSettings : public TSimpleRequestSettings<TActorChainPingSettings> {
+    FLUENT_SETTING_DEFAULT(size_t, ChainLength, 10);
+    FLUENT_SETTING_DEFAULT(size_t, WorkUsec, 5);
+    FLUENT_SETTING_DEFAULT(bool, NoTailChain, false);
+};
 
 ////////////////////////////////////////////////////////////////////////////////
 
@@ -75,6 +89,8 @@ class TDebugClient {
     TAsyncSchemeCachePingResult PingSchemeCache(const TSchemeCachePingSettings& settings);
     TAsyncTxProxyPingResult PingTxProxy(const TTxProxyPingSettings& settings);
 
+    TAsyncActorChainPingResult PingActorChain(const TActorChainPingSettings& settings);
+
 private:
     class TImpl;
     std::shared_ptr<TImpl> Impl_;

src/api/grpc/ydb_debug_v1.proto

@@ -11,4 +11,5 @@ service DebugService {
     rpc PingKqpProxy(Debug.KqpProxyRequest) returns (Debug.KqpProxyResponse);
     rpc PingSchemeCache(Debug.SchemeCacheRequest) returns (Debug.SchemeCacheResponse);
     rpc PingTxProxy(Debug.TxProxyRequest) returns (Debug.TxProxyResponse);
+    rpc PingActorChain(Debug.ActorChainRequest) returns (Debug.ActorChainResponse);
 }

src/api/protos/ydb_debug.proto

@@ -57,3 +57,22 @@ message TxProxyResponse {
     StatusIds.StatusCode status = 1;
     repeated Ydb.Issue.IssueMessage issues = 2;
 }
+
+// Ping Actor Chain
+
+message ActorChainRequest {
+
+    // number of actors to be created, default 10
+    uint32 ChainLength = 1;
+
+    // immitate work duration for each actor (approximate), default ~ 5 usec
+    uint32 WorkUsec = 2;
+
+    // don't use tail sends and registrations
+    bool NoTailChain = 3;
+}
+
+message ActorChainResponse {
+    StatusIds.StatusCode status = 1;
+    repeated Ydb.Issue.IssueMessage issues = 2;
+}

src/client/debug/client.cpp

@@ -40,6 +40,28 @@ class TDebugClient::TImpl: public TClientImplCommon<TDebugClient::TImpl> {
         return pingPromise;
     }
 
+    auto PingActorChain(const TActorChainPingSettings& settings) {
+        auto pingPromise = NewPromise<TActorChainPingResult>();
+        auto responseCb = [pingPromise] (Debug::ActorChainResponse*, TPlainStatus status) mutable {
+            TActorChainPingResult val(TStatus(std::move(status)));
+            pingPromise.SetValue(std::move(val));
+        };
+
+        Debug::ActorChainRequest request;
+        request.SetChainLength(settings.ChainLength_);
+        request.SetWorkUsec(settings.WorkUsec_);
+        request.SetNoTailChain(settings.NoTailChain_);
+
+        Connections_->Run<Debug::V1::DebugService, Debug::ActorChainRequest, Debug::ActorChainResponse>(
+            std::move(request),
+            responseCb,
+            &Debug::V1::DebugService::Stub::AsyncPingActorChain,
+            DbDriverState_,
+            TRpcRequestSettings::Make(settings));
+
+        return pingPromise;
+    }
+
     ~TImpl() = default;
 };
 
@@ -73,4 +95,8 @@ TAsyncTxProxyPingResult TDebugClient::PingTxProxy(const TTxProxyPingSettings& se
         settings, &Debug::V1::DebugService::Stub::AsyncPingTxProxy);
 }
 
-} // namespace NYdb::NDebug
+TAsyncActorChainPingResult TDebugClient::PingActorChain(const TActorChainPingSettings& settings) {
+    return Impl_->PingActorChain(settings);
+}
+
+} // namespace NYdb::V3::NDebug