Expose connectivity api for grpc channels

commit_hash:f8240948024f6fa2898b60d65635d3c9922359d3

yt/yt/core/rpc/grpc/channel.cpp

@@ -126,7 +126,7 @@ DEFINE_ENUM(EClientCallStage,
 );
 
 class TChannel
-    : public IChannel
+    : public NYT::NRpc::NGrpc::IGrpcChannel
 {
 public:
     explicit TChannel(TChannelConfigPtr config)
@@ -221,6 +221,11 @@ class TChannel
         return MemoryUsageTracker_;
     }
 
+    grpc_connectivity_state CheckConnectivityState(bool tryToConnect) override
+    {
+        return grpc_channel_check_connectivity_state(Channel_.Unwrap(), tryToConnect);
+    }
+
 private:
     const TChannelConfigPtr Config_;
     const TString EndpointAddress_;
@@ -736,7 +741,7 @@ class TChannelFactory
 
 } // namespace
 
-IChannelPtr CreateGrpcChannel(TChannelConfigPtr config)
+IGrpcChannelPtr CreateGrpcChannel(TChannelConfigPtr config)
 {
     return New<TChannel>(std::move(config));
 }

yt/yt/core/rpc/grpc/channel.h

@@ -3,13 +3,27 @@
 #include "public.h"
 
 #include <yt/yt/core/rpc/public.h>
+#include <yt/yt/core/rpc/channel.h>
+
+#include <contrib/libs/grpc/include/grpc/impl/connectivity_state.h>
 
 namespace NYT::NRpc::NGrpc {
 
 ////////////////////////////////////////////////////////////////////////////////
 
+struct IGrpcChannel
+    : public NRpc::IChannel
+{
+public:
+    virtual grpc_connectivity_state CheckConnectivityState(bool tryToConnect) = 0;
+};
+
+DEFINE_REFCOUNTED_TYPE(IGrpcChannel)
+
+////////////////////////////////////////////////////////////////////////////////
+
 //! Creates a channel implemented via GRPC.
-NRpc::IChannelPtr CreateGrpcChannel(TChannelConfigPtr config);
+IGrpcChannelPtr CreateGrpcChannel(TChannelConfigPtr config);
 
 //! Returns the factory for creating GRPC channels.
 NRpc::IChannelFactoryPtr GetGrpcChannelFactory();

yt/yt/core/rpc/grpc/public.h

@@ -15,6 +15,8 @@ DECLARE_REFCOUNTED_CLASS(TChannelCredentialsConfig)
 DECLARE_REFCOUNTED_CLASS(TChannelConfigTemplate)
 DECLARE_REFCOUNTED_CLASS(TChannelConfig)
 
+DECLARE_REFCOUNTED_STRUCT(IGrpcChannel)
+
 ////////////////////////////////////////////////////////////////////////////////
 
 extern const char* const TracingTraceIdMetadataKey;