public/myredis/myredis.go

package myredis

import (
	"time"

	"github.com/gomodule/redigo/redis"
)

// RedisDial 操作
type RedisDial interface {
	Destory()                                                                       // 析构
	GetRedisClient() redis.Conn                                                     // 获取一个原始的redis连接
	Ping() bool                                                                     // 判断是否能ping通
	Add(key interface{}, value interface{}, lifeSpan time.Duration) error           // 添加一个元素
	Value(key interface{}, value interface{}) error                                 // 获取一个value
	IsExist(key interface{}) bool                                                   // 判断是否存在
	Delete(key interface{}) error                                                   // 删除一个
	Clear() error                                                                   // 清空
	GetKeyS(key interface{}) ([]string, error)                                      // 查询所有key
	Close() (err error)                                                             // 关闭连接
	Do(commandName string, args ...interface{}) (reply interface{}, err error)      // 一次操作
	TryLock(key interface{}, value interface{}, lifeSpan time.Duration) (err error) //  试着加锁
	Unlock(key interface{}) (err error)                                             // 解锁
}

// DefaultConf ...
func DefaultConf() *MyRedis {
	if _default.conf == nil {
		InitDefaultRedis()
	}
	return _default
}

// InitDefaultRedis 初始化(必须要优先调用一次)
func InitDefaultRedis(ops ...Option) *MyRedis {
	var tmp = &redisOptions{isLog: true}
	for _, o := range ops {
		o.apply(tmp)
	}
	if len(tmp.addrs) == 0 {
		tmp.addrs = append(tmp.addrs, ":6379")
	}

	_default.mtx.Lock()
	defer _default.mtx.Unlock()
	_default.conf = tmp
	return _default
}

// InitRedis 初始化(必须要优先调用一次)
func InitRedis(ops ...Option) *MyRedis {
	var cnf = &MyRedis{}
	var tmp = &redisOptions{}
	for _, o := range ops {
		o.apply(tmp)
	}
	if len(tmp.addrs) == 0 {
		tmp.addrs = append(tmp.addrs, ":6379")
	}

	cnf.mtx.Lock()
	defer cnf.mtx.Unlock()
	cnf.conf = tmp
	return cnf
}

// NewRedis 初始化一个(InitDefaultRedis(需要优先调用)) groupName:分组名
func NewRedis(con *MyRedis) (dial RedisDial, err error) {
	if con == nil {
		con = DefaultConf()
	}
	con.once.Do(func() { // 创建连接
		ReDialRedis(con)
	})

	return con.dial, nil
}

// ReDialRedis 重新连接redis
func ReDialRedis(con *MyRedis) {
	if con.dial != nil { // 清理，关闭连接
		con.dial.Destory()
	}

	con.mtx.Lock()
	defer con.mtx.Unlock()

	if con.conf.maxIdle == 0 { // 创建连接池
		con.conf.maxIdle = 1
	}

	// 创建连接池
	con.dial = &redisConPool{
		base: base{MyRedis: con},
	}

	// 创建单个连接
	// con.dial = &redisConOlny{
	// 	base: base{MyRedis: con},
	// }
}