From c8427c851d82d10c7066755f5a05b2d81cfe13d0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=AF=BB=E6=AC=A2?= Date: Mon, 28 Apr 2025 11:33:43 +0800 Subject: [PATCH] =?UTF-8?q?:refactor:=20=E6=9B=B4=E6=96=B0=E7=94=A8?= =?UTF-8?q?=E6=88=B7=E4=BF=A1=E6=81=AF=E8=8E=B7=E5=8F=96=E9=80=BB=E8=BE=91?= =?UTF-8?q?=EF=BC=8C=E4=BD=BF=E7=94=A8Locals=E5=AD=98=E5=82=A8=E7=94=A8?= =?UTF-8?q?=E6=88=B7ID=E5=B9=B6=E8=AE=B0=E5=BD=95=E9=94=99=E8=AF=AF?= =?UTF-8?q?=E6=97=A5=E5=BF=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/handler/robot.go | 4 +--- internal/middleware/auth.go | 6 +++++- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/internal/handler/robot.go b/internal/handler/robot.go index 9669808..9252aa9 100644 --- a/internal/handler/robot.go +++ b/internal/handler/robot.go @@ -18,12 +18,10 @@ import ( // ListRobots 列出所有机器人 func ListRobots(c *fiber.Ctx) error { - log.Debugf("登录用户Id: %+v", c.Get("userId")) - db := model.GetDB() var robots []model.Robot - if err := db.Find(&robots).Error; err != nil { + if err := db.Where("user_id = ?", c.Locals("userId")).Find(&robots).Error; err != nil { return fiber.NewError(fiber.StatusInternalServerError, "获取机器人列表失败") } diff --git a/internal/middleware/auth.go b/internal/middleware/auth.go index 2ceaf6c..2570c48 100644 --- a/internal/middleware/auth.go +++ b/internal/middleware/auth.go @@ -6,6 +6,7 @@ import ( "gitee.ltd/lxh/wechat-robot/internal/logto" "gitee.ltd/lxh/wechat-robot/internal/redis" "github.com/gofiber/fiber/v2" + "github.com/gofiber/fiber/v2/log" ) // IsAuthenticated @@ -54,7 +55,10 @@ func Authenticate() fiber.Handler { return c.Redirect("/error?error=Logto登录错误: " + err.Error()) } if userInfo, e := client.GetIdTokenClaims(); e == nil { - c.Set("userId", userInfo.Sub) + //log.Debugf("获取用户信息成功: %+v", userInfo) + c.Locals("userId", userInfo.Sub) + } else { + log.Errorf("获取用户信息失败: %v", e) } }