redis_util/redisloginutil.go

package redis_util

import (
	"log"
	"strings"
	"time"

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

// RedisLoginPool 用于登陆的Redis连接池
var RedisLoginPool *redisLogin.Pool

// InitRedisLogin 初始化用于登陆的Redis连接池
func InitRedisLogin(addrs string) {
	addr := strings.Split(addrs, ",")
	for _, v := range addr {
		saddr := strings.Split(v, "=")
		if saddr[0] == "login" {
			RedisLoginPool = &redisLogin.Pool{MaxActive: 10, MaxIdle: 5,
				IdleTimeout: time.Duration(10) * time.Second, Dial: func() (redisLogin.Conn, error) {
					c, err := redisLogin.Dial("tcp", saddr[1])
					if err != nil {
						return nil, err
					}
					return c, nil
				}}
		}
	}

}

// SetLoginVal 设置K，V值 用于登陆的Redis
func SetLoginVal(key, value string) {
	conn := RedisLoginPool.Get()
	defer conn.Close()
	conn.Do("PUBLISH", key, value)
}

// GetLoginVal 获取置Key，并透传到入参处理函数 用于登陆的Redis
func GetLoginVal(key string, wxFunc func(wxParams []string) bool) {
	for {
		defer catch()
	L:
		for {
			conn := RedisLoginPool.Get()
			defer conn.Close()
			if conn.Err() == nil {
				psc := redisLogin.PubSubConn{Conn: conn}
				if err := psc.Subscribe(redisLogin.Args{}.AddFlat(key)...); err != nil {
					log.Println(err)
				}
				for {
					msg := psc.Receive()
					//					go func(msg interface{}) {
					switch n := msg.(type) {
					case error:
						log.Println("wxlogin err", msg)
						break L
					case redisLogin.Message:
						res := string(n.Data)
						param := strings.Split(res, ",")
						go wxFunc(param)
					}
					//					}(msg)
				}
			}
			time.Sleep(2 * time.Second)
		}
		time.Sleep(1 * time.Second)
	}
}