33 lines
895 B
Go
33 lines
895 B
Go
package signature
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/zeromicro/go-zero/core/stores/redis"
|
||
)
|
||
|
||
// NonceStore 防重放存储接口
|
||
type NonceStore interface {
|
||
// SetIfNotExists 返回 (true, nil) 表示 key 已存在（重放攻击）
|
||
SetIfNotExists(ctx context.Context, appId, nonce string, ttlSeconds int64) (bool, error)
|
||
}
|
||
|
||
type RedisNonceStore struct {
|
||
client *redis.Redis
|
||
}
|
||
|
||
func NewRedisNonceStore(client *redis.Redis) *RedisNonceStore {
|
||
return &RedisNonceStore{client: client}
|
||
}
|
||
|
||
func (s *RedisNonceStore) SetIfNotExists(_ context.Context, appId, nonce string, ttlSeconds int64) (bool, error) {
|
||
key := "sig:nonce:" + appId + ":" + nonce
|
||
// go-zero redis.Redis.SetNX 返回 true 表示设置成功（首次）
|
||
ok, err := s.client.SetnxEx(key, "1", int(ttlSeconds))
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
// SetnxEx ok=true 首次，ok=false 已存在（重放）
|
||
return !ok, nil
|
||
}
|