redisc update

utils/redis/redis_conn.go

@@ -14,6 +14,18 @@ import (
 	"time"
 )
 
+// 自定义错误
+const (
+	ErrRedisConnNil = ErrRedis("redis conn nil")
+)
+
+type ErrRedis string
+
+func (err ErrRedis) Error() string {
+	return string(err)
+}
+
+// Redis连接池适配器
 type Hub struct {
 	LOG     func() *logger.Logger
 	pool    *redis.Pool
@@ -21,6 +33,7 @@ type Hub struct {
 	once    sync.Once
 }
 
+// 获取Redis连接池适配器的新实例
 func NewHub() *Hub {
 	viper.SetDefault("redis.max_idle", 300)
 	viper.SetDefault("redis.max_active", 1000)
@@ -32,6 +45,7 @@ func NewHub() *Hub {
 	}
 }
 
+// 创建连接池
 func (h *Hub) createPool(addr string, opts ...redis.DialOption) (*redis.Pool, error) {
 	maxIdle := viper.GetInt("redis.max_idle")
 	maxActive := viper.GetInt("redis.max_active")
@@ -56,11 +70,12 @@ func (h *Hub) createPool(addr string, opts ...redis.DialOption) (*redis.Pool, er
 	}, nil
 }
 
+// 建立与Redis的连接
 func (h *Hub) conn() error {
 	dialOpts := []redis.DialOption{
 		redis.DialConnectTimeout(5 * time.Second),
-		redis.DialReadTimeout(500 * time.Millisecond),
-		redis.DialWriteTimeout(500 * time.Millisecond),
+		//redis.DialReadTimeout(5 * time.Second),
+		//redis.DialWriteTimeout(5 * time.Second),
 	}
 
 	nodes := viper.GetStringSlice("redis.nodes")
@@ -81,11 +96,11 @@ func (h *Hub) conn() error {
 		pool, _ = h.createPool(host, dialOpts...)
 		h.pool = pool
 	}
-
-	return nil
+	return h.ping()
 }
 
-func (h *Hub) test() error {
+// 连接测试
+func (h *Hub) ping() error {
 	// 连接测试
 	conn, err := h.Get()
 	if nil != err {
@@ -107,6 +122,7 @@ func (h *Hub) test() error {
 	return nil
 }
 
+// 监听系统退出信号量，自动关闭Redis连接
 func (h *Hub) closeWait() {
 	go func(h *Hub) {
 		defer utils.DefaultGoroutineRecover(nil, `Redis连接池关闭`)
@@ -130,6 +146,7 @@ func (h *Hub) closeWait() {
 	}(h)
 }
 
+// 关闭Redis连接
 func (h *Hub) Close() error {
 	var err error
 	if nil != h.cluster {
@@ -140,6 +157,7 @@ func (h *Hub) Close() error {
 	return err
 }
 
+// 执行Redis命令
 func (h *Hub) Do(cmd string, args ...interface{}) (interface{}, error) {
 	conn, err := h.Get()
 	if nil != err {
@@ -160,18 +178,10 @@ func (h *Hub) Do(cmd string, args ...interface{}) (interface{}, error) {
 	}
 }
 
-const (
-	ErrRedisConnNil = ErrRedis("redis conn nil")
-)
-
-type ErrRedis string
-
-func (err ErrRedis) Error() string {
-	return string(err)
-}
-
+// 从连接池中获取一个连接
 func (h *Hub) Get() (redis.Conn, error) {
 	var err error
+	// 延迟一次性初始化Redis连接池
 	h.once.Do(func() {
 		e := h.conn()
 		if nil != e {