更新redis

src/middleware/redis_client.go

@@ -2,354 +2,62 @@ package middleware
 
 import (
 	"fmt"
-	"github.com/go-redis/redis"
+	"github.com/gomodule/redigo/redis"
 	"lingye-gin/src/config"
-	"math/rand"
+	"log"
 	"time"
 )
 
-var RedisClient *redis.Client
+var RedisPoolFactory *redis.Pool
 
 type RedisPool struct{}
 
-func (p RedisPool) Init() {
+// 格式化成字符串
+func (RedisPool) String(a interface{}, err error) (string, error) {
+	return redis.String(a, err)
+}
+
+func (RedisPool) Init() {
 	config.Logger.Info("RedisPool Init")
-	// 连接服务器
-	RedisClient = redis.NewClient(&redis.Options{
-		Addr:     config.AppProps.Redis.Addr,
-		Password: config.AppProps.Redis.Passwd,
-		DB:       config.AppProps.Redis.Database,
-	})
-
-	// 心跳
-	pong, err := RedisClient.Ping().Result()
 
+	RedisPoolFactory = &redis.Pool{
+		MaxIdle:   config.AppProps.Redis.MaxIdle,
+		MaxActive: config.AppProps.Redis.MaxActive,
+		Dial: func() (redis.Conn, error) {
+			conn, err := redis.Dial("tcp", config.AppProps.Redis.Addr,
+				redis.DialReadTimeout(time.Second*10),
+				redis.DialConnectTimeout(time.Second*30),
+				redis.DialPassword(config.AppProps.Redis.Passwd),
+				redis.DialDatabase(config.AppProps.Redis.Database),
+			)
+			if err != nil {
+				log.Println("ERROR: fail init redis pool:", err.Error())
+				if conn != nil {
+					_ = conn.Close()
+				}
+				panic(fmt.Sprintf("ERROR: fail init redis pool: %s", err.Error()))
+			}
+			return conn, err
+		},
+		TestOnBorrow: func(c redis.Conn, t time.Time) error {
+			pong, err := c.Do("PING")
 			// Output: PONG <nil>
 			config.Logger.Info("RedisPool Ping And ", pong, err)
+			return err
+		},
+	}
 	config.Logger.Info("RedisPool Ok")
+	return
 }
 
-func (p RedisPool) Test() {
+func (RedisPool) Set() {
-	ExampleClient_String()
-	ExampleClient_List()
-	ExampleClient_Hash()
-	ExampleClient_Set()
-	ExampleClient_SortSet()
-	ExampleClient_HyperLogLog()
-	ExampleClient_CMD()
-	ExampleClient_Scan()
-	ExampleClient_Tx()
-	ExampleClient_Script()
-	ExampleClient_PubSub()
-}
-
-func ExampleClient_String() {
 	config.Logger.Println("ExampleClient_String")
-	defer config.Logger.Println("ExampleClient_String")
+	// 先从 pool 取出一个连接
+	conn := RedisPoolFactory.Get()
+	defer conn.Close()
 
-	//kv读写
+	if _, err := conn.Do("Set", "key", "value", 1*time.Second); err != nil {
-	err := RedisClient.Set("key", "value", 1*time.Second).Err()
 		config.Logger.Println(err)
-
-	//获取过期时间
-	tm, err := RedisClient.TTL("key").Result()
-	config.Logger.Println(tm)
-
-	val, err := RedisClient.Get("key").Result()
-	config.Logger.Println(val, err)
-
-	val2, err := RedisClient.Get("missing_key").Result()
-	if err == redis.Nil {
-		config.Logger.Println("missing_key does not exist")
-	} else if err != nil {
-		config.Logger.Println("missing_key", val2, err)
-	}
-
-	//不存在才设置 过期时间 nx ex
-	value, err := RedisClient.SetNX("counter", 0, 1*time.Second).Result()
-	config.Logger.Println("setnx", value, err)
-
-	//Incr
-	result, err := RedisClient.Incr("counter").Result()
-	config.Logger.Println("Incr", result, err)
-}
-
-func ExampleClient_List() {
-	config.Logger.Println("ExampleClient_List")
-	defer config.Logger.Println("ExampleClient_List")
-
-	//添加
-	config.Logger.Println(RedisClient.RPush("list_test", "message1").Err())
-	config.Logger.Println(RedisClient.RPush("list_test", "message2").Err())
-
-	//设置
-	config.Logger.Println(RedisClient.LSet("list_test", 2, "message set").Err())
-
-	//remove
-	ret, err := RedisClient.LRem("list_test", 3, "message1").Result()
-	config.Logger.Println(ret, err)
-
-	rLen, err := RedisClient.LLen("list_test").Result()
-	config.Logger.Println(rLen, err)
-
-	//遍历
-	lists, err := RedisClient.LRange("list_test", 0, rLen-1).Result()
-	config.Logger.Println("LRange", lists, err)
-
-	//pop没有时阻塞
-	result, err := RedisClient.BLPop(1*time.Second, "list_test").Result()
-	config.Logger.Println("result:", result, err, len(result))
-}
-
-func ExampleClient_Hash() {
-	config.Logger.Println("ExampleClient_Hash")
-	defer config.Logger.Println("ExampleClient_Hash")
-
-	datas := map[string]interface{}{
-		"name": "LI LEI",
-		"sex":  1,
-		"age":  28,
-		"tel":  123445578,
-	}
-
-	//添加
-	if err := RedisClient.HMSet("hash_test", datas).Err(); err != nil {
-		config.Logger.Fatal(err)
-	}
-
-	//获取
-	rets, err := RedisClient.HMGet("hash_test", "name", "sex").Result()
-	config.Logger.Println("rets:", rets, err)
-
-	//成员
-	retAll, err := RedisClient.HGetAll("hash_test").Result()
-	config.Logger.Println("retAll", retAll, err)
-
-	//存在
-	bExist, err := RedisClient.HExists("hash_test", "tel").Result()
-	config.Logger.Println(bExist, err)
-
-	bRet, err := RedisClient.HSetNX("hash_test", "id", 100).Result()
-	config.Logger.Println(bRet, err)
-
-	//删除
-	config.Logger.Println(RedisClient.HDel("hash_test", "age").Result())
-}
-
-func ExampleClient_Set() {
-	config.Logger.Println("ExampleClient_Set")
-	defer config.Logger.Println("ExampleClient_Set")
-
-	//添加
-	ret, err := RedisClient.SAdd("set_test", "11", "22", "33", "44").Result()
-	config.Logger.Println(ret, err)
-
-	//数量
-	count, err := RedisClient.SCard("set_test").Result()
-	config.Logger.Println(count, err)
-
-	//删除
-	ret, err = RedisClient.SRem("set_test", "11", "22").Result()
-	config.Logger.Println(ret, err)
-
-	//成员
-	members, err := RedisClient.SMembers("set_test").Result()
-	config.Logger.Println(members, err)
-
-	bret, err := RedisClient.SIsMember("set_test", "33").Result()
-	config.Logger.Println(bret, err)
-
-	RedisClient.SAdd("set_a", "11", "22", "33", "44")
-	RedisClient.SAdd("set_b", "11", "22", "33", "55", "66", "77")
-	//差集
-	diff, err := RedisClient.SDiff("set_a", "set_b").Result()
-	config.Logger.Println(diff, err)
-
-	//交集
-	inter, err := RedisClient.SInter("set_a", "set_b").Result()
-	config.Logger.Println(inter, err)
-
-	//并集
-	union, err := RedisClient.SUnion("set_a", "set_b").Result()
-	config.Logger.Println(union, err)
-
-	ret, err = RedisClient.SDiffStore("set_diff", "set_a", "set_b").Result()
-	config.Logger.Println(ret, err)
-
-	rets, err := RedisClient.SMembers("set_diff").Result()
-	config.Logger.Println(rets, err)
-}
-
-func ExampleClient_SortSet() {
-	config.Logger.Println("ExampleClient_SortSet")
-	defer config.Logger.Println("ExampleClient_SortSet")
-
-	addArgs := make([]redis.Z, 100)
-	for i := 1; i < 100; i++ {
-		addArgs = append(addArgs, redis.Z{Score: float64(i), Member: fmt.Sprintf("a_%d", i)})
-	}
-	//Logger.Println(addArgs)
-
-	Shuffle := func(slice []redis.Z) {
-		r := rand.New(rand.NewSource(time.Now().Unix()))
-		for len(slice) > 0 {
-			n := len(slice)
-			randIndex := r.Intn(n)
-			slice[n-1], slice[randIndex] = slice[randIndex], slice[n-1]
-			slice = slice[:n-1]
-		}
-	}
-
-	//随机打乱
-	Shuffle(addArgs)
-
-	//添加
-	ret, err := RedisClient.ZAddNX("sortset_test", addArgs...).Result()
-	config.Logger.Println(ret, err)
-
-	//获取指定成员score
-	score, err := RedisClient.ZScore("sortset_test", "a_10").Result()
-	config.Logger.Println(score, err)
-
-	//获取制定成员的索引
-	index, err := RedisClient.ZRank("sortset_test", "a_50").Result()
-	config.Logger.Println(index, err)
-
-	count, err := RedisClient.SCard("sortset_test").Result()
-	config.Logger.Println(count, err)
-
-	//返回有序集合指定区间内的成员
-	rets, err := RedisClient.ZRange("sortset_test", 10, 20).Result()
-	config.Logger.Println(rets, err)
-
-	//返回有序集合指定区间内的成员分数从高到低
-	rets, err = RedisClient.ZRevRange("sortset_test", 10, 20).Result()
-	config.Logger.Println(rets, err)
-
-	//指定分数区间的成员列表
-	rets, err = RedisClient.ZRangeByScore("sortset_test", redis.ZRangeBy{Min: "(30", Max: "(50", Offset: 1, Count: 10}).Result()
-	config.Logger.Println(rets, err)
-}
-
-//用来做基数统计的算法，HyperLogLog 的优点是，在输入元素的数量或者体积非常非常大时，计算基数所需的空间总是固定 的、并且是很小的。
-//每个 HyperLogLog 键只需要花费 12 KB 内存，就可以计算接近 2^64 个不同元素的基 数
-func ExampleClient_HyperLogLog() {
-	config.Logger.Println("ExampleClient_HyperLogLog")
-	defer config.Logger.Println("ExampleClient_HyperLogLog")
-
-	for i := 0; i < 10000; i++ {
-		RedisClient.PFAdd("pf_test_1", fmt.Sprintf("pfkey%d", i))
-	}
-	ret, err := RedisClient.PFCount("pf_test_1").Result()
-	config.Logger.Println(ret, err)
-
-	for i := 0; i < 10000; i++ {
-		RedisClient.PFAdd("pf_test_2", fmt.Sprintf("pfkey%d", i))
-	}
-	ret, err = RedisClient.PFCount("pf_test_2").Result()
-	config.Logger.Println(ret, err)
-
-	RedisClient.PFMerge("pf_test", "pf_test_2", "pf_test_1")
-	ret, err = RedisClient.PFCount("pf_test").Result()
-	config.Logger.Println(ret, err)
-}
-
-func ExampleClient_PubSub() {
-	config.Logger.Println("ExampleClient_PubSub")
-	defer config.Logger.Println("ExampleClient_PubSub")
-	//发布订阅
-	pubsub := RedisClient.Subscribe("subkey")
-	_, err := pubsub.Receive()
-	if err != nil {
-		config.Logger.Fatal("pubsub.Receive")
-	}
-	ch := pubsub.Channel()
-	time.AfterFunc(1*time.Second, func() {
-		config.Logger.Println("Publish")
-
-		err = RedisClient.Publish("subkey", "test publish 1").Err()
-		if err != nil {
-			config.Logger.Fatal("RedisClient.Publish", err)
-		}
-
-		RedisClient.Publish("subkey", "test publish 2")
-	})
-	for msg := range ch {
-		config.Logger.Println("recv channel:", msg.Channel, msg.Pattern, msg.Payload)
-	}
-}
-
-func ExampleClient_CMD() {
-	config.Logger.Println("ExampleClient_CMD")
-	defer config.Logger.Println("ExampleClient_CMD")
-
-	//执行自定义redis命令
-	Get := func(rdb *redis.Client, key string) *redis.StringCmd {
-		cmd := redis.NewStringCmd("get", key)
-		RedisClient.Process(cmd)
-		return cmd
-	}
-
-	v, err := Get(RedisClient, "NewStringCmd").Result()
-	config.Logger.Println("NewStringCmd", v, err)
-
-	v, err = RedisClient.Do("get", "RedisClient.do").String()
-	config.Logger.Println("RedisClient.Do", v, err)
-}
-
-func ExampleClient_Scan() {
-	config.Logger.Println("ExampleClient_Scan")
-	defer config.Logger.Println("ExampleClient_Scan")
-
-	//scan
-	for i := 1; i < 1000; i++ {
-		RedisClient.Set(fmt.Sprintf("skey_%d", i), i, 0)
-	}
-
-	cusor := uint64(0)
-	for {
-		keys, retCusor, err := RedisClient.Scan(cusor, "skey_*", int64(100)).Result()
-		config.Logger.Println(keys, cusor, err)
-		cusor = retCusor
-		if cusor == 0 {
-			break
-		}
-	}
-}
-
-func ExampleClient_Tx() {
-	pipe := RedisClient.TxPipeline()
-	incr := pipe.Incr("tx_pipeline_counter")
-	pipe.Expire("tx_pipeline_counter", time.Hour)
-
-	// Execute
-	//
-	//     MULTI
-	//     INCR pipeline_counter
-	//     EXPIRE pipeline_counts 3600
-	//     EXEC
-	//
-	// using one rdb-server roundtrip.
-	_, err := pipe.Exec()
-	fmt.Println(incr.Val(), err)
-}
-
-func ExampleClient_Script() {
-	IncrByXX := redis.NewScript(`
-        if redis.call("GET", KEYS[1]) ~= false then
-            return redis.call("INCRBY", KEYS[1], ARGV[1])
-        end
-        return false
-    `)
-
-	n, err := IncrByXX.Run(RedisClient, []string{"xx_counter"}, 2).Result()
-	fmt.Println(n, err)
-
-	err = RedisClient.Set("xx_counter", "40", 0).Err()
-	if err != nil {
 		panic(err)
 	}
-
-	n, err = IncrByXX.Run(RedisClient, []string{"xx_counter"}, 2).Result()
-	fmt.Println(n, err)
 }