eta_api/utils/redis/standalone_redis.go

package redis

import (
	"context"
	"encoding/json"
	"errors"
	"fmt"
	"github.com/go-redis/redis/v8"
	"strconv"
	"time"
)

// StandaloneRedisClient
// @Description: 单机redis客户端
type StandaloneRedisClient struct {
	redisClient *redis.Client
}

func InitStandaloneRedis(config string) (standaloneRedis StandaloneRedisClient, err error) {
	var cf map[string]string
	err = json.Unmarshal([]byte(config), &cf)
	if err != nil {
		return
	}
	//if _, ok := cf["key"]; !ok {
	//	cf["key"] = DefaultKey
	//}

	if _, ok := cf["conn"]; !ok {
		err = errors.New("config has no conn key")
		return
	}

	// db库
	dbNum := 0
	// 如果指定了db库
	if _, ok := cf["dbNum"]; ok {
		dbNum, err = strconv.Atoi(cf["dbNum"])
		if err != nil {
			return
		}
	}

	// 密码
	if _, ok := cf["password"]; !ok {
		cf["password"] = ""
	}

	client := redis.NewClient(&redis.Options{
		Addr:     cf["conn"],
		Password: cf["password"],
		DB:       dbNum,
		//PoolSize: 10, //连接池最大socket连接数，默认为10倍CPU数， 10 * runtime.NumCPU（暂不配置）
	})

	_, err = client.Ping(context.TODO()).Result()
	if err != nil {
		err = errors.New("redis 链接失败:" + err.Error())
		return
	}

	standaloneRedis = StandaloneRedisClient{redisClient: client}

	return
}

// Get cache from redis.
func (rc StandaloneRedisClient) Get(key string) interface{} {
	data, err := rc.redisClient.Get(context.TODO(), key).Bytes()
	if err != nil {
		return nil
	}

	return data
}

func (rc StandaloneRedisClient) RedisBytes(key string) (data []byte, err error) {
	data, err = rc.redisClient.Get(context.TODO(), key).Bytes()

	return
}

func (rc StandaloneRedisClient) RedisString(key string) (data string, err error) {
	data, err = rc.redisClient.Get(context.TODO(), key).Result()

	return
}

func (rc StandaloneRedisClient) RedisInt(key string) (data int, err error) {
	data, err = rc.redisClient.Get(context.TODO(), key).Int()

	return
}

// Put put cache to redis.
func (rc StandaloneRedisClient) Put(key string, val interface{}, timeout time.Duration) error {
	var err error
	err = rc.redisClient.SetEX(context.TODO(), key, val, timeout).Err()
	if err != nil {
		return err
	}

	err = rc.redisClient.HSet(context.TODO(), DefaultKey, key, true).Err()

	return err
}

func (rc StandaloneRedisClient) SetNX(key string, val interface{}, timeout time.Duration) bool {
	result, err := rc.redisClient.SetEX(context.TODO(), key, val, timeout).Result()
	if err != nil || result != "OK" {
		return false
	}

	return true
}

// Delete delete cache in redis.
func (rc StandaloneRedisClient) Delete(key string) error {
	var err error

	err = rc.redisClient.Del(context.TODO(), key).Err()
	if err != nil {
		return err
	}

	err = rc.redisClient.HDel(context.TODO(), DefaultKey, key).Err()

	return err
}

// IsExist check cache's existence in redis.
func (rc StandaloneRedisClient) IsExist(key string) bool {
	result, err := rc.redisClient.Exists(context.TODO(), key).Result()
	if err != nil {
		return false
	}
	if result == 0 {
		err = rc.redisClient.HDel(context.TODO(), DefaultKey, key).Err()
		if err != nil {
			return false
		}
	}
	return true
}

// Put put cache to redis.
func (rc StandaloneRedisClient) LPush(key string, val interface{}) error {
	data, _ := json.Marshal(val)
	err := rc.redisClient.LPush(context.TODO(), key, data).Err()

	return err
}

func (rc StandaloneRedisClient) Brpop(key string, callback func([]byte)) {
	values, err := rc.redisClient.BRPop(context.TODO(), 1*time.Second, key).Result()
	if err != nil {
		return
	}
	if len(values) < 2 {
		fmt.Println("assert is wrong")
		return
	}

	callback([]byte(values[1]))
}

func (rc StandaloneRedisClient) GetRedisTTL(key string) time.Duration {
	value, err := rc.redisClient.TTL(context.TODO(), key).Result()
	if err != nil {
		return 0
	}

	return value

}

// Decr decrease counter in redis.
func (rc StandaloneRedisClient) Incrby(key string, num int) (interface{}, error) {
	return rc.redisClient.IncrBy(context.TODO(), key, int64(num)).Result()
}

// actually do the redis cmds
func (rc StandaloneRedisClient) Do(commandName string, args ...interface{}) (reply interface{}, err error) {
	newArgs := make([]interface{}, 0)
	newArgs = append(newArgs, commandName, args)
	return rc.redisClient.Do(context.TODO(), commandName, newArgs).Result()
}