Fixed CA checking for HandleShutdown TEvDqCompute::TEvState

Author: GrigoriyPA

ydb/core/kqp/executer_actor/kqp_data_executer.cpp

@@ -2684,10 +2684,16 @@ class TKqpDataExecuter : public TKqpExecuterBase<TKqpDataExecuter, EExecType::Da
         if (ev->Get()->Record.GetState() == NDqProto::COMPUTE_STATE_FAILURE) {
             YQL_ENSURE(Planner);
 
-            TActorId actor = ev->Sender;
-            ui64 taskId = ev->Get()->Record.GetTaskId();
-
-            Planner->CompletedCA(taskId, actor);
+            const auto& task = TasksGraph.GetTask(ev->Get()->Record.GetTaskId());
+            if (!task.Meta.Completed) {
+                if (task.ComputeActorId) {
+                    YQL_ENSURE(ev->Sender == task.ComputeActorId);
+                    Planner->CompletedCA(task.Id, task.ComputeActorId);
+                } else {
+                    // CA didn't have time to register
+                    Planner->TaskNotStarted(task.Id);
+                }
+            }
 
             if (Planner->GetPendingComputeTasks().empty() && Planner->GetPendingComputeActors().empty()) {
                 PassAway();