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),
		)
	}
}