40 lines
869 B
Go
40 lines
869 B
Go
package middleware
|
|
|
|
import (
|
|
"base-framework/pkg/logx"
|
|
"base-framework/pkg/router"
|
|
"base-framework/pkg/routinex"
|
|
"base-framework/pkg/utils/uid"
|
|
"time"
|
|
)
|
|
|
|
// TrackTime 记录开始时间到现在时间的耗时
|
|
func TrackTime(pre time.Time) time.Duration {
|
|
elapsed := time.Since(pre)
|
|
|
|
return elapsed
|
|
}
|
|
|
|
func Logger() router.HandlerFunc {
|
|
return func(c *router.Context) {
|
|
start := time.Now() // 记录开始时间
|
|
|
|
// 生成 trackID 并存入 routinex
|
|
trackID := uid.NextID()
|
|
routinex.Set(logx.TrackID, trackID)
|
|
|
|
c.Next() // 执行业务逻辑
|
|
|
|
// 请求结束后计算耗时
|
|
elapsed := TrackTime(start)
|
|
|
|
logx.Logger.Info("请求结束",
|
|
logx.String("logType", "request"),
|
|
logx.Uint64("trackID", trackID),
|
|
logx.String("path", c.Request.URL.Path),
|
|
logx.String("method", c.Request.Method),
|
|
logx.Duration("elapsed", elapsed),
|
|
)
|
|
}
|
|
}
|