54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
|
package redis
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/qit-team/snow-core/redis"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// AcquireLock 尝试获取分布式锁
|
|||
|
|
func AcquireLock(key string, expired time.Duration) (bool, error) {
|
|||
|
|
ctx := context.Background()
|
|||
|
|
client := redis.GetRedis()
|
|||
|
|
// 设置锁的值，这里可以是一个随机的UUID，用于检测是否是同一个客户端持有的锁锁
|
|||
|
|
lockValue := fmt.Sprintf("%d", time.Now().UnixNano())
|
|||
|
|
result := client.SetNX(ctx, key, lockValue, expired)
|
|||
|
|
return result.Val(), result.Err()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Del(key string) error {
|
|||
|
|
client := redis.GetRedis()
|
|||
|
|
result := client.Del(context.Background(), key)
|
|||
|
|
return result.Err()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Set(key string, value interface{}, expired time.Duration) error {
|
|||
|
|
client := redis.GetRedis()
|
|||
|
|
return client.Set(context.Background(), key, value, expired).Err()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Get(key string) (string, error) {
|
|||
|
|
client := redis.GetRedis()
|
|||
|
|
result := client.Get(context.Background(), key)
|
|||
|
|
return result.Result()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Decr(key string) (int64, error) {
|
|||
|
|
client := redis.GetRedis()
|
|||
|
|
result := client.Decr(context.Background(), key)
|
|||
|
|
return result.Result()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Incr(key string) (int64, error) {
|
|||
|
|
client := redis.GetRedis()
|
|||
|
|
result := client.Incr(context.Background(), key)
|
|||
|
|
return result.Result()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func SetNX(key string, value interface{}, expired time.Duration) (bool, error) {
|
|||
|
|
client := redis.GetRedis()
|
|||
|
|
result := client.SetNX(context.Background(), key, value, expired)
|
|||
|
|
return result.Result()
|
|||
|
|
}
|