From 09bda69ee5877ea39dcc87412853c5a57913aa6e Mon Sep 17 00:00:00 2001
From: fatelei <fatelei@gmail.com>
Date: Wed, 15 Feb 2023 13:32:15 +0800
Subject: [PATCH] chore: recoverer middleware skip write header on upgrade
 connection

---
 middleware/recoverer.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/middleware/recoverer.go b/middleware/recoverer.go
index 8fa69bad..612e155a 100644
--- a/middleware/recoverer.go
+++ b/middleware/recoverer.go
@@ -36,7 +36,9 @@ func Recoverer(next http.Handler) http.Handler {
 					PrintPrettyStack(rvr)
 				}
 
-				w.WriteHeader(http.StatusInternalServerError)
+				if r.Header.Get("Connection") != "Upgrade" {
+					w.WriteHeader(http.StatusInternalServerError)
+				}
 			}
 		}()