72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package logger
|
|
|
|
import (
|
|
"log/slog"
|
|
"net/http"
|
|
"runtime/debug"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
func Middleware(next http.Handler, recordBody bool) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "GET" && r.URL.Path == "/healthz" {
|
|
next.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
slog.Error(
|
|
"panic",
|
|
"err", err,
|
|
"trace", string(debug.Stack()),
|
|
)
|
|
}
|
|
}()
|
|
start := time.Now()
|
|
path := r.URL.Path
|
|
query := r.URL.RawQuery
|
|
|
|
bw := newBodyWriter(w, recordBody)
|
|
|
|
next.ServeHTTP(bw, r)
|
|
|
|
end := time.Now()
|
|
requestAttributes := []slog.Attr{
|
|
slog.Time("time", start.UTC()),
|
|
slog.String("method", r.Method),
|
|
slog.String("host", r.Host),
|
|
slog.String("path", path),
|
|
slog.String("query", query),
|
|
slog.String("ip", r.RemoteAddr),
|
|
}
|
|
responseAttributes := []slog.Attr{
|
|
slog.Time("time", end.UTC()),
|
|
slog.Duration("latency", end.Sub(start)),
|
|
slog.Int("length", bw.bytes),
|
|
slog.Int("status", bw.status),
|
|
}
|
|
if recordBody {
|
|
responseAttributes = append(responseAttributes, slog.String("body", bw.body.String()))
|
|
}
|
|
attributes := []slog.Attr{
|
|
{
|
|
Key: "request",
|
|
Value: slog.GroupValue(requestAttributes...),
|
|
},
|
|
{
|
|
Key: "response",
|
|
Value: slog.GroupValue(responseAttributes...),
|
|
},
|
|
}
|
|
level := slog.LevelInfo
|
|
if bw.status >= http.StatusInternalServerError {
|
|
level = slog.LevelError
|
|
} else if bw.status >= http.StatusBadRequest && bw.status < http.StatusInternalServerError {
|
|
level = slog.LevelWarn
|
|
}
|
|
slog.LogAttrs(r.Context(), level, strconv.Itoa(bw.status)+": "+http.StatusText(bw.status), attributes...)
|
|
})
|
|
}
|