Redis 脚本|极客教程

Redis 脚本

什么是 Redis 脚本

Redis 是一个基于内存的高性能键值存储数据库，它提供了丰富的数据结构和操作命令。Redis 脚本是一种特殊的 Lua 脚本，可以在 Redis 中执行。通过在 Redis 中使用 Lua 脚本，可以实现一些复杂的操作，提高性能和数据库的灵活性。

Redis 脚本的优势

原子性操作：Redis 脚本在执行时是原子的，即使是复杂的操作也能保证数据以原子的方式进行处理，避免数据不一致的情况。
性能优化：通过 Redis 脚本可以减少网络往返的次数，提高性能。可以将多个命令打包成一个脚本一次性执行。
灵活性：通过 Lua 脚本可以实现丰富的逻辑控制，满足不同的业务需求。

Redis 中执行脚本

在 Redis 中执行 Lua 脚本需要使用 EVAL 命令或 EVALSHA 命令。EVAL 命令直接执行 Lua 脚本，而 EVALSHA 命令执行的是经过缓存的 Lua 脚本。

EVAL 命令

语法：EVAL script numkeys key [key ...] arg [arg ...]

script：Lua 脚本代码
numkeys：键的数量
key：键名
arg：脚本参数

示例：

127.0.0.1:6379> EVAL "return {KEYS[1],ARGV[1]}" 1 key1 value1
1) "key1"
2) "value1"

EVALSHA 命令

EVALSHA 命令执行的是经过缓存的 Lua 脚本，可以减少网络传输的开销。首先将 Lua 脚本加载到缓存中，然后通过哈希值引用执行。

语法：EVALSHA sha1 numkeys key [key ...] arg [arg ...]

sha1：Lua 脚本的 SHA1 校验和，通过 SCRIPT LOAD 命令可以获取
numkeys：键的数量
key：键名
arg：脚本参数

示例：

首先加载 Lua 脚本到缓存中：

127.0.0.1:6379> SCRIPT LOAD "return {KEYS[1],ARGV[1]}"
"4f3b10d5de755bb409ef899a6b8e8d668b91d0c3"

然后通过 SHA1 校验和执行 Lua 脚本：

127.0.0.1:6379> EVALSHA 4f3b10d5de755bb409ef899a6b8e8d668b91d0c3 1 key1 value1
1) "key1"
2) "value1"

Redis 脚本实例

示例一：计数器

下面的示例演示了如何使用 Lua 脚本实现一个简单的计数器功能。

-- 计数器
local count_key = KEYS[1]
local num = tonumber(ARGV[1])
local count = redis.call('GET', count_key)
if count then
    count = tonumber(count)
else
    count = 0
end
count = count + num
redis.call('SET', count_key, count)
return count

首先加载 Lua 脚本到缓存中：

127.0.0.1:6379> SCRIPT LOAD "local count_key = KEYS[1] local num = tonumber(ARGV[1]) local count = redis.call('GET', count_key) if count then count = tonumber(count) else count = 0 end count = count + num redis.call('SET', count_key, count) return count"
"b6e6d24814f688b0b7bdf5734005a8d59d54590d"

然后执行计数器脚本：

127.0.0.1:6379> EVALSHA b6e6d24814f688b0b7bdf5734005a8d59d54590d 1 counter 1
(integer) 1
127.0.0.1:6379> EVALSHA b6e6d24814f688b0b7bdf5734005...
(integer) 2
127.0.0.1:6379> EVALSHA b6e6d24814f688b0b7bdf5734005...
(integer) 3

示例二：获取最大值和最小值

下面的示例演示了如何使用 Lua 脚本获取 Redis 中存储的最大值和最小值。

-- 获取最大值和最小值
local max_key = KEYS[1]
local min_key = KEYS[2]
local value = tonumber(ARGV[1])
local max = tonumber(redis.call('GET', max_key) or value)
local min = tonumber(redis.call('GET', min_key) or value)
if value > max then
    redis.call('SET', max_key, value)
    max = value
end
if value < min then
    redis.call('SET', min_key, value)
    min = value
end
return {max, min}

首先加载 Lua 脚本到缓存中：

127.0.0.1:6379> SCRIPT LOAD "local max_key = KEYS[1] local min_key = KEYS[2] local value = tonumber(ARGV[1]) local max = tonumber(redis.call('GET', max_key) or value) local min = tonumber(redis.call('GET', min_key) or value) if value > max then redis.call('SET', max_key, value) max = value end if value < min then redis.call('SET', min_key, value) min = value end return {max, min}"
"09f7b4acb2352e11bd4464c1db3d4aa5c36d6475"

然后执行获取最大值和最小值脚本：

127.0.0.1:6379> EVALSHA 09f7b4acb2352e11bd4464c1db3d4aa5c36d6475 2 max min 10
1) (integer) 10
2) (integer) 10
127.0.0.1:6379> EVALSHA 09f7b4acb2352e11bd4464c1db3d4aa5c36d6475 2 max min 5
1) (integer) 10
2) (integer) 5