diff --git a/v1/middleware/middleware.go b/v1/middleware/middleware.go index 38a6ea3..7a2a7e8 100644 --- a/v1/middleware/middleware.go +++ b/v1/middleware/middleware.go @@ -5,6 +5,7 @@ import ( mcontext "github.com/abelith/gomax/v1/context" "github.com/abelith/gomax/v1/core" merrors "github.com/abelith/gomax/v1/errors" + "github.com/rs/zerolog" "github.com/rs/zerolog/log" ) @@ -40,10 +41,10 @@ func MwErrorResponder(defaultMsg string) core.Middleware { } } -func MwLoggerInjector() core.Middleware { +func MwLoggerInjector(logger zerolog.Logger) core.Middleware { return func(next core.HandlerFunc) core.HandlerFunc { return func(c *mcontext.Context) error { - ctx := log.Logger.With().Int64("user_id", c.Update().UserID). + ctx := logger.With().Int64("user_id", c.Update().UserID). Logger().WithContext(c.Context()) c.SetContext(ctx) return next(c)