This commit is contained in:
Sakurasan
2024-10-06 02:08:27 +08:00
parent 2a0c18e51b
commit 1afaf3b385

View File

@@ -95,12 +95,15 @@ func forwardMessages(ctx context.Context, src, dst *websocket.Conn) error {
case <-ctx.Done(): case <-ctx.Done():
return ctx.Err() return ctx.Err()
default: default:
_, message, err := src.ReadMessage() messageType, message, err := src.ReadMessage()
if err != nil { if err != nil {
if websocket.IsCloseError(err, websocket.CloseNormalClosure, websocket.CloseGoingAway) {
return nil // 正常关闭,不报错
}
return err return err
} }
log.Println("Received message:", string(message)) log.Println("Received message:", string(message))
err = dst.WriteMessage(websocket.TextMessage, message) err = dst.WriteMessage(messageType, message)
if err != nil { if err != nil {
return err return err
} }