优化redis eval的key重命名

redis/redis.go

@@ -20,7 +20,7 @@ const (
 
 var Nil = redis.Nil
 
-type ReidsOption struct {
+type RedisOption struct {
 	Addr      string
 	Addrs     []string
 	Password  string // 密码
@@ -67,7 +67,7 @@ type redisClient struct {
 	separator string
 }
 
-func New(opt *ReidsOption) redisClient {
+func New(opt *RedisOption) redisClient {
 	var rdb redisClient
 
 	if len(opt.Addrs) == 0 && len(opt.Addr) > 0 {
@@ -114,12 +114,6 @@ func New(opt *ReidsOption) redisClient {
 	return rdb
 }
 
-func (rdb redisClient) Eval(ctx context.Context, script string, keys []string, args ...interface{}) *redis.Cmd {
-	newKeys := rdb.FormatKeys(keys...)
-
-	return rdb.UniversalClient.Eval(ctx, script, newKeys, args...)
-}
-
 func (rdb redisClient) Prefix() string {
 	return rdb.prefix
 }

redis/redis_test.go

@@ -143,7 +143,7 @@ func TestRedisPool(t *testing.T) {
 
 		t.Log(client.ConfigGet(context.Background(), "slowlog-log-slower-than").Result())
 
-	}, redis.ReidsOption{
+	}, redis.RedisOption{
 		Addr:     "192.168.123.100:6379",
 		PoolSize: 100,
 		PoolFIFO: true,

redis/rename_hook.go

@@ -49,7 +49,7 @@ func (r renameKey) renameKey(cmd redis.Cmder) {
 	}
 
 	switch strings.ToUpper(cmd.Name()) {
-	case "SELECT", "EVAL":
+	case "SELECT":
 		// 无KEY指令
 	case
 		"RENAME", "RENAMENX",
@@ -69,6 +69,11 @@ func (r renameKey) renameKey(cmd redis.Cmder) {
 	case "MSET", "MSETNX":
 		// 间隔KEY，KEY位置规则1,3,5,7
 		r.rename(args, createSepuence(1, len(args), 2)...)
+	case "EVAL":
+		// 命令中包含键数量  EVAL script numkeys [key [key ...]] [arg [arg ...]]
+		if n, ok := args[2].(int); ok && n > 0 {
+			r.rename(args, createSepuence(3, 3+n, 1)...)
+		}
 	default:
 		// 默认第一个参数为键值
 		r.rename(args, 1)

redis/rename_hook_test.go

@@ -1,9 +1,24 @@
 package redis
 
-import "testing"
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
 
 func TestRename(t *testing.T) {
-	New(&ReidsOption{
+	New(&RedisOption{
 		Addrs: []string{"192.168.123.100:6379"},
 	})
 }
+
+func TestEvalName(t *testing.T) {
+	rdb := New(&RedisOption{
+		Addrs:  []string{"192.168.123.100:6379"},
+		Prefix: "aabbcc",
+	})
+
+	_, err := rdb.Eval(context.Background(), "return 1", []string{"a1", "a2", "a3"}, "b1", "b2", "b3").Result()
+	assert.Nil(t, err, err)
+}