filling connections has been fixed

1 files changed, 14 insertions, 13 deletions

diff --git a/ydb/core/yq/libs/actors/run_actor.cpp b/ydb/core/yq/libs/actors/run_actor.cpp
index e4342a8732..0a73bc3aac 100644
--- a/ydb/core/yq/libs/actors/run_actor.cpp
+++ b/ydb/core/yq/libs/actors/run_actor.cpp
@@ -289,6 +289,8 @@ public:
     void Bootstrap() {
         LOG_D("Start run actor. Compute state: " << YandexQuery::QueryMeta::ComputeStatus_Name(Params.Status));
 
+        FillConnections();
+
         QueryCounters.SetUptimePublicAndServiceCounter((TInstant::Now() - CreatedAt).Seconds());
         QueryCounters.Counters->GetCounter("RetryCount", false)->Set(Params.RestartCount);
         LogReceivedParams();
@@ -374,6 +376,18 @@ private:
         IgnoreFunc(TEvDqStats);
     )
 
+    void FillConnections() {
+        LOG_D("FillConnections");
+
+        for (const auto& connection : Params.Connections) {
+            if (!connection.content().name()) {
+                LOG_D("Connection with empty name " << connection.meta().id());
+                continue;
+            }
+            YqConnections.emplace(connection.meta().id(), connection);
+        }
+    }
+
     void KillExecuter() {
         if (ExecuterId) {
             Send(ExecuterId, new NActors::TEvents::TEvPoison());
@@ -525,7 +539,6 @@ private:
             Finish(GetFinalStatusFromFinalizingStatus(Params.Status));
             break;
         case YandexQuery::QueryMeta::STARTING:
-            HandleConnections();
             QueryStateUpdateRequest.mutable_resources()->set_rate_limiter(
                 Params.RateLimiterConfig.GetEnabled() ? Fq::Private::TaskResources::PREPARE : Fq::Private::TaskResources::NOT_NEEDED);
             QueryStateUpdateRequest.mutable_resources()->set_compilation(Fq::Private::TaskResources::PREPARE);
@@ -543,18 +556,6 @@ private:
         }
     }
 
-    void HandleConnections() {
-        LOG_D("HandleConnections");
-
-        for (const auto& connection : Params.Connections) {
-            if (!connection.content().name()) {
-                LOG_D("Connection with empty name " << connection.meta().id());
-                continue;
-            }
-            YqConnections.emplace(connection.meta().id(), connection);
-        }
-    }
-
     void FailOnException() {
         Fail(CurrentExceptionMessage());
     }