在 Redis 中，Lua 脚本 提供了一种强大的方式来执行原子操作，可以在 Redis 服务器上直接执行 Lua 代码，从而避免了多次网络往返和保证操作的原子性。Redis 内置了对 Lua 脚本的支持，通过 EVAL 命令来执行脚本，EVALSHA 则用于执行已经加载到 Redis 服务器的脚本。
1. Redis 与 Lua 脚本的基本交互
1.1 基本的 Lua 脚本执行
你可以通过 EVAL 命令执行 Lua 脚本。基本语法如下：
EVAL <lua-script> <num-keys> <key1> <key2> ... <arg1> <arg2> ...
<lua-script>：Lua 脚本内容。<num-keys>：脚本操作的 Redis 键的数量。<key1> <key2> ...：键的名称。<arg1> <arg2> ...：传递给 Lua 脚本的额外参数。
1.2 示例：简单的 Lua 脚本
下面是一个简单的 Lua 脚本，计算 Redis 中某个键的值并加上一个指定的值。
-- Lua脚本：获取某个键的值，增加指定值并返回
local current = redis.call("GET", KEYS[1]) -- 获取键的当前值
if current then
local new_value = tonumber(current) + tonumber(ARGV[1]) -- 将值加上传入的参数
redis.call("SET", KEYS[1], new_value) -- 设置新的值
return new_value
else
return nil -- 如果键不存在，返回nil
end
你可以使用以下命令在 Redis 客户端执行这个脚本：
EVAL "local current = redis.call('GET', KEYS[1]); if current then local new_value = tonumber(current) + tonumber(ARGV[1]); redis.call('SET', KEYS[1], new_value); return new_value; else return nil; end" 1 mykey 10
这个脚本的作用是获取 mykey 键的值，将其与传入的参数 10 相加，然后将新的值设置回 mykey。
1.3 通过 EVALSHA 执行 Lua 脚本
当脚本很长或需要频繁执行时，可以通过 Redis 的 SCRIPT LOAD 命令将 Lua 脚本预加载到 Redis 服务器上，然后通过 EVALSHA 来执行该脚本，这样可以避免每次都发送完整的 Lua 脚本。
加载脚本：
SCRIPT LOAD "local current = redis.call('GET', KEYS[1]); if current then local new_value = tonumber(current) + tonumber(ARGV[1]); redis.call('SET', KEYS[1], new_value); return new_value; else return nil; end"
这将返回一个脚本的 SHA1 值，例如："abc123"。
- 通过
EVALSHA执行脚本：
EVALSHA abc123 1 mykey 10
这种方式可以提高性能，因为 Redis 只需要在第一次加载脚本时处理一次脚本内容。
2. Lua 脚本的高级用法
2.1 操作多个键
Redis 的 Lua 脚本可以操作多个键，脚本通过 KEYS 数组访问键名，ARGV 数组访问传入的参数。
local value1 = redis.call("GET", KEYS[1])
local value2 = redis.call("GET", KEYS[2])
if value1 and value2 then
return value1 + value2
else
return nil
end
执行：
EVAL "local value1 = redis.call('GET', KEYS[1]); local value2 = redis.call('GET', KEYS[2]); if value1 and value2 then return value1 + value2; else return nil; end" 2 key1 key2
2.2 使用 Redis 的事务（MULTI/EXEC）
你可以在 Lua 脚本中模拟 Redis 的事务，即使用 MULTI 和 EXEC 来执行多个命令的事务。
redis.call("MULTI")
redis.call("SET", KEYS[1], ARGV[1])
redis.call("SET", KEYS[2], ARGV[2])
redis.call("EXEC")
2.3 Lua 脚本的原子性
由于 Redis 执行 Lua 脚本时会在服务器端完成脚本中的所有操作，因此 Lua 脚本的执行是原子的。这意味着在执行脚本期间，其他 Redis 客户端无法对相关的键进行修改，保证了数据一致性。
3. 使用 Lua 脚本的优点
原子性：Lua 脚本在 Redis 中执行时是原子的，所有操作要么全部成功，要么全部失败，保证了数据的一致性。
减少网络开销：将多次操作封装在一个 Lua 脚本中，可以减少网络往返，提高效率。
灵活性：可以在 Redis 内部执行复杂的逻辑，如条件判断、循环、数据处理等。
4. 在 Redis 中调用 Lua 脚本的注意事项
脚本执行时间：长时间执行的脚本会导致 Redis 阻塞。因为 Lua 脚本在 Redis 内部是同步执行的，如果脚本非常复杂，可能会阻塞整个 Redis 服务器。避免执行耗时过长的操作。
键的数量限制：Redis 的 EVAL 命令最大支持 16 个键。如果超过这个限制，脚本将不能执行。可以将多个键合并为一个键来绕过这个限制。
内存消耗：Lua 脚本的执行会占用 Redis 内存，因此要避免脚本中使用过多内存的操作。
5. 使用 Redis 客户端执行 Lua 脚本
以 Node.js 为例，使用 ioredis 客户端执行 Lua 脚本：
const Redis = require('ioredis');
const redis = new Redis();
const script = `
local current = redis.call('GET', KEYS[1]);
if current then
local new_value = tonumber(current) + tonumber(ARGV[1]);
redis.call('SET', KEYS[1], new_value);
return new_value;
else
return nil;
end
`;
redis.eval(script, 1, 'mykey', 10)
.then(result => {
console.log('New value:', result);
})
.catch(err => {
console.error('Error:', err);
});
总结
Redis 提供了强大的 Lua 脚本支持，可以在服务器端执行复杂的操作，同时保证操作的原子性和减少网络往返。通过使用 EVAL 或 EVALSHA 命令，你可以在 Redis 中执行各种自定义的 Lua 脚本，帮助你实现更高效的数据处理和业务逻辑。
发布者：myrgd，转载请注明出处：https://www.object-c.cn/4928
