74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package chizap
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
"go.uber.org/zap"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
const timeFormat = "02/Jan/2006:15:04:05 -0700"
|
|
|
|
func Logger(logger *zap.Logger) func(next http.Handler) http.Handler {
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
path := r.URL.Path
|
|
if r.URL.RawQuery != "" {
|
|
path += "?" + r.URL.RawQuery
|
|
}
|
|
start := time.Now()
|
|
ww := middleware.NewWrapResponseWriter(w, r.ProtoMajor)
|
|
next.ServeHTTP(ww, r)
|
|
stop := time.Since(start)
|
|
statusCode := ww.Status()
|
|
clientUserAgent := r.UserAgent()
|
|
if clientUserAgent == "" {
|
|
clientUserAgent = "-"
|
|
}
|
|
referer := r.Referer()
|
|
if referer == "" {
|
|
referer = "-"
|
|
}
|
|
dataLength := ww.BytesWritten()
|
|
if dataLength < 0 {
|
|
dataLength = 0
|
|
}
|
|
|
|
fields := []zap.Field{
|
|
zap.Int("statusCode", statusCode),
|
|
zap.Int64("duration", stop.Nanoseconds()),
|
|
zap.String("durationPretty", stop.String()),
|
|
zap.String("clientIP", r.RemoteAddr),
|
|
zap.String("method", r.Method),
|
|
zap.String("path", path),
|
|
zap.String("proto", r.Proto),
|
|
zap.String("referer", referer),
|
|
zap.Int("dataLength", dataLength),
|
|
zap.String("usetAgent", clientUserAgent),
|
|
}
|
|
|
|
msg := fmt.Sprintf(
|
|
`%s - - [%s] "%s %s %s" %d %d "%s" "%s" %s`,
|
|
r.RemoteAddr,
|
|
time.Now().Format(timeFormat),
|
|
r.Method,
|
|
path,
|
|
r.Proto,
|
|
statusCode,
|
|
dataLength,
|
|
referer,
|
|
clientUserAgent,
|
|
stop.String(),
|
|
)
|
|
if statusCode >= http.StatusInternalServerError {
|
|
logger.Error(msg, fields...)
|
|
} else if statusCode >= http.StatusBadRequest {
|
|
logger.Warn(msg, fields...)
|
|
} else {
|
|
logger.Info(msg, fields...)
|
|
}
|
|
})
|
|
}
|
|
}
|