fix: websocket close frame not response

ZiRunHua · Nov 14, 2024 · 468382f · 468382f
1 parent 4a4ed4b
commit 468382f
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/router/websocket/websocket.go b/router/websocket/websocket.go
@@ -18,8 +18,7 @@ func Use(handler func(conn *websocket.Conn, ctx *gin.Context) error) gin.Handler
 	return func(ctx *gin.Context) {
 		conn, err := upgrader.Upgrade(ctx.Writer, ctx.Request, nil)
 		if err != nil {
-			ctx.JSONP(500, "error")
-			return
+			panic(err)
 		}
 		conn.SetPingHandler(
 			func(message string) error {
@@ -33,11 +32,7 @@ func Use(handler func(conn *websocket.Conn, ctx *gin.Context) error) gin.Handler
 			},
 		)
 		conn.SetPongHandler(nil)
-		conn.SetCloseHandler(
-			func(code int, text string) error {
-				return nil
-			},
-		)
+		conn.SetCloseHandler(nil)
 		defer conn.Close()
 		err = handler(conn, ctx)
 		if err != nil {