Add delay before shutdown

Author: pixcc

ydb/core/driver_lib/run/run.cpp

@@ -504,6 +504,9 @@ static TString ReadFile(const TString& fileName) {
 void TKikimrRunner::InitializeGracefulShutdown(const TKikimrRunConfig& runConfig) {
     Y_UNUSED(runConfig);
     GracefulShutdownSupported = true;
+    if (runConfig.AppConfig.HasDelayBeforeShutdownSeconds()) {
+        DelayBeforeShutdown = TDuration::Seconds(runConfig.AppConfig.GetDelayBeforeShutdownSeconds());
+    }
 }
 
 void TKikimrRunner::InitializeKqpController(const TKikimrRunConfig& runConfig) {
@@ -1732,6 +1735,7 @@ void TKikimrRunner::KikimrStop(bool graceful) {
 
     if (EnabledGrpcService) {
         ActorSystem->Send(new IEventHandle(NGRpcService::CreateGrpcPublisherServiceActorId(), {}, new TEvents::TEvPoisonPill));
+        Sleep(DelayBeforeShutdown);
     }
 
     TIntrusivePtr<TDrainProgress> drainProgress(new TDrainProgress());

ydb/core/driver_lib/run/run.h

@@ -43,6 +43,7 @@ class TKikimrRunner : public virtual TThrRefBase, private IGlobalObjectStorage {
 
     bool EnabledGrpcService = false;
     bool GracefulShutdownSupported = false;
+    TDuration DelayBeforeShutdown;
     THolder<NSQS::TAsyncHttpServer> SqsHttp;
 
     THolder<NYdb::TDriver> YdbDriver;

ydb/core/protos/config.proto

@@ -1989,6 +1989,7 @@ message TAppConfig {
     optional TMemoryControllerConfig MemoryControllerConfig = 81;
 	optional TGroupedMemoryLimiterConfig GroupedMemoryLimiterConfig = 82;
     optional NKikimrReplication.TReplicationDefaults ReplicationConfig = 83;
+    optional uint32 DelayBeforeShutdownSeconds = 108;
 
     repeated TNamedConfig NamedConfigs = 100;
     optional string ClusterYamlConfig = 101;