Fixed compute actor finalizing in case of node disconnect

GrigoriyPA · GrigoriyPA · commit 9e6fc1f5965f · 2024-08-30T09:00:22.000Z
diff --git 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);
 }
 

Original file line number	Diff line number	Diff line change
`@@ -643,6 +643,9 @@ void TKqpPlanner::TaskNotStarted(ui64 taskId) {`
`643`	`643`	`YQL_ENSURE(!task.ComputeActorId);`
`644`	`644`	`YQL_ENSURE(!task.Meta.Completed);`
`645`	`645`
	`646`	`+ // Mark as completed, because node disconnect may be temporary, and CA will send TEvDqCompute::TEvState while shutting down`
	`647`	`+ task.Meta.Completed = true;`
	`648`	`+`
`646`	`649`	`PendingComputeTasks.erase(taskId);`
`647`	`650`	`}`
`648`	`651`