优化代码位置

internal/app/middleware/audit.go

@@ -8,7 +8,6 @@ import (
 
 	"git.huangwc.com/pig/pig-farm-controller/internal/app/service/audit"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
-	"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
 	"github.com/gin-gonic/gin"
 )
 
@@ -30,14 +29,14 @@ func AuditLogMiddleware(auditService audit.Service) gin.HandlerFunc {
 		// --- 在这里，请求已经处理完毕 ---
 
 		// 从上下文中尝试获取由控制器设置的业务审计信息
-		actionType, exists := c.Get(repository.ContextAuditActionType)
+		actionType, exists := c.Get(models.ContextAuditActionType.String())
 		if !exists {
 			// 如果上下文中没有 actionType，说明此接口无需记录审计日志，直接返回
 			return
 		}
 
 		// 从 Gin Context 中获取用户对象
-		userCtx, userExists := c.Get(repository.ContextUserKey)
+		userCtx, userExists := c.Get(models.ContextUserKey.String())
 		var user *models.User
 		if userExists {
 			user, _ = userCtx.(*models.User)
@@ -51,11 +50,11 @@ func AuditLogMiddleware(auditService audit.Service) gin.HandlerFunc {
 		}
 
 		// 获取其他审计信息
-		description, _ := c.Get(repository.ContextAuditDescription)
-		targetResource, _ := c.Get(repository.ContextAuditTargetResource)
+		description, _ := c.Get(models.ContextAuditDescription.String())
+		targetResource, _ := c.Get(models.ContextAuditTargetResource.String())
 
 		// 默认操作状态为成功
-		status := repository.AuditStatusSuccess
+		status := models.AuditStatusSuccess
 		resultDetails := ""
 
 		// 尝试从捕获的响应体中解析平台响应
@@ -64,14 +63,14 @@ func AuditLogMiddleware(auditService audit.Service) gin.HandlerFunc {
 			// 如果解析成功，根据平台状态码判断操作是否失败
 			// 成功状态码范围是 2000-2999
 			if platformResponse.Code < 2000 || platformResponse.Code >= 3000 {
-				status = repository.AuditStatusFailed
+				status = models.AuditStatusFailed
 				resultDetails = platformResponse.Message
 			}
 		} else {
 			// 如果响应体不是预期的平台响应格式，或者解析失败，则记录原始HTTP状态码作为详情
 			// 并且如果HTTP状态码不是2xx，则标记为失败
 			if c.Writer.Status() < 200 || c.Writer.Status() >= 300 {
-				status = repository.AuditStatusFailed
+				status = models.AuditStatusFailed
 			}
 			resultDetails = "HTTP Status: " + strconv.Itoa(c.Writer.Status()) + ", Body Parse Error: " + err.Error()
 		}

internal/app/middleware/auth.go

@@ -6,6 +6,7 @@ import (
 	"strings"
 
 	"git.huangwc.com/pig/pig-farm-controller/internal/app/service/token"
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"
 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
@@ -52,7 +53,7 @@ func AuthMiddleware(tokenService token.TokenService, userRepo repository.UserRep
 		}
 
 		// 将完整的用户对象存储在 context 中，以便后续的处理函数使用
-		c.Set(repository.ContextUserKey, user)
+		c.Set(models.ContextUserKey.String(), user)
 
 		// 继续处理请求链中的下一个处理程序
 		c.Next()