Go-redis使用

文章首发于：clawhub.club

---

项目中用到了redis中间件，所以找了个go版的redis客户端，简单封装后使用。
使用的go组件的github地址:
https://github.com/go-redis/redis

简单使用

• 支持单机与集群
• 使用go.uber.org/zap日志组件
• github.com/magiconair/properties 配置文件组件
  我把例子代码上传到了github:
  https://github.com/ClawHub/go-study.git

redis_cmd.go

package redis

import (
	"github.com/go-redis/redis"
	"time"
)

//set
func Set(key string, value interface{}, expiration time.Duration) *redis.StatusCmd {
	if cfg.Cluster {
		return redisClusterClient.Set(key, value, expiration)
	} else {
		return client.Set(key, value, expiration)
	}
}

//get
func Get(key string) *redis.StringCmd {
	if cfg.Cluster {
		return redisClusterClient.Get(key)
	} else {
		return client.Get(key)
	}
}

//订阅
func Subscribe(channels ...string) *redis.PubSub {
	if cfg.Cluster {
		return redisClusterClient.Subscribe(channels...)
	} else {
		return client.Subscribe(channels...)
	}
}

//发布
func Publish(channel string, message interface{}) *redis.IntCmd {
	if cfg.Cluster {
		return redisClusterClient.Publish(channel, message)
	} else {
		return client.Publish(channel, message)
	}
}

redis_client.go

package redis

import (
	"github.com/go-redis/redis"
	"go-study/src/document"
	"go-study/src/log"
	"go.uber.org/zap"
)

//redis单机客户端
var client *redis.Client

//redis集群客户端
var redisClusterClient *redis.ClusterClient

//redis配置
var cfg configRedis

func init() {

	//用了配置文件读取
	if err := document.Properties.Decode(&cfg); err != nil {
		log.LmdbLogger.Error("read document fail", zap.Error(err))
	}

	//判断是否为集群配置
	if cfg.Cluster {
		//ClusterClient是一个Redis集群客户机，表示一个由0个或多个底层连接组成的池。它对于多个goroutine的并发使用是安全的。
		redisClusterClient = redis.NewClusterClient(&redis.ClusterOptions{
			Password: cfg.Password,
			Addrs:    cfg.Addrs,
		})
		//Ping
		ping, err := redisClusterClient.Ping().Result()
		log.LmdbLogger.Info("Redis Ping", zap.String("ping", ping), zap.Error(err))

	} else {
		//Redis客户端，由零个或多个基础连接组成的池。它对于多个goroutine的并发使用是安全的。
		//更多参数参考Options结构体
		client = redis.NewClient(&redis.Options{
			Addr:     cfg.Addr,
			Password: cfg.Password, // no password set
			DB:       cfg.DB,       // use default DB
		})
		//Ping
		pong, err := client.Ping().Result()
		log.LmdbLogger.Info("Redis Ping", zap.String("pong", pong), zap.Error(err))
	}

}

//redis配置结构体
type configRedis struct {
	Addr     string   `properties:"redis.addr"`
	Password string   `properties:"redis.password"`
	DB       int      `properties:"redis.db,default=0"`
	Cluster  bool     `properties:"redis.cluster,default=false"`
	Addrs    []string `properties:"redis.addrs,default=localhost:2181;localhost:2182"`
}

demo_redis.go

package redis

import (
	"github.com/go-redis/redis"
	"go-study/src/log"
	"go.uber.org/zap"
)

func DemoRedis() {
	log.LmdbLogger.Info("------redis---------")
	err := client.Set("key", "value", 0).Err()
	if err != nil {
		log.LmdbLogger.Error("client.Set fail", zap.Error(err))
	}

	val, err := Get("key").Result()
	if err != nil {
		log.LmdbLogger.Error("client.Get fail", zap.Error(err))
	}
	log.LmdbLogger.Info("client.Get", zap.String("val", val))

	val2, err := Get("key2").Result()
	if err == redis.Nil {
		log.LmdbLogger.Info("key2 does not exist")
	} else if err != nil {
		log.LmdbLogger.Error("client.Get fail", zap.Error(err))
	} else {
		log.LmdbLogger.Info("client.Get", zap.String("val2", val2))
	}
	//订阅
	sub := Subscribe("channel")
	//新协程
	go func() {
		for {
			//发布
			pub := Publish("channel", "message")
			if pub.Err() != nil {
				log.LmdbLogger.Error("Publish err", zap.String("message", "message"), zap.Error(pub.Err()))
			} else {
				log.LmdbLogger.Info("Publish msg", zap.String("message", "message"))
			}
		}
	}()

	//从订阅获取信息，获取一次则程序结束
	msg, err := sub.ReceiveMessage()
	if err != nil {
		log.LmdbLogger.Error("Subscribe err", zap.Error(err))
	} else {
		log.LmdbLogger.Info("Subscribe msg", zap.String("msg", msg.String()))
	}

}