diff --git a/internal/net/gphttp/middleware/middleware.go b/internal/net/gphttp/middleware/middleware.go index fdb4b0a0..21d0eb4c 100644 --- a/internal/net/gphttp/middleware/middleware.go +++ b/internal/net/gphttp/middleware/middleware.go @@ -8,6 +8,8 @@ import ( "sort" "strings" + "github.com/rs/zerolog" + "github.com/rs/zerolog/log" "github.com/yusing/go-proxy/internal/gperr" gphttp "github.com/yusing/go-proxy/internal/net/gphttp" "github.com/yusing/go-proxy/internal/net/gphttp/reverseproxy" @@ -189,6 +191,18 @@ func (m *Middleware) ServeHTTP(next http.HandlerFunc, w http.ResponseWriter, r * next(w, r) } +func (m *Middleware) LogWarn(req *http.Request) *zerolog.Event { + return log.Warn().Str("middleware", m.name). + Str("host", req.Host). + Str("path", req.URL.Path) +} + +func (m *Middleware) LogError(req *http.Request) *zerolog.Event { + return log.Error().Str("middleware", m.name). + Str("host", req.Host). + Str("path", req.URL.Path) +} + func PatchReverseProxy(rp *ReverseProxy, middlewaresMap map[string]OptionsRaw) (err gperr.Error) { var middlewares []*Middleware middlewares, err = compileMiddlewares(middlewaresMap)