Fix track running tasks

Rafał Hibner · Rafał Hibner · commit 04eedc6d320d · 2025-09-30T12:08:37.000Z
diff --git a/cpp/src/arrow/util/async_util.cc b/cpp/src/arrow/util/async_util.cc
@@ -170,7 +170,6 @@ class AsyncTaskSchedulerImpl : public AsyncTaskScheduler {
     if (IsAborted()) {
       return false;
     }
-    running_tasks_set_.insert(task.get());
     SubmitTaskUnlocked(std::move(task), std::move(lk));
     return true;
   }
@@ -217,6 +216,8 @@ class AsyncTaskSchedulerImpl : public AsyncTaskScheduler {
           };
         })) {
       return OnTaskFinished(submit_result->status());
+    } else {
+      running_tasks_set_.insert(task.get());
     }
   }
 

Original file line number	Diff line number	Diff line change
`@@ -170,7 +170,6 @@ class AsyncTaskSchedulerImpl : public AsyncTaskScheduler {`
`170`	`170`	`if (IsAborted()) {`
`171`	`171`	`return false;`
`172`	`172`	`}`
`173`		`- running_tasks_set_.insert(task.get());`
`174`	`173`	`SubmitTaskUnlocked(std::move(task), std::move(lk));`
`175`	`174`	`return true;`
`176`	`175`	`}`
`@@ -217,6 +216,8 @@ class AsyncTaskSchedulerImpl : public AsyncTaskScheduler {`
`217`	`216`	`};`
`218`	`217`	`})) {`
`219`	`218`	`return OnTaskFinished(submit_result->status());`
	`219`	`+ } else {`
	`220`	`+ running_tasks_set_.insert(task.get());`
`220`	`221`	`}`
`221`	`222`	`}`
`222`	`223`