优化代码位置

This commit is contained in:
2025-09-28 01:10:04 +08:00
parent d995498199
commit fdbea5b7e9
6 changed files with 39 additions and 31 deletions

View File

@@ -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()