Redis 是一个高性能的内存数据库，支持多种数据结构，如字符串、哈希、列表、集合和有序集合。为了增强其功能，Redis 引入了 Lua 脚本支持，使开发者可以编写自定义的脚本，确保操作的原子性并提高复杂操作的性能。本文将详细介绍如何使用 Lua 脚本对 Redis 进行扩展，重点讲解 eval 命令、redis.call 和 redis.pcall 的用法。
一、Lua 脚本在 Redis 中的作用
Lua 脚本在 Redis 中的主要作用有：
- 原子操作：确保一组命令的原子性，避免并发问题。
- 减少网络开销：将多个命令组合到一个脚本中执行，减少客户端与服务器之间的网络通信次数。
- 复杂逻辑处理：在服务器端执行复杂的业务逻辑，减轻客户端的负担。
二、eval 命令
2.1 eval 命令概述
eval 命令用于执行 Lua 脚本。其基本语法如下：
EVAL script numkeys key [key ...] arg [arg ...]
script：要执行的 Lua 脚本。numkeys：脚本中使用的键的数量。key [key ...]：键列表，供脚本使用。arg [arg ...]：参数列表，供脚本使用。
2.2 示例
以下是一个简单的 Lua 脚本示例，该脚本实现了将一个键的值增加指定的数量：
local current = redis.call('GET', KEYS[1])
if current == false then
current = 0
else
current = tonumber(current)
end
current = current + tonumber(ARGV[1])
redis.call('SET', KEYS[1], current)
return current
使用 eval 命令执行上述脚本：
EVAL "local current = redis.call('GET', KEYS[1]) if current == false then current = 0 else current = tonumber(current) end current = current + tonumber(ARGV[1]) redis.call('SET', KEYS[1], current) return current" 1 mykey 10
该命令会将键 mykey 的值增加 10，如果键不存在，则初始化为 0 后再进行增加。
三、redis.call 和 redis.pcall
3.1 redis.call
redis.call 用于在 Lua 脚本中执行 Redis 命令，并在出现错误时抛出错误。它的使用方式与在命令行中执行 Redis 命令类似。
示例：
local value = redis.call('GET', KEYS[1])
3.2 redis.pcall
redis.pcall 与 redis.call 类似，但它在出现错误时不会抛出错误，而是返回一个描述错误的表。通过 redis.pcall，可以实现更加健壮的错误处理。
示例：
local result = redis.pcall('GET', KEYS[1])
if result.err then
return "Error: " .. result.err
else
return result
end
四、Lua 脚本示例
4.1 原子性操作
以下是一个 Lua 脚本示例，该脚本实现了一个原子性递增操作，并返回递增后的值：
local current = redis.call('GET', KEYS[1])
if not current then
current = 0
end
current = current + tonumber(ARGV[1])
redis.call('SET', KEYS[1], current)
return current
4.2 错误处理
以下是一个使用 redis.pcall 的示例，该脚本尝试删除一个键，如果删除失败则返回错误信息：
local result = redis.pcall('DEL', KEYS[1])
if result.err then
return "Error: " .. result.err
else
return "Deleted: " .. result
end
4.3 复杂逻辑处理
以下是一个更复杂的示例，该脚本实现了一个简化的限流器，每个用户每分钟最多可以访问 10 次：
local user = KEYS[1]
local current_time = redis.call('TIME')[1]
local window_start = current_time - (current_time % 60)
local key = user .. ":" .. window_start
local current_count = redis.call('GET', key)
if not current_count then
current_count = 0
end
if tonumber(current_count) >= 10 then
return "Rate limit exceeded"
else
redis.call('INCR', key)
redis.call('EXPIRE', key, 60)
return "Request allowed"
end