up

mirrors2 · Oct 5, 2024 · 1afaf3b · 1afaf3b
1 parent 2a0c18e
commit 1afaf3b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pkg/openai/realtime.go b/pkg/openai/realtime.go
@@ -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
 			}