From eb52dd73d17b235e4c1ff616703edfd74adabeff Mon Sep 17 00:00:00 2001 From: "Gustavo \"Guz\" L. de Mello" Date: Mon, 22 Jul 2024 12:25:12 -0300 Subject: [PATCH] feat(router): check error of middlewared response writer --- internals/router/router.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/internals/router/router.go b/internals/router/router.go index 9e71779..0948076 100644 --- a/internals/router/router.go +++ b/internals/router/router.go @@ -1,6 +1,7 @@ package router import ( + "fmt" "log" "net/http" "strings" @@ -43,7 +44,10 @@ func (router Router) wrapMiddleares(ms []Middleware, h http.HandlerFunc) http.Ha return func(w http.ResponseWriter, r *http.Request) { mw := NewMiddlewaredResponse(w) fh(mw, r) - _, _ = mw.ReallyWriteHeader() + _, err := mw.ReallyWriteHeader() + if err != nil { + _, _ = w.Write([]byte(fmt.Sprintf("Error while trying to write to body:\n%s", err.Error()))) + } } }