Check context termination while writting into a channel.

libp2p · Sep 16, 2022 · 568a01c · 568a01c
1 parent 8826fba
commit 568a01c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/compsequential.go b/compsequential.go
@@ -219,7 +219,12 @@ func getChannelOrErrorSequential[T any](
 					if !ok {
 						break f
 					}
-					chanOut <- v
+					select {
+					case <-ctx.Done():
+						break f
+					case chanOut <- v:
+					}
+
 				}
 			}