[KS-347] Start tracking registerTrigger goroutine (#13688)

* [KS-347] Start tracking registerTrigger goroutine * Check for channel closed * Check for channel closed --------- Co-authored-by: Bolek <[email protected]>
smartcontractkit · Jun 27, 2024 · 7a3ba8e · 7a3ba8e
1 parent 96f332c
commit 7a3ba8e
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/core/services/workflows/engine.go b/core/services/workflows/engine.go
@@ -355,9 +355,25 @@ func (e *Engine) registerTrigger(ctx context.Context, t *triggerCapability, trig
 			}}
 	}
 
+	e.wg.Add(1)
 	go func() {
-		for event := range eventsCh {
-			e.triggerEvents <- event
+		defer e.wg.Done()
+
+		for {
+			select {
+			case <-e.stopCh:
+				return
+			case event, isOpen := <-eventsCh:
+				if !isOpen {
+					return
+				}
+
+				select {
+				case <-e.stopCh:
+					return
+				case e.triggerEvents <- event:
+				}
+			}
 		}
 	}()