file-server/pkg/middleware/mw_trace.go

package middleware

import (
	"github.com/gogf/gf/net/ghttp"
	"gxt-file-server/pkg/gplus"
	"gxt-file-server/pkg/utils"
)

func TraceIdMiddleware(skippers ...SkipperFunc) ghttp.HandlerFunc {
	return func(r *ghttp.Request) {
		if len(skippers) > 0 && skippers[0](r) {
			r.Middleware.Next()
			return
		}
		if r.GetCtxVar(gplus.TraceIDKey).String() == "" {
			r.SetCtxVar(gplus.TraceIDKey, utils.NewTraceID())
		}
		r.Middleware.Next()
	}
}