任务三

internal/biz/system/jwt_blacklist.go

@@ -0,0 +1,60 @@
+package system
+
+import "context"
+
+// JwtBlacklist JWT黑名单实体
+type JwtBlacklist struct {
+	ID  uint
+	Jwt string
+}
+
+// JwtBlacklistRepo JWT黑名单仓储接口
+type JwtBlacklistRepo interface {
+	Create(ctx context.Context, jwt string) error
+	IsInBlacklist(ctx context.Context, jwt string) bool
+	LoadAll(ctx context.Context) ([]string, error)
+}
+
+// JwtBlacklistUsecase JWT黑名单用例
+type JwtBlacklistUsecase struct {
+	repo  JwtBlacklistRepo
+	cache map[string]struct{} // 内存缓存
+}
+
+// NewJwtBlacklistUsecase 创建JWT黑名单用例
+func NewJwtBlacklistUsecase(repo JwtBlacklistRepo) *JwtBlacklistUsecase {
+	uc := &JwtBlacklistUsecase{
+		repo:  repo,
+		cache: make(map[string]struct{}),
+	}
+	// 加载黑名单到缓存
+	uc.LoadAll(context.Background())
+	return uc
+}
+
+// JoinBlacklist 加入黑名单
+func (uc *JwtBlacklistUsecase) JoinBlacklist(ctx context.Context, jwt string) error {
+	if err := uc.repo.Create(ctx, jwt); err != nil {
+		return err
+	}
+	uc.cache[jwt] = struct{}{}
+	return nil
+}
+
+// IsInBlacklist 检查是否在黑名单中
+func (uc *JwtBlacklistUsecase) IsInBlacklist(jwt string) bool {
+	_, ok := uc.cache[jwt]
+	return ok
+}
+
+// LoadAll 加载所有黑名单到缓存
+func (uc *JwtBlacklistUsecase) LoadAll(ctx context.Context) error {
+	jwts, err := uc.repo.LoadAll(ctx)
+	if err != nil {
+		return err
+	}
+	for _, jwt := range jwts {
+		uc.cache[jwt] = struct{}{}
+	}
+	return nil
+}