使用redis.call("sinter", ...) 命令,通过 lua 脚本交集多个集合

我想要求两个或者更多集合的交集。要求交集的集合数量通过命令行的 ARGV 传递。因为集合数量从命令行传递,所以 redis.call() 函数的参数数量是不确定的。怎样才能在 Lua 脚本中使用 redis.call() 函数进行交集操作呢?

然而,我已经编写了一个采用如下算法的脚本:

  • 在 KEYS[1] 中接受要求交集的集合数。
  • 使用 setIntersected = redis.call(ARGV[1], ARGV[2]) 求出前两个集合的交集。
  • 运行循环并使用 setIntersected = redis.call("sinter", tostring(setIntersected), set[i])
  • 最后应该得到交集。

上述算法的代码为:

local noOfArgs = KEYS[1] -- 将从 cli 中传递的参数数量存储下来

--[[
    运行 noOfArgs 次循环并初始化表元素,因为我们不知道要求交集的集合数,所以将使用 Table (arrays)
--]]

local setsTable = {}

for i = 1, noOfArgs, 1 do
    setsTable[i] = tostring(ARGV[i])
end

-- 现在找出交集
local intersectedVal = redis.call("sinter", setsTable[1], setsTable[2]) -- 找到第一个交集,因为至少有两个集合
local new_updated_set = ""
for i = 3, noOfArgs, 1 do
    new_updated_set = tostring(intersectedVal)
    intersectedVal = redis.call("sinter", new_updated_set, setsTable[i])
end

return intersectedVal

当我通过命令行传递两个集合时,这个脚本可以正常工作。

例如:

redic-cli --eval scriptfile.lua 2 , points:Above20 points:Above30

 输出:

 1) "playerid:1"

 2) "playerid:2"

 3) "playerid:7"

其中 points:Above20points:Above30 是集合。此时它不会通过从 i=3 开始的循环。

但是当我传递 3 个集合时,我总是得到以下输出:

(空列表或者集合)

所以我编写的求交集的循环有问题。我究竟做错了什么?有没有一种优化的方法可以直接求出多个集合的交集?

点赞
用户3160475
用户3160475

你可能正在寻找的是难以捉摸的unpack()Lua命令,这个命令相当于其他语言中被称为"Splat"运算符的东西。

在你的代码中,使用以下代码:

local intersectedVal = redis.call("sinter", unpack(setsTable))

尽管如此,SINTER是可变参数的,可以接受多个键作为参数。除非你的脚本除了交集操作外还有其他操作,否则最好使用它。

2017-06-15 12:34:33