signal when we are done

lestrrat-go · Sep 29, 2024 · a47356d · a47356d
1 parent 9de8a68
commit a47356d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/proxysink/proxysink.go b/proxysink/proxysink.go
@@ -30,13 +30,15 @@ func New[T any](b Backend[T]) *Proxy[T] {
 }
 
 func (p *Proxy[T]) Run(ctx context.Context) {
+	defer p.cond.Broadcast()
 	go p.controlloop(ctx)
 	go p.flushloop(ctx)
 
 	<-ctx.Done()
 }
 
 func (p *Proxy[T]) controlloop(ctx context.Context) {
+	defer p.cond.Broadcast()
 	for {
 		select {
 		case <-ctx.Done():