Don't panic if the accept fails

leap-fish · Apr 6, 2024 · 501fe70 · 501fe70
1 parent f8c6d35
commit 501fe70
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/wrapws/server.go b/wrapws/server.go
@@ -41,7 +41,10 @@ func (ws *WebSocketServer) Serve(addr string) error {
 func (ws *WebSocketServer) acceptFunc(w http.ResponseWriter, req *http.Request) {
 	conn, err := websocket.Accept(w, req, ws.options)
 	if err != nil {
-		ws.handler.OnError(req.Context(), nil, err)
+		return
+	}
+
+	if conn == nil {
 		return
 	}