81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
|
package middleware
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"message-nest/pkg/logging"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
status200 = 42
|
||
|
|
status404 = 43
|
||
|
|
status500 = 41
|
||
|
|
|
||
|
|
methodGET = 44
|
||
|
|
)
|
||
|
|
|
||
|
|
func formatDuration(duration time.Duration) string {
|
||
|
|
seconds := duration.Seconds()
|
||
|
|
switch {
|
||
|
|
case seconds >= 1:
|
||
|
|
return fmt.Sprintf("%.2fs", seconds)
|
||
|
|
default:
|
||
|
|
return fmt.Sprintf("%.3fms", float64(duration.Milliseconds()))
|
||
|
|
//default:
|
||
|
|
// return fmt.Sprintf("%dns", duration.Nanoseconds())
|
||
|
|
//case duration.Milliseconds() >= 1:
|
||
|
|
// return fmt.Sprintf("%.2fms", float64(duration.Milliseconds()))
|
||
|
|
//default:
|
||
|
|
// return fmt.Sprintf("%dns", duration.Nanoseconds())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func LogMiddleware() gin.HandlerFunc {
|
||
|
|
return func(c *gin.Context) {
|
||
|
|
start := time.Now()
|
||
|
|
path := c.Request.URL.Path
|
||
|
|
raw := c.Request.URL.RawQuery
|
||
|
|
|
||
|
|
// Process request
|
||
|
|
c.Next()
|
||
|
|
|
||
|
|
// Log only when path is not being skipped
|
||
|
|
|
||
|
|
// Stop timer
|
||
|
|
end := time.Now()
|
||
|
|
timeSub := end.Sub(start)
|
||
|
|
dr := formatDuration(timeSub)
|
||
|
|
clientIP := c.ClientIP()
|
||
|
|
method := c.Request.Method
|
||
|
|
statusCode := c.Writer.Status()
|
||
|
|
//bodySize := c.Writer.Size()
|
||
|
|
if raw != "" {
|
||
|
|
path = path + "?" + raw
|
||
|
|
}
|
||
|
|
|
||
|
|
var statusColor string
|
||
|
|
switch statusCode {
|
||
|
|
case 200:
|
||
|
|
statusColor = fmt.Sprintf("\033[%dm %d \033[0m", status200, statusCode)
|
||
|
|
case 500:
|
||
|
|
statusColor = fmt.Sprintf("\033[%dm %d \033[0m", status500, statusCode)
|
||
|
|
default:
|
||
|
|
statusColor = fmt.Sprintf("\033[%dm %d \033[0m", status404, statusCode)
|
||
|
|
}
|
||
|
|
|
||
|
|
var methodColor string
|
||
|
|
methodColor = fmt.Sprintf("\033[%dm %s \033[0m", methodGET, method)
|
||
|
|
|
||
|
|
logging.Logger.Infof("[GIN]|%s|%s|%s|%s| %s",
|
||
|
|
//start.Format("2006-01-02 15:04:06"),
|
||
|
|
statusColor,
|
||
|
|
dr,
|
||
|
|
clientIP,
|
||
|
|
methodColor,
|
||
|
|
path,
|
||
|
|
)
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|