fix: correctly handle normal ws closure

jpts · Jun 22, 2023 · 806d623 · 806d623
1 parent 31451d7
commit 806d623
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/cmd/websocket.go b/cmd/websocket.go
@@ -71,14 +71,15 @@ func (d *WebsocketRoundTripper) WsCallback(ws *websocket.Conn) error {
 	}()
 
 	for err := range errChan {
-		if e, ok := err.(*websocket.CloseError); ok {
-			klog.V(4).Infof("Closing websocket connection with error code %d, err: %s", e.Code, err)
-		}
 		if websocket.IsCloseError(err, websocket.CloseNormalClosure) {
 			return nil
 		} else if errors.Is(err, io.EOF) {
+			klog.V(4).Info("Closing websocket connection with EOF")
 			return nil
 		}
+		if e, ok := err.(*websocket.CloseError); ok {
+			klog.V(4).Infof("Closing websocket connection with error code %d, err: %s", e.Code, err)
+		}
 		return err
 	}
 	return nil