Refactor Close method in websocketPeer to handle duplicate close calls

gammazero · Apr 2, 2024 · 9609aab · 9609aab
1 parent 2798a81
commit 9609aab
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/transport/websocketpeer.go b/transport/websocketpeer.go
@@ -233,6 +233,12 @@ func (w *websocketPeer) IsLocal() bool { return false }
 //
 // *** Do not call Send after calling Close. ***
 func (w *websocketPeer) Close() {
+	select {
+	case <-w.closed:
+		return
+	default:
+	}
+
 	// Tell sendHandler to exit and discard any queued messages.  Do not close
 	// wr channel in case there are incoming messages during close.
 	w.cancelSender()