fix: break from loop when context cancelled

filecoin-project · Sep 19, 2023 · eab31ac · eab31ac
1 parent a25cd9b
commit eab31ac
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/aggregateeventrecorder/aggregateeventrecorder.go b/pkg/aggregateeventrecorder/aggregateeventrecorder.go
@@ -128,6 +128,7 @@ func (a *aggregateEventRecorder) ingestEvents() {
 	for {
 		select {
 		case <-a.ctx.Done():
+			return
 
 		// Read incoming data
 		case event := <-a.ingestChan:
@@ -297,6 +298,7 @@ func (a *aggregateEventRecorder) postEvents() {
 		select {
 		case <-a.ctx.Done():
 			return
+
 		case batchedData := <-a.postChan:
 			byts, err := json.Marshal(batchedEvents{batchedData})
 			if err != nil {