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():
return ctx.Err()
default:
_, message, err := src.ReadMessage()
messageType, message, err := src.ReadMessage()
if err != nil {
if websocket.IsCloseError(err, websocket.CloseNormalClosure, websocket.CloseGoingAway) {
return nil // 正常关闭,不报错
}
return err
}
log.Println("Received message:", string(message))
err = dst.WriteMessage(websocket.TextMessage, message)
err = dst.WriteMessage(messageType, message)
if err != nil {
return err
}