Redis zset按前缀模式合并成员并求和的可行方案咨询

嘿，这个问题我刚好碰到过，给你几个实用的解决思路！你说的没错，ZINTERSTORE 和 ZUNIONSTORE 确实只处理成员名称完全匹配的情况，没法直接按前缀分组统计。不过我们可以通过下面两种方法实现需求：

方法一：用Lua脚本（推荐，高效原子操作） #

Redis支持执行Lua脚本，能在服务器端原子性地完成遍历、分组、求和的全流程，避免客户端和服务器之间多次往返。

假设你的ZSet成员格式是 [前缀]:[名称]（比如cat:mimi、dog:wangwang），可以用这个脚本：

local prefix_counts = {}
local cursor = "0"
repeat
    -- 分批扫描ZSet成员，COUNT参数可以根据数据量调整
    local result = redis.call("ZSCAN", KEYS[1], cursor, "COUNT", 1000)
    cursor = result[1]
    local members = result[2]
    -- 遍历每一组成员和分值
    for i = 1, #members, 2 do
        local member = members[i]
        local score = members[i+1]
        -- 拆分前缀（这里用冒号作为分隔符，可根据你的实际格式修改）
        local colon_pos = string.find(member, ":")
        if colon_pos then
            local prefix = string.sub(member, 1, colon_pos - 1)
            -- 累加对应前缀的总分值
            prefix_counts[prefix] = (prefix_counts[prefix] or 0) + score
        end
    end
until cursor == "0"

-- 整理成Redis能返回的键值对格式
local output = {}
for prefix, total_score in pairs(prefix_counts) do
    table.insert(output, prefix)
    table.insert(output, total_score)
end
return output

执行脚本的命令（替换成你的ZSet名称pets）：

redis-cli EVAL "$(cat your_script.lua)" 1 pets

执行后会返回类似这样的结果：1) "cat" 2) "2" 3) "dog" 4) "1"，对应每个前缀的总分值（因为你的所有score都是1，结果就是该前缀下的成员数量）。

方法二：客户端端扫描处理 #

如果不想用Lua脚本，也可以在客户端通过ZSCAN命令分批遍历ZSet，然后在本地统计前缀的总分值。比如用Python的Redis库实现：

import redis

r = redis.Redis(host="localhost", port=6379, db=0)
cursor = 0
prefix_total = {}

while True:
    # 分批扫描ZSet
    cursor, members = r.zscan("pets", cursor=cursor)
    for member, score in members:
        # 拆分前缀
        colon_idx = member.find(":")
        if colon_idx != -1:
            prefix = member[:colon_idx]
            prefix_total[prefix] = prefix_total.get(prefix, 0) + score
    # 游标为0时结束遍历
    if cursor == 0:
        break

print(prefix_total)
# 输出示例：{'cat': 2, 'dog': 1}

补充说明 #

• 如果你的成员前缀分隔符不是冒号，只需修改脚本/代码里的分隔符判断逻辑即可。
• 因为你的所有成员score都是1，求和结果等价于统计每个前缀下的成员数量；如果后续score有变化，这两种方法也能正确累加总分值。

内容的提问来源于stack exchange，提问作者Ismail