93 lines
2.0 KiB
Go
93 lines
2.0 KiB
Go
|
package cache
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/fonchain_enterprise/utils/zap_log"
|
|||
|
|
"strconv"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"dubbo.apache.org/dubbo-go/v3/common/logger"
|
|||
|
|
"github.com/go-redis/redis"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// RedisClient Redis缓存客户端单例
|
|||
|
|
var (
|
|||
|
|
RedisClient *redis.Client
|
|||
|
|
)
|
|||
|
|
var deleteScript = `
|
|||
|
|
if redis.call("get",KEYS[1]) == ARGV[1] then
|
|||
|
|
return redis.call("del",KEYS[1])
|
|||
|
|
else
|
|||
|
|
return 0
|
|||
|
|
end
|
|||
|
|
`
|
|||
|
|
|
|||
|
|
type RedisConfig struct {
|
|||
|
|
RedisDB string
|
|||
|
|
RedisAddr string
|
|||
|
|
RedisPw string
|
|||
|
|
RedisDbName string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//LoadRedis 在中间件中初始化redis链接
|
|||
|
|
func LoadRedis(configEnv RedisConfig) {
|
|||
|
|
db, _ := strconv.ParseUint(configEnv.RedisDbName, 10, 64)
|
|||
|
|
client := redis.NewClient(&redis.Options{
|
|||
|
|
Addr: configEnv.RedisAddr,
|
|||
|
|
Password: configEnv.RedisPw,
|
|||
|
|
DB: int(db),
|
|||
|
|
})
|
|||
|
|
_, err := client.Ping().Result()
|
|||
|
|
if err != nil {
|
|||
|
|
logger.Info(err)
|
|||
|
|
panic(err)
|
|||
|
|
}
|
|||
|
|
RedisClient = client
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//LockConcurrency 锁死并发
|
|||
|
|
func LockConcurrency(key string, value string, ms time.Duration) (bool, error) {
|
|||
|
|
fmt.Println(ms)
|
|||
|
|
setNx := RedisClient.SetNX(key, value, ms)
|
|||
|
|
|
|||
|
|
result, err := setNx.Result()
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
zap_log.Error(fmt.Sprintf("锁死加解密的时候，redis 失效,err%s,%v", key, err))
|
|||
|
|
//panic(err)
|
|||
|
|
return false, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if result {
|
|||
|
|
zap_log.Info(fmt.Sprintf("新增key存在，设置成功%s", key))
|
|||
|
|
} else {
|
|||
|
|
zap_log.Info(fmt.Sprintf("新增key已存在，设置失败%s", key))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
zap_log.Info(fmt.Sprintf("日志%v%v%s", result, err, key))
|
|||
|
|
|
|||
|
|
if setNx.Val() == false {
|
|||
|
|
return false, errors.New("已经被人锁定")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return true, nil
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Release(key, value string) error {
|
|||
|
|
result, err := RedisClient.Eval(deleteScript, []string{key}, value).Result()
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 判断Lua脚本执行的结果
|
|||
|
|
if result == int64(1) {
|
|||
|
|
fmt.Println("Lock released.")
|
|||
|
|
return nil
|
|||
|
|
} else {
|
|||
|
|
fmt.Println("Failed to release lock, or lock was not held by this client.")
|
|||
|
|
return errors.New("释放失败也可能是超时自动解锁了,请您刷新")
|
|||
|
|
}
|
|||
|
|
}
|