58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package logging
|
|
|
|
import (
|
|
"acc-server-manager/local/utl/logging"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
type RequestLoggingMiddleware struct {
|
|
infoLogger *logging.InfoLogger
|
|
}
|
|
|
|
func NewRequestLoggingMiddleware() *RequestLoggingMiddleware {
|
|
return &RequestLoggingMiddleware{
|
|
infoLogger: logging.GetInfoLogger(),
|
|
}
|
|
}
|
|
|
|
func (rlm *RequestLoggingMiddleware) Handler() fiber.Handler {
|
|
return func(c *fiber.Ctx) error {
|
|
start := time.Now()
|
|
|
|
userAgent := c.Get("User-Agent")
|
|
if userAgent == "" {
|
|
userAgent = "Unknown"
|
|
}
|
|
|
|
rlm.infoLogger.LogRequest(c.Method(), c.OriginalURL(), userAgent)
|
|
|
|
err := c.Next()
|
|
|
|
duration := time.Since(start)
|
|
|
|
statusCode := c.Response().StatusCode()
|
|
rlm.infoLogger.LogResponse(c.Method(), c.OriginalURL(), statusCode, duration.String())
|
|
|
|
if err != nil {
|
|
logging.ErrorWithContext("REQUEST_MIDDLEWARE", "Request failed: %v", err)
|
|
}
|
|
|
|
return err
|
|
}
|
|
}
|
|
|
|
var globalRequestLoggingMiddleware *RequestLoggingMiddleware
|
|
|
|
func GetRequestLoggingMiddleware() *RequestLoggingMiddleware {
|
|
if globalRequestLoggingMiddleware == nil {
|
|
globalRequestLoggingMiddleware = NewRequestLoggingMiddleware()
|
|
}
|
|
return globalRequestLoggingMiddleware
|
|
}
|
|
|
|
func Handler() fiber.Handler {
|
|
return GetRequestLoggingMiddleware().Handler()
|
|
}
|