chore: recoverer middleware skip write header on upgrade connection (#…

…795)
go-chi · Feb 20, 2023 · 51068a7 · 51068a7
1 parent 883aa98
commit 51068a7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)
+				}
 			}
 		}()