package redisCli import ( "context" "git.hilo.cn/hilo-common/myerr" "git.hilo.cn/hilo-common/mylogrus" "github.com/go-redis/redis/v8" "time" ) //这个用户避免多个服务器并发问题。 func SetNX(key string, value interface{}, expiration time.Duration, callBack func()) { flag, err := RedisClient.SetNX(context.Background(), key, value, expiration).Result() if err != nil { mylogrus.MyLog.Errorf("key:%v lock start setNx err: %v", key, err) } if !flag { mylogrus.MyLog.Infof("key:%v lock setNx has lock", key) return } mylogrus.MyLog.Infof("key:%v lock setNx begin", key) callBack() //执行结束之后,移除key //RedisClient.Del(context.Background(), key) mylogrus.MyLog.Infof("key:%v lock setNx end", key) } func Lock(key string, expiration time.Duration) bool { flag, err := RedisClient.SetNX(context.Background(), key, 1, expiration).Result() if err != nil { return false } if !flag { return false } return true } // 尝试获取锁 func TryLock(key string, expiration time.Duration) bool { deadline := time.Now().Add(expiration) for { if time.Now().After(deadline) { return false // 超时无法获取锁 } if Lock(key, expiration) { return true } time.Sleep(time.Millisecond) } } // 删除锁 func DelLock(key string) { RedisClient.Del(context.Background(), key) } func GetCacheInt64(key string) (int64, error) { data, err := RedisClient.Get(context.Background(), key).Int64() if err != nil && err != redis.Nil { return 0, err } return data, nil } func IncrBy(key string, num int64) (int64, error) { resNum, err := RedisClient.IncrBy(context.Background(), key, num).Result() if err != nil { return 0, err } return resNum, nil } func IncrNumExpire(key string, num int64, expiration time.Duration) (int64, error) { times, err := IncrBy(key, num) if err != nil { return 0, err } ttl, err := RedisClient.TTL(context.Background(), key).Result() if err != nil { return 0, err } if ttl == -1 { RedisClient.Expire(context.Background(), key, expiration) } return times, nil } func DelCache(key string) error { err := RedisClient.Del(context.Background(), key).Err() if err != nil { mylogrus.MyLog.Errorf("DelCache key:%s, err:%s", key, err) return err } return nil } func DecrBy(key string, num int64) (int64, error) { resNum, err := RedisClient.DecrBy(context.Background(), key, num).Result() if err != nil { return 0, err } return resNum, nil } func DecrNumExpire(key string, num int64, expiration time.Duration) (int64, error) { times, err := DecrBy(key, num) if err != nil { return 0, err } ttl, err := RedisClient.TTL(context.Background(), key).Result() if err != nil { return 0, err } if ttl == -1 { RedisClient.Expire(context.Background(), key, expiration) } return times, nil } func DecrNum(key string, num int64) (int64, error) { resNum, err := RedisClient.DecrBy(context.Background(), key, num).Result() if err != nil { return 0, myerr.WrapErr(err) } return resNum, nil }