42 lines
883 B
Go
42 lines
883 B
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
var revokedTokens sync.Map
|
|
|
|
func RevokeToken(ctx context.Context, tokenID string, ttl time.Duration) error {
|
|
if tokenID == "" || ttl <= 0 {
|
|
return nil
|
|
}
|
|
if redisClient != nil {
|
|
return redisClient.client.Set(ctx, "jwt:revoked:"+tokenID, "1", ttl).Err()
|
|
}
|
|
revokedTokens.Store(tokenID, time.Now().Add(ttl))
|
|
return nil
|
|
}
|
|
|
|
func IsTokenRevoked(ctx context.Context, tokenID string) (bool, error) {
|
|
if tokenID == "" {
|
|
return false, fmt.Errorf("token 缺少 jti")
|
|
}
|
|
if redisClient != nil {
|
|
count, err := redisClient.client.Exists(ctx, "jwt:revoked:"+tokenID).Result()
|
|
return count > 0, err
|
|
}
|
|
value, ok := revokedTokens.Load(tokenID)
|
|
if !ok {
|
|
return false, nil
|
|
}
|
|
expiresAt := value.(time.Time)
|
|
if time.Now().After(expiresAt) {
|
|
revokedTokens.Delete(tokenID)
|
|
return false, nil
|
|
}
|
|
return true, nil
|
|
}
|