fix: should break the loop

hertz-contrib · Jan 4, 2024 · 039247d · 039247d
1 parent f728214
commit 039247d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ws_reverse_proxy.go b/ws_reverse_proxy.go
@@ -137,9 +137,12 @@ func (w *WSReverseProxy) ServeHTTP(ctx context.Context, c *app.RequestContext) {
 
 			var ce *websocket.CloseError
 			var hzce *hzws.CloseError
-			if !errors.As(err, &ce) || !errors.As(err, &hzce) {
+			if !errors.As(err, &ce) && !errors.As(err, &hzce) {
 				hlog.CtxErrorf(ctx, errMsg, err)
+				continue
 			}
+
+			break
 		}
 	}); err != nil {
 		hlog.CtxErrorf(ctx, "can not upgrade to websocket: %v", err)