events/broadcaster Buffer val when writing to subscriber channel (#111)

Signed-off-by: joshvanl <me@joshvanl.dev>
dapr · Jan 10, 2025 · fb19570 · fb19570
1 parent 65ba378
commit fb19570
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/events/broadcaster/broadcaster.go b/events/broadcaster/broadcaster.go
@@ -91,7 +91,14 @@ func (b *Broadcaster[T]) subscribe(ctx context.Context, ch chan<- T) {
 				return
 			case <-b.closeCh:
 				return
-			case ch <- <-bufferedCh:
+			case val := <-bufferedCh:
+				select {
+				case <-ctx.Done():
+					return
+				case <-b.closeCh:
+					return
+				case ch <- val:
+				}
 			}
 		}
 	}()