69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package cache
|
|
|
|
import (
|
|
"servicebase/pkg/log"
|
|
"errors"
|
|
"time"
|
|
)
|
|
|
|
func Lock(key string) (e error) {
|
|
var success bool
|
|
TryMax := 6
|
|
redisClient := GetCommonRedisInstance().RedisClient
|
|
for i := 0; i < TryMax; i++ {
|
|
success, e = redisClient.SetNX(Ctx(), key, time.Now(), time.Millisecond*1000*3).Result()
|
|
if e != nil {
|
|
log.Info("Lock GetCommonRedisInstance error: " + e.Error())
|
|
}
|
|
if success {
|
|
return nil
|
|
}
|
|
time.Sleep(time.Millisecond * 50)
|
|
}
|
|
return errors.New("操作太频繁了~")
|
|
}
|
|
|
|
func Free(key string) {
|
|
client := GetCommonRedisInstance().RedisClient
|
|
_ = client.Del(Ctx(), key).Err()
|
|
}
|
|
|
|
const (
|
|
TryMax = int(30)
|
|
TryDuration = time.Millisecond * 13
|
|
LockTimeout = time.Millisecond * 700
|
|
)
|
|
|
|
func RedisLockGetTime(key string, duration time.Duration) (e error) {
|
|
redisClient := GetCommonRedisInstance().RedisClient
|
|
success, e := redisClient.SetNX(Ctx(), key, "1", duration).Result()
|
|
if !success {
|
|
e = errors.New("(LOCK_FAILED)系统繁忙，请稍后再试！")
|
|
}
|
|
return
|
|
}
|
|
|
|
func RedisLockGet(key string) (e error) {
|
|
redisClient := GetCommonRedisInstance().RedisClient
|
|
success := false
|
|
for i := 0; i < TryMax; i++ {
|
|
success, e = redisClient.SetNX(Ctx(), key, time.Now(), LockTimeout).Result()
|
|
if nil != e {
|
|
return
|
|
}
|
|
if success {
|
|
break
|
|
}
|
|
time.Sleep(TryDuration)
|
|
}
|
|
if !success {
|
|
e = errors.New("(LOCK_FAILED)系统繁忙，请稍后再试！")
|
|
}
|
|
return
|
|
}
|
|
|
|
func RedisLockDel(key string) {
|
|
redisClient := GetCommonRedisInstance().RedisClient
|
|
_ = redisClient.Del(Ctx(), key).Err()
|
|
}
|