|
|
package RedisUtil
|
|
|
|
|
|
import (
|
|
|
"dsSso/Const/ErrorConst"
|
|
|
"dsSso/Utils/ConfigUtil"
|
|
|
"dsSso/Utils/LogUtil"
|
|
|
"fmt"
|
|
|
garybudRedis "github.com/garyburd/redigo/redis"
|
|
|
"github.com/go-redis/redis/v7"
|
|
|
"github.com/sirupsen/logrus"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
var (
|
|
|
// 定义redis链接池
|
|
|
RedisClient *redis.Client
|
|
|
//这个是为了使用OAuth2类库，引入的redis池，也是没办法之举~
|
|
|
Pool *garybudRedis.Pool
|
|
|
)
|
|
|
|
|
|
func init() {
|
|
|
// 从配置文件获取redis的ip以及db
|
|
|
var redisHost = ConfigUtil.RedisIp + ":" + ConfigUtil.RedisPort
|
|
|
Pool = &garybudRedis.Pool{
|
|
|
Dial: func() (garybudRedis.Conn, error) {
|
|
|
conn, err := garybudRedis.Dial("tcp", redisHost)
|
|
|
if err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
return conn, nil
|
|
|
},
|
|
|
}
|
|
|
//这个是以后项目中广泛使用的redis池
|
|
|
RedisClient = redis.NewClient(&redis.Options{
|
|
|
Addr: redisHost, // Redis地址
|
|
|
Password: "", // Redis账号
|
|
|
DB: 0, // Redis库
|
|
|
PoolSize: 100, // Redis连接池大小
|
|
|
MaxRetries: 3, // 最大重试次数
|
|
|
IdleTimeout: 10 * time.Second, // 空闲链接超时时间
|
|
|
})
|
|
|
_, err := RedisClient.Ping().Result()
|
|
|
if err == redis.Nil {
|
|
|
LogUtil.Error(ErrorConst.CreateRedisError, "Redis异常")
|
|
|
} else if err != nil {
|
|
|
LogUtil.Error(ErrorConst.CreateRedisError, err.Error())
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func FlushAll() {
|
|
|
RedisClient.FlushAll()
|
|
|
}
|
|
|
|
|
|
//======下面的代码由黄海增加于2020-02-18=============================================================
|
|
|
/**
|
|
|
功能：获取redis中的Hash缓存
|
|
|
作者：黄海
|
|
|
时间：2020-02-18
|
|
|
*/
|
|
|
func HGETALL(key string) map[string]interface{} {
|
|
|
_map, err := RedisClient.HGetAll(key).Result()
|
|
|
n := make(map[string]interface{})
|
|
|
for k, v := range _map {
|
|
|
n[k] = interface{}(v)
|
|
|
}
|
|
|
if err == nil {
|
|
|
return n
|
|
|
}
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：设置hash值
|
|
|
作者：黄海
|
|
|
时间：2020-02-18
|
|
|
*/
|
|
|
func HSET(key string, value interface{}) {
|
|
|
RedisClient.HSet(key, value)
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：获取hash值
|
|
|
作者：黄海
|
|
|
时间：2020-02-18
|
|
|
*/
|
|
|
func HGET(key string, field string) *redis.StringCmd {
|
|
|
return RedisClient.HGet(key, field)
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：设置hash值
|
|
|
作者：黄海
|
|
|
时间：2020-02-18
|
|
|
*/
|
|
|
func HMSET(key string, m map[string]interface{}) {
|
|
|
RedisClient.HMSet(key, m)
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：设置多个hash值
|
|
|
作者：黄海
|
|
|
时间：2020-02-18
|
|
|
*/
|
|
|
func HMGET(key string, fields ...string) map[string]interface{} {
|
|
|
resMap := make(map[string]interface{})
|
|
|
for _, field := range fields {
|
|
|
var result interface{}
|
|
|
val, err := RedisClient.HGet(key, fmt.Sprintf("%s", field)).Result()
|
|
|
if err == redis.Nil {
|
|
|
logrus.Errorln("Key Doesn't Exists:", field)
|
|
|
resMap[field] = result
|
|
|
} else if err != nil {
|
|
|
logrus.Errorln("Redis HMGet Error:", err)
|
|
|
resMap[field] = result
|
|
|
}
|
|
|
if val != "" {
|
|
|
resMap[field] = val
|
|
|
} else {
|
|
|
resMap[field] = result
|
|
|
}
|
|
|
}
|
|
|
return resMap
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：设置指定键值的过期时间
|
|
|
作者：黄海
|
|
|
时间：2020-02-18
|
|
|
*/
|
|
|
func EXPIRE(key string, ttl time.Duration) {
|
|
|
RedisClient.Expire(key, ttl)
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能:是否存在指定的键值
|
|
|
作者：黄海
|
|
|
时间：2020-03-18
|
|
|
*/
|
|
|
func EXISTS(key string) (int64, error) {
|
|
|
return RedisClient.Exists(key).Result()
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：设置指定键值的过期时间
|
|
|
作者：黄海
|
|
|
时间：2020-02-18
|
|
|
*/
|
|
|
func DEL(key string) {
|
|
|
RedisClient.Del(key)
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：设置一个STRING形式的缓存
|
|
|
作者：黄海
|
|
|
时间：2020-02-25
|
|
|
*/
|
|
|
func SET(key string, value string, ttl time.Duration) {
|
|
|
RedisClient.Set(key, value, ttl)
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：获取指定的KEY值
|
|
|
作者：黄海
|
|
|
时间：2020-02-25
|
|
|
*/
|
|
|
func GET(key string) (string, error) {
|
|
|
var r = RedisClient.Get(key)
|
|
|
if r.Err() != nil {
|
|
|
return "", r.Err()
|
|
|
} else {
|
|
|
return r.Result()
|
|
|
}
|
|
|
}
|
