util.go

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
}