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 // @Description: 根据key获取数据(其实是返回的字节编码) // @receiver rc // @param key // @return interface{} func (rc *StandaloneRedisClient) Get(key string) interface{} { data, err := rc.redisClient.Get(context.TODO(), key).Bytes() if err != nil { return nil } return data } // RedisBytes // @Description: 根据key获取字节编码数据 // @receiver rc // @param key // @return data // @return err func (rc *StandaloneRedisClient) RedisBytes(key string) (data []byte, err error) { data, err = rc.redisClient.Get(context.TODO(), key).Bytes() return } // RedisString // @Description: 根据key获取字符串数据 // @receiver rc // @param key // @return data // @return err func (rc *StandaloneRedisClient) RedisString(key string) (data string, err error) { data, err = rc.redisClient.Get(context.TODO(), key).Result() return } // RedisInt // @Description: 根据key获取int数据 // @receiver rc // @param key // @return data // @return err func (rc *StandaloneRedisClient) RedisInt(key string) (data int, err error) { data, err = rc.redisClient.Get(context.TODO(), key).Int() return } // Put // @Description: put一个数据到redis // @receiver rc // @param key // @param val // @param timeout // @return error 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 } // SetNX // @Description: 设置一个会过期时间的值 // @receiver rc // @param key // @param val // @param timeout // @return bool func (rc *StandaloneRedisClient) SetNX(key string, val interface{}, timeout time.Duration) bool { result, err := rc.redisClient.SetNX(context.TODO(), key, val, timeout).Result() if err != nil { return false } return result } // Delete // @Description: 删除redis中的键值对 // @receiver rc // @param key // @return error 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 // @Description: 根据key判断是否写入缓存中 // @receiver rc // @param key // @return bool func (rc *StandaloneRedisClient) IsExist(key string) bool { result, err := rc.redisClient.Exists(context.TODO(), key).Result() if err != nil { return false } if result == 0 { _ = rc.redisClient.HDel(context.TODO(), DefaultKey, key).Err() return false } return true } // LPush // @Description: 写入list // @receiver rc // @param key // @param val // @return error func (rc *StandaloneRedisClient) LPush(key string, val interface{}) error { data, _ := json.Marshal(val) err := rc.redisClient.LPush(context.TODO(), key, data).Err() return err } // Brpop // @Description: 从list中读取 // @receiver rc // @param key // @param callback 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])) } // GetRedisTTL // @Description: 获取key的过期时间 // @receiver rc // @param key // @return time.Duration func (rc *StandaloneRedisClient) GetRedisTTL(key string) time.Duration { value, err := rc.redisClient.TTL(context.TODO(), key).Result() if err != nil { return 0 } return value } // Incrby // @Description: 设置自增值 // @receiver rc // @param key // @param num // @return interface{} // @return error func (rc *StandaloneRedisClient) Incrby(key string, num int) (interface{}, error) { return rc.redisClient.IncrBy(context.TODO(), key, int64(num)).Result() } // Do // @Description: cmd执行redis命令 // @receiver rc // @param commandName // @param args // @return reply // @return err func (rc *StandaloneRedisClient) Do(commandName string, args ...interface{}) (reply interface{}, err error) { newArgs := []interface{}{commandName} newArgs = append(newArgs, args...) return rc.redisClient.Do(context.TODO(), newArgs...).Result() }