49a3c43741
- 新增 Access Token + Refresh Token 双令牌认证 - 支持 Token 自动刷新和轮转 - 添加登出和撤销所有会话接口 - 前端实现无感刷新和主动刷新定时器
74 lines
3.0 KiB
Go
74 lines
3.0 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"time"
|
||
)
|
||
|
||
// ErrRefreshTokenNotFound is returned when a refresh token is not found in cache.
|
||
// This is used to abstract away the underlying cache implementation (e.g., redis.Nil).
|
||
var ErrRefreshTokenNotFound = errors.New("refresh token not found")
|
||
|
||
// RefreshTokenData 存储在Redis中的Refresh Token数据
|
||
type RefreshTokenData struct {
|
||
UserID int64 `json:"user_id"`
|
||
TokenVersion int64 `json:"token_version"` // 用于检测密码更改后的Token失效
|
||
FamilyID string `json:"family_id"` // Token家族ID，用于防重放攻击
|
||
CreatedAt time.Time `json:"created_at"`
|
||
ExpiresAt time.Time `json:"expires_at"`
|
||
}
|
||
|
||
// RefreshTokenCache 管理Refresh Token的Redis缓存
|
||
// 用于JWT Token刷新机制，支持Token轮转和防重放攻击
|
||
//
|
||
// Key 格式:
|
||
// - refresh_token:{token_hash} -> RefreshTokenData (JSON)
|
||
// - user_refresh_tokens:{user_id} -> Set<token_hash>
|
||
// - token_family:{family_id} -> Set<token_hash>
|
||
type RefreshTokenCache interface {
|
||
// StoreRefreshToken 存储Refresh Token
|
||
// tokenHash: Token的SHA256哈希值（不存储原始Token）
|
||
// data: Token关联的数据
|
||
// ttl: Token过期时间
|
||
StoreRefreshToken(ctx context.Context, tokenHash string, data *RefreshTokenData, ttl time.Duration) error
|
||
|
||
// GetRefreshToken 获取Refresh Token数据
|
||
// 返回 (data, nil) 如果Token存在
|
||
// 返回 (nil, ErrRefreshTokenNotFound) 如果Token不存在
|
||
// 返回 (nil, err) 如果发生其他错误
|
||
GetRefreshToken(ctx context.Context, tokenHash string) (*RefreshTokenData, error)
|
||
|
||
// DeleteRefreshToken 删除单个Refresh Token
|
||
// 用于Token轮转时使旧Token失效
|
||
DeleteRefreshToken(ctx context.Context, tokenHash string) error
|
||
|
||
// DeleteUserRefreshTokens 删除用户的所有Refresh Token
|
||
// 用于密码更改或用户主动登出所有设备
|
||
DeleteUserRefreshTokens(ctx context.Context, userID int64) error
|
||
|
||
// DeleteTokenFamily 删除整个Token家族
|
||
// 用于检测到Token重放攻击时，撤销整个会话链
|
||
DeleteTokenFamily(ctx context.Context, familyID string) error
|
||
|
||
// AddToUserTokenSet 将Token添加到用户的Token集合
|
||
// 用于跟踪用户的所有活跃Refresh Token
|
||
AddToUserTokenSet(ctx context.Context, userID int64, tokenHash string, ttl time.Duration) error
|
||
|
||
// AddToFamilyTokenSet 将Token添加到家族Token集合
|
||
// 用于跟踪同一登录会话的所有Token
|
||
AddToFamilyTokenSet(ctx context.Context, familyID string, tokenHash string, ttl time.Duration) error
|
||
|
||
// GetUserTokenHashes 获取用户的所有Token哈希
|
||
// 用于批量删除用户Token
|
||
GetUserTokenHashes(ctx context.Context, userID int64) ([]string, error)
|
||
|
||
// GetFamilyTokenHashes 获取家族的所有Token哈希
|
||
// 用于批量删除家族Token
|
||
GetFamilyTokenHashes(ctx context.Context, familyID string) ([]string, error)
|
||
|
||
// IsTokenInFamily 检查Token是否属于指定家族
|
||
// 用于验证Token家族关系
|
||
IsTokenInFamily(ctx context.Context, familyID string, tokenHash string) (bool, error)
|
||
}
|