24 lines
686 B
Go
24 lines
686 B
Go
package middleware
|
||
|
||
import (
|
||
"modelRT/util"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// StartTrace define func of set trace info from request header
|
||
func StartTrace() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
traceID := c.Request.Header.Get("traceid")
|
||
pSpanID := c.Request.Header.Get("spanid")
|
||
spanID := util.GenerateSpanID(c.Request.RemoteAddr)
|
||
if traceID == "" { // 如果traceId 为空，证明是链路的发端，把它设置成此次的spanId，发端的spanId是root spanId
|
||
traceID = spanID // trace 标识整个请求的链路, span则标识链路中的不同服务
|
||
}
|
||
c.Set("traceid", traceID)
|
||
c.Set("spanid", spanID)
|
||
c.Set("pspanid", pSpanID)
|
||
c.Next()
|
||
}
|
||
}
|