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) } }