优化结构

internal/server/middleware/rate_limit.go

@@ -16,6 +16,10 @@ func SecurityRateLimit(settings *service.SecurityService) gin.HandlerFunc {
 			c.Next()
 			return
 		}
+		if settings == nil {
+			c.Next()
+			return
+		}
 		config, err := settings.CurrentSecurity(c.Request.Context())
 		if err != nil || config == nil {
 			c.AbortWithStatusJSON(http.StatusServiceUnavailable, gin.H{"code": CodeError, "msg": "安全服务暂不可用"})
@@ -29,7 +33,11 @@ func SecurityRateLimit(settings *service.SecurityService) gin.HandlerFunc {
 		if window < 1 {
 			window = 60
 		}
-		key := "KRA_SecLimit" + c.ClientIP() + c.FullPath()
+		route := c.FullPath()
+		if route == "" {
+			route = path
+		}
+		key := "KRA_SecLimit:" + c.ClientIP() + ":" + route
 		count, cacheErr := settings.IncrementRateLimit(c.Request.Context(), key, time.Duration(window)*time.Second)
 		if cacheErr != nil {
 			c.AbortWithStatusJSON(http.StatusServiceUnavailable, gin.H{"code": CodeError, "msg": "安全服务暂不可用"})