Fix proxy listener not being closed when the server is stopped

geniusrabbit · Sep 12, 2024 · d1ef2e7 · d1ef2e7
1 parent 38985ad
commit d1ef2e7
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/gochan/go.go b/gochan/go.go
@@ -67,8 +67,13 @@ func (p *Proxy) write(ctx context.Context, msg any) error {
 // Listen starts processing queue
 func (p *Proxy) Listen(ctx context.Context) error {
 	for msg := range p.pool {
-		if err := p.ProcessMessage(msg); err != nil {
-			return err
+		select {
+		case <-ctx.Done():
+			return ctx.Err()
+		default:
+			if err := p.ProcessMessage(msg); err != nil {
+				return err
+			}
 		}
 	}
 	return nil