Add middleware

This commit is contained in:
Pablu23
2024-11-05 16:44:41 +01:00
parent c3066da440
commit 91d128dbeb
7 changed files with 162 additions and 95 deletions

26
middleware/logging.go Normal file
View File

@@ -0,0 +1,26 @@
package middleware
import (
"net/http"
"time"
"github.com/rs/zerolog/log"
"github.com/urfave/negroni"
)
func RequestLogger(next http.Handler) http.Handler {
log.Info().Msg("Enabling Logging")
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
start := time.Now()
lrw := negroni.NewResponseWriter(w)
next.ServeHTTP(lrw, r)
duration := time.Since(start)
if duration.Milliseconds() > 500 {
log.Warn().Str("host", r.Host).Str("uri", r.RequestURI).Str("method", r.Method).Int("status", lrw.Status()).Int("size", lrw.Size()).Str("duration", duration.String()).Msg("Slow Request")
} else {
log.Info().Str("host", r.Host).Str("uri", r.RequestURI).Str("method", r.Method).Int("status", lrw.Status()).Int("size", lrw.Size()).Str("duration", duration.String()).Msg("Received Request")
}
})
}