登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  数据库 >  Redis

Redis Lua 脚本返回多个值时客户端怎么解析

来源:17golang原创

时间:2026-09-07 08:09:30 125浏览 收藏

Redis Lua 脚本不能像函数那样直接返回多个独立值。实际做法是返回一个按位置组织的 Lua 数组,Redis 再把它编码成 RESP 数组,客户端收到的通常就是一个列表或切片。解析是否稳定,关键不在客户端 API,而在脚本是否提前固定了每个位置的含义。

要点速览
  • 只用连续的索引表返回多个值,例如 return {status, value, count}
  • Lua 的 false 会变成空值回复,数组中的第一个 nil 会让后续位置丢失。
  • 客户端按位置解包并显式处理空值;Cluster 下脚本访问的键还要落在同一 hash slot。

一、Lua 多值返回本质上是一个 RESP 数组

下面这个脚本看起来返回了三个值,实际上返回的是一个 Lua 索引表:

local score = redis.call('GET', KEYS[1])
local count = redis.call('SCARD', KEYS[2])
-- 固定位置:状态、原始值、数量
return {'ok', score or '', count}

Redis 会把索引表编码成数组,客户端看到的顺序就是 ok、键值、集合数量。关联字段不会被编码成对象;例如 {status = 'ok', value = 'x'} 不是跨客户端都能直接得到的字典。RESP2 下,Lua 字符串对应 bulk string,数字对应整数,false 对应 null bulk reply。

Redis Lua 索引表编码为 RESP 数组并映射到客户端列表的静态结构图
图1:Redis Lua 索引表、RESP 数组和客户端列表之间的静态类型映射。

二、先固定返回契约,避免 nil 把数组截断

最容易踩坑的是把未命中的值直接放入数组:

local value = redis.call('GET', KEYS[1])
-- 不要让 false 成为中间位置的隐式协议
return {'ok', value, redis.call('EXISTS', KEYS[1])}

GET 没有结果时,Redis 返回给 Lua 的是 false;它再被编码为空值。若代码使用真正的 nil 填充数组,RESP2 的数组转换会在第一个 nil 处停止,后面的元素不会出现。因此更稳妥的契约是:位置 1 永远是状态字符串,位置 2 用空字符串表示“无值”,位置 3 永远是整数。浮点结果则先在 Lua 中转成字符串,否则 Lua 的单一数字类型会按整数回复处理。

返回位置建议含义客户端处理
0状态或错误码先判断字符串,再决定是否继续
1业务值,可为空Python 判断 None/空字节串,Go 判断 nil
2数量或版本转换为 int,不要按字符串拼接

三、Python redis-py 按位置拆开并处理空值

redis-py 执行脚本后通常得到 Python 的 list。如果连接未设置 decode_responses=True,字符串元素还可能是字节串,所以不要直接把整个结果当 JSON 对象使用:

import redis

r = redis.Redis(decode_responses=True)
script = r.register_script("""
local value = redis.call('GET', KEYS[1])
local count = redis.call('SCARD', KEYS[2])
-- 用空字符串保持返回位置稳定
return {'ok', value or '', count}
""")

result = script(keys=['profile:42', 'online:users'])
status, value, count = result
if status != 'ok':  # 先处理协议状态,再使用业务字段
    raise RuntimeError(status)
print(value or '', int(count))

如果保留默认字节串模式,就把 result[0]result[1] 显式解码。不要用“长度等于 3”作为唯一判断,因为真正的空值回复可能表现为 None。脚本失败则由客户端抛出异常,业务代码应把网络错误、脚本错误和“业务上没有值”分开记录。

Python redis-py 与 Go go-redis 按返回位置解析 Redis Lua 数组的静态结构图
图2:客户端按固定位置接收状态、业务值和计数,并在边界处处理空值。

四、Go go-redis 解析切片时别忘了原子性边界

go-redis 的脚本结果可用 Result() 接收,再把返回值断言为 []interface{}。示例只做最小的类型转换,不把 Redis 的动态回复强行当成固定结构体:

ctx := context.Background()
script := redis.NewScript(`
local value = redis.call('GET', KEYS[1])
local count = redis.call('SCARD', KEYS[2])
-- 返回顺序必须和 Go 的断言顺序一致
return {'ok', value or '', count}
`)

raw, err := script.Run(ctx, rdb, []string{"profile:42", "online:users"}).Result()
if err != nil { // 脚本错误和业务空值不是一回事
    return err
}
items, ok := raw.([]interface{})
if !ok || len(items) != 3 {
    return fmt.Errorf("unexpected lua reply: %T", raw)
}
status, _ := items[0].(string)
value, _ := items[1].(string)
count, ok := items[2].(int64)
if !ok || status != "ok" {
    return fmt.Errorf("invalid reply: %v", raw)
}
fmt.Println(value, count)

脚本的一次执行仍是 Redis 的原子执行单元,但原子性不等于可以随意访问键。Cluster 模式下,脚本使用的键必须按参数传入,并且通常要落在同一个 hash slot;否则客户端可能在发送前或服务端返回跨槽错误。把返回协议写进代码注释、测试“命中/未命中/脚本报错”三种结果,后续换客户端时就不会靠猜。

相关问题

Lua 能不能返回一个带字段名的对象?

RESP2 下普通关联表不会按对象传输。需要字段名时,可以返回交替的键和值,或在脚本内编码成 JSON,但客户端要承担解码成本。

为什么返回的小数变成了整数?

Lua 只有一种 number 类型,Redis 的 Lua 到 RESP2 转换会按整数回复处理。需要保留小数时,先格式化成字符串并在客户端转换。

返回数组为空时怎么判断?

先判断结果是否为 nil,再判断数组长度;不要把空数组、空字符串和空值回复当成同一种业务状态。

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>