go - 从 redigo 调用 lua 脚本抛出错误参数数量错误

Question

我正在尝试使用 redigo在redis上执行一个lua脚本。我的代码如下所示:-

package main

import (
    "github.com/PuerkitoBio/redisc"
    "github.com/garyburd/redigo/redis"
    "log"
    "time"
)

const script = `if redis.call("EXISTS", KEYS[1]) == 1 then
    local keyvalues = redis.call("HGETALL", KEYS[1])
    local a = {}
    for i=2, table.getn(ARGV) do
      a[i-1] = ARGV[i]
    end
    local res = redis.call("HMSET", KEYS[1], unpack(a))
    redis.call("EXPIRE", KEYS[1], ARGV[1])  
    return keyvalues
else
    return 2 -- "Key doesn't exists"
end`

func main() {
    cluster := redisc.Cluster{
        StartupNodes: []string{":30001", ":30002", ":30003", ":30004", ":30005", ":30006"},
        DialOptions:  []redis.DialOption{redis.DialConnectTimeout(5 * time.Second)},
        CreatePool:   createPool,
    }
    defer cluster.Close()

    // initialize its mapping
    if err := cluster.Refresh(); err != nil {
        log.Fatalf("Refresh failed: %v", err)
    }

    // grab a connection from the pool
    conn := cluster.Get()
    defer cluster.Close()
    retryConn, errRe := redisc.RetryConn(conn, 3, 1*time.Millisecond)
    if errRe != nil {
        log.Println("Failed to get retry connection " + errRe.Error())
        return
    }
    rScript := redis.NewScript(1, script)
    argv := make([]string, 5)
    argv[0] = "30000"
    argv[1] = "SSF_lastAccessedDate"
    argv[2] = "1481627386"
    argv[3] = "SSF_expiryDate"
    argv[4] = "2481657386"
    reply, errS := rScript.Do(retryConn, "JJNb324a680c35d11e6a1123c15c2d271f21481871788G", argv)
    if errS != nil {
        log.Println("Error in executing script " + errS.Error())
    } else {
        log.Printf("Result %+v", reply)
    }
}

func createPool(addr string, opts ...redis.DialOption) (*redis.Pool, error) {
    return &redis.Pool{
        MaxIdle:     100,
        MaxActive:   4000,
        IdleTimeout: time.Minute,
        Dial: func() (redis.Conn, error) {
            return redis.Dial("tcp", addr, opts...)
        },
        TestOnBorrow: func(c redis.Conn, t time.Time) error {
            if time.Since(t) < time.Minute {
                return nil
            }
            _, err := c.Do("PING")
            return err
        },
    }, nil
}

但是在执行代码时会抛出错误:-

2016/12/16 12:52:32 Error in executing script ERR Error running script (call to f_5a127779e5c1c2daa0b636d9b02f6b570a9f7f13): @user_script:7: @user_script: 7: Wrong number of args calling Redis command From Lua script 

谁能告诉我出了什么问题以及如何解决这个问题？

环境:-

• 开始 - 1.7.4

编辑

按照答案中的建议，我将调用修改为:-

rScript.Do(retryConn, "JJNb324a680c35d11e6a1123c15c2d271f21481871788G", argv[0], argv[1], argv[2], argv[3], argv[4])

但它给出的输出是:-

2016/12/16 21:39:51 Result [[116 121 112 101] [50] [100 97 116 97] [72 101 108 108 111] [120] [50] [83 83 70 95 99 114 101 97 116 105 111 110 68 97 116 101] [49 52 56 49 56 55 49 55 56 56] [83 83 70 95 108 97 115 116 77 111 100 105 102 105 101 100 68 97 116 101] [49 52 56 49 56 55 49 55 56 56] [83 83 70 95 101 120 112 105 114 121 68 97 116 101] [50 52 56 49 54 53 55 51 56 54] [83 83 70 95 108 97 115 116 65 99 99 101 115 115 101 100 68 97 116 101] [49 52 56 49 54 50 55 51 56 54]]

但是从 redis-cli 我得到的输出是:-

1) "type"
 2) "2"
 3) "data"
 4) "Hello"
 5) "x"
 6) "2"
 7) "SSF_creationDate"
 8) "1481871788"
 9) "SSF_lastModifiedDate"
10) "1481871788"
11) "SSF_expiryDate"
12) "2481657386"
13) "SSF_lastAccessedDate"
14) "1481627386"

Answer 1

根据报错信息，错误在这一行:

local res = redis.call("HMSET", KEYS[1], unpack(a))

问题是应用程序将两个参数传递给脚本，键和 argv 转换为字符串。调用 unpack(a) 产生一个值。

修复方法是将每个参数分别传递给脚本:

reply, errS := rScript.Do(retryConn,
   "JJNb324a680c35d11e6a1123c15c2d271f21481871788G",
   "30000",
   "SSF_lastAccessedDate", "1481627386",
   "SSF_expiryDate", "2481657386")

我建议用脚本初始化一个包级变量，这样脚本文本的哈希值就计算一次，而不是每次使用脚本时都计算一次:

var rScript = redis.NewScript(1, `if redis.call("EXISTS", KEYS[1]) == 1 then
    local keyvalues = redis.call("HGETALL", KEYS[1])
    local a = {}
    for i=2, table.getn(ARGV) do
      a[i-1] = ARGV[i]
    end
    local res = redis.call("HMSET", KEYS[1], unpack(a))
    redis.call("EXPIRE", KEYS[1], ARGV[1])  
    return keyvalues
else
    return 2 -- "Key doesn't exists"
end`)

并在 main() 中使用此脚本:

func main() {
  ...
  reply, errS := rScript.Do(retryConn,
   "JJNb324a680c35d11e6a1123c15c2d271f21481871788G",
   "30000",
   "SSF_lastAccessedDate", "1481627386",
   "SSF_expiryDate", "2481657386")

  ...
}

此外，使用 table.remove 来简化脚本:

var rScript = redis.NewScript(1, `if redis.call("EXISTS", KEYS[1]) == 1 then
    local keyvalues = redis.call("HGETALL", KEYS[1])
    local expires = table.remove(ARGV, 1)
    local res = redis.call("HMSET", KEYS[1], unpack(ARGV))
    redis.call("EXPIRE", KEYS[1], expires)  
    return keyvalues
else
    return 2 -- "Key doesn't exists"
end`)

使用redis.Strings将从服务器返回的 []byte slice 转换为字符串 slice :

  reply, errS := redis.Strings(rScript.Do(retryConn,
   "JJNb324a680c35d11e6a1123c15c2d271f21481871788G",
   "30000",
   "SSF_lastAccessedDate", "1481627386",
   "SSF_expiryDate", "2481657386"))

打印此reply 将给出您期望的结果。