103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package middleware
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"strings"
|
|
|
|
"kra/pkg/logging"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func randomHex(bytes int) string {
|
|
value := make([]byte, bytes)
|
|
_, _ = rand.Read(value)
|
|
return hex.EncodeToString(value)
|
|
}
|
|
|
|
func RequestMeta() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
requestID := c.GetHeader("X-Request-Id")
|
|
if !saneHeaderID(requestID) {
|
|
requestID = uuid.NewString()
|
|
}
|
|
traceID, parentSpanID := "", ""
|
|
if upstreamTraceID, upstreamSpanID, ok := parseTraceparent(c.GetHeader("traceparent")); ok {
|
|
traceID, parentSpanID = upstreamTraceID, upstreamSpanID
|
|
} else if candidate := c.GetHeader("X-Trace-Id"); saneHeaderID(candidate) {
|
|
traceID = candidate
|
|
}
|
|
if traceID == "" {
|
|
traceID = randomHex(16)
|
|
}
|
|
spanID := randomHex(8)
|
|
c.Header("X-Request-Id", requestID)
|
|
c.Header("X-Trace-Id", traceID)
|
|
if validTraceID(traceID) {
|
|
c.Header("traceparent", "00-"+traceID+"-"+spanID+"-01")
|
|
}
|
|
c.Set("request_id", requestID)
|
|
c.Set("trace_id", traceID)
|
|
c.Set("span_id", spanID)
|
|
c.Set("parent_span_id", parentSpanID)
|
|
c.Request = c.Request.WithContext(logging.WithContextFields(c.Request.Context(), &logging.ContextFields{
|
|
RequestID: requestID, TraceID: traceID, SpanID: spanID, ParentSpanID: parentSpanID,
|
|
DeviceID: c.GetHeader("X-Device-Id"), ClientIP: c.ClientIP(),
|
|
HTTPMethod: c.Request.Method, HTTPPath: c.Request.URL.Path,
|
|
}))
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
func parseTraceparent(value string) (traceID, parentSpanID string, ok bool) {
|
|
parts := strings.Split(value, "-")
|
|
if len(parts) < 4 {
|
|
return "", "", false
|
|
}
|
|
version, traceID, parentSpanID, flags := parts[0], parts[1], parts[2], parts[3]
|
|
if !lowerHex(version, 2) || version == "ff" || version == "00" && len(parts) != 4 ||
|
|
!validTraceID(traceID) || !lowerHex(parentSpanID, 16) || allZero(parentSpanID) || !lowerHex(flags, 2) {
|
|
return "", "", false
|
|
}
|
|
return traceID, parentSpanID, true
|
|
}
|
|
|
|
func saneHeaderID(value string) bool {
|
|
if value == "" || len(value) > 64 {
|
|
return false
|
|
}
|
|
for index := 0; index < len(value); index++ {
|
|
if value[index] <= 0x20 || value[index] > 0x7e {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func validTraceID(value string) bool { return lowerHex(value, 32) && !allZero(value) }
|
|
|
|
func lowerHex(value string, length int) bool {
|
|
if len(value) != length {
|
|
return false
|
|
}
|
|
for index := 0; index < len(value); index++ {
|
|
if current := value[index]; current < '0' || current > '9' {
|
|
if current < 'a' || current > 'f' {
|
|
return false
|
|
}
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func allZero(value string) bool {
|
|
for index := 0; index < len(value); index++ {
|
|
if value[index] != '0' {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|