fix: remove deadlock on .stop re-entrance

Author: 0x416e746f6e

proxy/websocket_pump.go

@@ -41,7 +41,7 @@ func newWebsocketPump(
 		frontend: frontend,
 		backend:  backend,
 		logger:   logger,
-		done:     make(chan struct{}),
+		done:     make(chan struct{}, 1),
 	}
 }
 
@@ -93,10 +93,19 @@ func (p *websocketPump) run() error {
 }
 
 func (p *websocketPump) stop() error {
-	p.done <- struct{}{}
+	if !p.active.Load() {
+		return nil // already stopped
+	}
 
 	errs := make([]error, 0)
 
+	select {
+	case p.done <- struct{}{}:
+		// no-op
+	default:
+		errs = append(errs, errors.New("double-closing the pump"))
+	}
+
 	countdown := 60
 	for p.active.Load() && countdown > 0 {
 		time.Sleep(time.Second)
@@ -147,7 +156,14 @@ func (p *websocketPump) pumpMessages(
 
 				msgType, bytes, err := from.ReadMessage()
 				if err != nil {
-					failure <- err
+					select {
+					case failure <- err:
+						// no-op
+					default:
+						l.Warn("Dropping websocket pump failure b/c the channel is full",
+							zap.Error(err),
+						)
+					}
 					continue
 				}