fix: proper use of extensions channel

hatchet-dev · Dec 20, 2024 · 712319e · 712319e
1 parent 2d079ed
commit 712319e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/scheduling/v2/scheduler.go b/pkg/scheduling/v2/scheduler.go
@@ -668,7 +668,6 @@ func (s *Scheduler) tryAssign(
 		startTotal := time.Now()
 
 		extensionResultsCh := make(chan *assignResults, len(actionIdToQueueItems))
-		defer close(extensionResultsCh)
 
 		// process each action id in parallel
 		for actionId, qis := range actionIdToQueueItems {
@@ -760,6 +759,7 @@ func (s *Scheduler) tryAssign(
 		wg.Wait()
 		span.End()
 		close(resultsCh)
+		close(extensionResultsCh)
 
 		extInput := s.getExtensionInput(extensionResultsCh)