feat: 添加认证、日志、响应等基础中间件与服务

internal/middleware/logger.go

@@ -0,0 +1,83 @@
+package middleware
+
+import (
+	"EnGin/internal/global"
+	"fmt"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+func Logger() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		start := time.Now()
+		path := c.Request.URL.Path
+		raw := c.Request.URL.RawQuery
+
+		c.Next()
+
+		latency := time.Since(start)
+		clientIP := c.ClientIP()
+		method := c.Request.Method
+		statusCode := c.Writer.Status()
+		traceID := c.GetString("trace_id")
+
+		if raw != "" {
+			path = path + "?" + raw
+		}
+
+		// 状态码颜色
+		statusColor := getStatusColor(statusCode)
+		methodColor := getMethodColor(method)
+		resetColor := "\033[0m"
+
+		logMsg := fmt.Sprintf("%s %3d %s| %13v | %10s | %s%-5s%s %s | trace_id:%s",
+			statusColor, statusCode, resetColor,
+			latency,
+			clientIP,
+			methodColor, method, resetColor,
+			path,
+			traceID,
+		)
+
+		if statusCode >= 500 {
+			global.Log.Error(logMsg)
+		} else {
+			global.Log.Info(logMsg)
+		}
+	}
+}
+
+func getStatusColor(code int) string {
+	switch {
+	case code >= 200 && code < 300:
+		return "\033[97;42m" // 绿底白字
+	case code >= 300 && code < 400:
+		return "\033[90;47m" // 白底黑字
+	case code >= 400 && code < 500:
+		return "\033[97;43m" // 黄底白字
+	default:
+		return "\033[97;41m" // 红底白字
+	}
+}
+
+func getMethodColor(method string) string {
+	switch method {
+	case "GET":
+		return "\033[97;44m" // 蓝底白字
+	case "POST":
+		return "\033[97;46m" // 青底白字
+	case "PUT":
+		return "\033[97;43m" // 黄底白字
+	case "DELETE":
+		return "\033[97;41m" // 红底白字
+	case "PATCH":
+		return "\033[97;42m" // 绿底白字
+	case "HEAD":
+		return "\033[97;45m" // 紫底白字
+	case "OPTIONS":
+		return "\033[90;47m" // 白底黑字
+	default:
+		return "\033[0m"
+	}
+}