From 9e6fc1f5965f33c97036a6e9059499137fdd430f Mon Sep 17 00:00:00 2001
From: Grigoriy Pisarenko <grigoriypisar@ydb.tech>
Date: Fri, 30 Aug 2024 07:17:26 +0000
Subject: [PATCH] Fixed compute actor finalizing in case of node disconnect

---
 ydb/core/kqp/executer_actor/kqp_planner.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ydb/core/kqp/executer_actor/kqp_planner.cpp b/ydb/core/kqp/executer_actor/kqp_planner.cpp
index b759ecb32347..c9340c318079 100644
--- a/ydb/core/kqp/executer_actor/kqp_planner.cpp
+++ b/ydb/core/kqp/executer_actor/kqp_planner.cpp
@@ -643,6 +643,9 @@ void TKqpPlanner::TaskNotStarted(ui64 taskId) {
     YQL_ENSURE(!task.ComputeActorId);
     YQL_ENSURE(!task.Meta.Completed);
 
+    // Mark as completed, because node disconnect may be temporary, and CA will send TEvDqCompute::TEvState while shutting down
+    task.Meta.Completed = true;
+
     PendingComputeTasks.erase(taskId);
 }