redis.go

package gredis

import (
	"fmt"
	"github.com/gomodule/redigo/redis"
	"go_sku_server/pkg/config"
	"time"
)

type ichuntRedis struct {
	RedisList map[string]*redis.Pool
}

var ichuntRedis_ = &ichuntRedis{}

func Conn(connection string) redis.Conn {

	return ichuntRedis_.RedisList[connection].Get()
}

var writeConn, readConn *redis.Pool

func Setup() (err error) {
	ichuntRedis_.RedisList = make(map[string]*redis.Pool, 0)
	RedisDatabaseMap := config.BuildRedisConfigs()
	for redisKey, redisConfig := range RedisDatabaseMap {
		ichuntRedis_.RedisList[redisKey], err = getConn(redisConfig.Host, redisConfig.Password, redisConfig.MaxIdle, redisConfig.MaxActive)
		if err != nil {
			panic(err)
		}
	}
	return nil
}

//格式化成字符串
func String(a interface{}, err error) (string, error) {
	return redis.String(a, err)
}

func getConn(writeHost, password string, maxIdle, maxActive int) (pool *redis.Pool, err error) {
	//maxIdle, _ := config.Get("redis.max_idle").Int()
	//maxActive, _ := config.Get("redis.max_active").Int()
	pool = &redis.Pool{
		MaxIdle:   maxIdle,
		MaxActive: maxActive,
		Dial: func() (redis.Conn, error) {
			c, err := redis.Dial("tcp", writeHost)
			if err != nil {
				return nil, err
			}
			if password != "" {
				if _, err := c.Do("AUTH", password); err != nil {
					c.Close()
					return nil, err
				}
			}
			return c, err
		},
		TestOnBorrow: func(c redis.Conn, t time.Time) error {
			_, err := c.Do("PING")
			return err
		},
	}
	return
}

/*
批量或者单个查询redis数据，统一返回map[string]string
@param hkey string 集合键值，如sku
@param targetIds string 查询的id 切片
eg:
    redisConn := gredis.Conn("search_r")
	skuArr := gredis.HgetPi(&redisConn,"Self_SelfGoods",[]string{"1001","10005"})
*/
func Hmget(redisCon string, hkey string, targetIds []string) map[string]string {
	if len(targetIds) == 0 {
		fmt.Println("批量查询不得传入空")
		return nil
	}
	redisConn := Conn(redisCon)
	defer redisConn.Close()

	skuArr := make(map[string]string, 0)
	if len(targetIds) == 1 {
		goods_id := targetIds[0]
		info, err := String(redisConn.Do("HGET", hkey, goods_id))
		if err != nil && err != redis.ErrNil {
			fmt.Print("连接redis错误991：", err)
		}
		if info == "" {
			skuArr[goods_id] = ""
		} else {
			skuArr[goods_id] = info
		}
		//fmt.Println("单个查询")
		return skuArr
	} else { //多个查询
		param := []interface{}{hkey}
		for _, goods_id := range targetIds {
			param = append(param, goods_id)
		}

		res1, err1 := redisConn.Do("hmget", param...)
		reply, _ := redis.Strings(res1, err1)
		if err1 != nil {
			fmt.Println(err1)
		}

		for k, goodsInfo := range reply {
			skuArr[targetIds[k]] = goodsInfo
		}
		//fmt.Println(skuArr)
		return skuArr
	}
}

/*
批量或者单个查询redis数据，统一返回map[string]string
@param hkey string 集合键值，如sku
@param targetIds string 查询的id 切片
eg:
    redisConn := gredis.Conn("search_r")
	skuArr := gredis.HgetPi(&redisConn,"Self_SelfGoods",[]string{"1001","10005"})
*/
func HgetPi(redisCon string, hkey string, targetIds []string) map[string]string {
	redisConn := Conn(redisCon)
	defer redisConn.Close()

	skuArr := make(map[string]string, 0)
	if len(targetIds) == 1 {
		oneId := targetIds[0]
		info, err := String(redisConn.Do("HGET", hkey, oneId))
		if err != nil {
			fmt.Print(err)
		}
		if info == "" {
			skuArr[oneId] = ""
		} else {
			skuArr[oneId] = info
		}
		return skuArr
	}

	for _, v := range targetIds {
		redisConn.Send("HGet", hkey, v)
	}
	redisConn.Flush()

	for _, goods_id := range targetIds {
		info, _ := redisConn.Receive()
		if info == nil {
			skuArr[goods_id] = ""
			continue
		}
		oneInfo := string(info.([]byte))
		skuArr[goods_id] = oneInfo
	}
	return skuArr
}

/*
批量或者单个查询redis数据，统一返回map[string]string
@param hkey string 集合键值，如sku
@param targetIds string 查询的id 切片
eg:
    redisConn := gredis.Conn("search_r")
	skuArr := gredis.HgetPi(&redisConn,"Self_SelfGoods",[]string{"1001","10005"})
*/
func HgetPi2(redisCon string, hkey string, targetIds []string) map[string]string {
	redisConn := Conn(redisCon)
	defer redisConn.Close()

	skuArr := make(map[string]string, 0)
	for _, goods_id := range targetIds {
		info, err := String(redisConn.Do("HGET", hkey, goods_id))
		if err != nil {
			fmt.Print("连接redis错误991：", err)
		}
		if info == "" {
			skuArr[goods_id] = ""
		} else {
			skuArr[goods_id] = info
		}
	}
	return skuArr
}