优化结构

2026-08-20 23:02:06 +08:00
parent 63dbb4cfbc
commit e65cf0b1d0
338 changed files with 622 additions and 481 deletions
@@ -1,102 +0,0 @@
-package middleware
-
-import (
-	"crypto/rand"
-	"encoding/hex"
-	"strings"
-
-	"kra/internal/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
-}