Handle all sorts of EvState events on shutdown

Author: abyss7

ydb/core/kqp/executer_actor/kqp_data_executer.cpp

@@ -2641,23 +2641,15 @@ class TKqpDataExecuter : public TKqpExecuterBase<TKqpDataExecuter, EExecType::Da
     }
 
     void HandleShutdown(TEvDqCompute::TEvState::TPtr& ev) {
-        if (ev->Get()->Record.GetState() == NDqProto::COMPUTE_STATE_FINISHED) {
-            YQL_ENSURE(Planner);
-
-            if (ev->Get()->Record.GetStatusCode() != NDqProto::StatusIds::ABORTED) {
-                return;
-            }
+        YQL_ENSURE(Planner);
 
-            TActorId actor = ev->Sender;
-            ui64 taskId = ev->Get()->Record.GetTaskId();
+        TActorId actor = ev->Sender;
+        ui64 taskId = ev->Get()->Record.GetTaskId();
 
-            Planner->CompletedCA(taskId, actor);
+        Planner->CompletedCA(taskId, actor);
 
-            if (Planner->GetPendingComputeTasks().empty() && Planner->GetPendingComputeActors().empty()) {
-                PassAway();
-            }
-        } else {
-            // TODO: handle other states.
+        if (Planner->GetPendingComputeTasks().empty() && Planner->GetPendingComputeActors().empty()) {
+            PassAway();
         }
     }