Merge pull request #11 from KlassnayaAfrodita/patch-2

Update concurrency.go
hokamsingh · Oct 11, 2024 · 710e53d · 710e53d
2 parents d9b95fd + c2ff575
commit 710e53d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/internal/core/concurrency/concurrency.go b/internal/core/concurrency/concurrency.go
@@ -130,7 +130,12 @@ func (tm *TaskManager) runParallel(ctx context.Context) ([]interface{}, error) {
 	// Submit tasks to the worker pool
 	for i, task := range tm.tasks {
 		go func(index int, task *Task) {
-			pool.Submit(task, index)
+			select {
+			case <-ctx.Done():
+				errChan <- ctx.Err()
+			default:
+				pool.Submit(task, index)
+			}
 		}(i, task)
 	}