Fix incorrect stopping actorsystem before grpc (#27535)

Author: kruall

ydb/core/driver_lib/run/run.cpp

@@ -177,11 +177,17 @@ namespace NKikimr {
 
 namespace {
 
-    void StopGRpcServers(std::weak_ptr<TGRpcServersWrapper> grpcServersWrapper) {
+    void StopGRpcServers(std::weak_ptr<TGRpcServersWrapper> grpcServersWrapper, bool isDisabled = false) {
         auto wrapper = grpcServersWrapper.lock();
         if (!wrapper) {
             return;
         }
+        if (wrapper->IsDisabled.load(std::memory_order_acquire)) {
+            return;
+        }
+        if (isDisabled) {
+            wrapper->IsDisabled.store(true, std::memory_order_release);
+        }
         TGuard<TMutex> guard = wrapper->Guard();
         for (auto& [name, server] : wrapper->Servers) {
             if (!server) {
@@ -191,7 +197,6 @@ namespace {
         }
         wrapper->Servers.clear();
     }
-
 }
 
 class TGRpcServersManager : public TActorBootstrapped<TGRpcServersManager> {
@@ -265,6 +270,9 @@ class TGRpcServersManager : public TActorBootstrapped<TGRpcServersManager> {
         if (!wrapper) {
             return;
         }
+        if (wrapper->IsDisabled.load(std::memory_order_acquire)) {
+            return;
+        }
         TGuard<TMutex> guard = wrapper->Guard();
         wrapper->Servers = wrapper->GrpcServersFactory();
         for (auto& [name, server] : wrapper->Servers) {
@@ -317,6 +325,7 @@ class TGRpcServersManager : public TActorBootstrapped<TGRpcServersManager> {
         hFunc(TEvStop, HandleStop)
         cFunc(TEvGRpcServersManager::EvDisconnectRequestStarted, HandleDisconnectRequestStarted)
         cFunc(TEvGRpcServersManager::EvDisconnectRequestFinished, HandleDisconnectRequestFinished)
+        cFunc(TEvents::TSystem::PoisonPill, PassAway)
     )
 };
 
@@ -2153,14 +2162,13 @@ void TKikimrRunner::KikimrStop(bool graceful) {
         SqsHttp.Destroy();
     }
 
-    if (ActorSystem) {
-        ActorSystem->Stop();
-    }
-
     // stop processing grpc requests/response - we must stop feeding ActorSystem
     if (GRpcServersManager) {
-        StopGRpcServers(GRpcServersWrapper);
-        GRpcServersWrapper->Servers.clear();
+        StopGRpcServers(GRpcServersWrapper, true);
+    }
+
+    if (ActorSystem) {
+        ActorSystem->Stop();
     }
 
     if (YqSharedResources) {

ydb/core/driver_lib/run/run.h

@@ -33,6 +33,7 @@ struct TGRpcServersWrapper {
     TGRpcServers Servers;
     TGRpcServersFactory GrpcServersFactory;
     TMutex Mutex;
+    std::atomic<bool> IsDisabled = false;
 
     TGuard<TMutex> Guard() {
         return TGuard<TMutex>(Mutex);