77 lines
1.4 KiB
Go
77 lines
1.4 KiB
Go
package middleware
|
|
|
|
import (
|
|
"allapp/db"
|
|
"allapp/db/repo"
|
|
"allapp/utils/errorx"
|
|
"allapp/utils/redisx"
|
|
"allapp/utils/responsex"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/gofiber/fiber/v3"
|
|
)
|
|
|
|
const CtxSpaceIDKey = "space_id"
|
|
|
|
func SpaceGuard() fiber.Handler {
|
|
return func(c fiber.Ctx) error {
|
|
userID, err := GetUserID(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
spaceIDStr := strings.TrimSpace(c.Get("SPACE_ID"))
|
|
spaceID, err := strconv.ParseInt(spaceIDStr, 10, 64)
|
|
if err != nil || spaceID <= 0 {
|
|
return responsex.SpaceError(c, "请选择空间")
|
|
}
|
|
|
|
// 1 先查 Redis
|
|
inSpace, err := redisx.SpaceIsMember(spaceID, userID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if inSpace {
|
|
c.Locals(CtxSpaceIDKey, spaceID)
|
|
return c.Next()
|
|
}
|
|
|
|
// 2 Redis miss → 查数据库
|
|
role, err := db.Queries.GetUserSpaceRole(c.Context(), repo.GetUserSpaceRoleParams{
|
|
SpaceID: spaceID,
|
|
UserID: userID,
|
|
})
|
|
if err != nil {
|
|
if errorx.IsNotFound(err) {
|
|
return responsex.SpaceError(c, "未加入空间")
|
|
}
|
|
return err
|
|
}
|
|
|
|
// 3 写回 Redis（重建缓存）
|
|
_ = redisx.SpaceAddOrUpdateMember(spaceID, userID, int(role))
|
|
c.Locals(CtxSpaceIDKey, spaceID)
|
|
|
|
// 4 放行
|
|
return c.Next()
|
|
}
|
|
}
|
|
|
|
func GetSpaceID(c fiber.Ctx) (int64, error) {
|
|
v := c.Locals(CtxSpaceIDKey)
|
|
|
|
if v == nil {
|
|
return 0, responsex.SpaceError(c, "空间无效")
|
|
}
|
|
|
|
spaceID, ok := v.(int64)
|
|
|
|
if !ok {
|
|
return 0, responsex.SpaceError(c, "空间无效")
|
|
}
|
|
|
|
return spaceID, nil
|
|
}
|