43 lines
1.0 KiB
Go
43 lines
1.0 KiB
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"time"
|
|
|
|
"git.nwaifu.su/sergey/MyGoServer/internal/apiserver/logger"
|
|
rhttp "git.nwaifu.su/sergey/MyGoServer/pkg/http"
|
|
)
|
|
|
|
// LoggingMiddleware logs HTTP requests
|
|
func LoggingMiddleware(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
start := time.Now()
|
|
|
|
// Use the custom response writer to capture status code
|
|
rw := rhttp.NewResponseWriter(w)
|
|
|
|
next.ServeHTTP(rw, r)
|
|
|
|
// Log the request
|
|
logger := logger.GetLogger()
|
|
logger.WithFields(map[string]interface{}{
|
|
"method": r.Method,
|
|
"uri": r.RequestURI,
|
|
"remote_addr": r.RemoteAddr,
|
|
"user_agent": r.UserAgent(),
|
|
"status_code": rw.GetStatusCode(),
|
|
"duration": time.Since(start).String(),
|
|
"request_id": getRequestID(r.Context()),
|
|
}).Info("HTTP request")
|
|
})
|
|
}
|
|
|
|
// getRequestID retrieves request ID from context
|
|
func getRequestID(ctx context.Context) string {
|
|
if id, ok := ctx.Value("request_id").(string); ok {
|
|
return id
|
|
}
|
|
return ""
|
|
}
|