restore originalWriter in case of panic inside echoHandlerFuncWrapper…

….ServeHTTP method
labstack · Apr 28, 2021 · 48fa624 · 48fa624
1 parent 6430665
commit 48fa624
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/middleware/timeout.go b/middleware/timeout.go
@@ -92,6 +92,15 @@ func (t echoHandlerFuncWrapper) ServeHTTP(rw http.ResponseWriter, r *http.Reques
 	originalWriter := t.ctx.Response().Writer
 	t.ctx.Response().Writer = rw
 
+	// in case of panic we restore original writer and call panic again
+	// so it could be handled with global middleware Recover()
+	defer func() {
+		if err := recover(); err != nil {
+			t.ctx.Response().Writer = originalWriter
+			panic(err)
+		}
+	}()
+
 	err := t.handler(t.ctx)
 	if ctxErr := r.Context().Err(); ctxErr == context.DeadlineExceeded {
 		if err != nil && t.errHandler != nil {