当前位置: 首页 > 工具软件 > Lua > 使用案例 >

redis使用lua脚本

暨承平
2023-12-01

Lua是什么:

Lua由标准C编写而成,几乎在所有操作系统和平台上都可以编译,运行。其设计目的是为了嵌入应用程序中,从而为应用程序提供灵活的扩展和定制功能。

为什么使用:

(1) 减少网络开销: 在Redis操作需求需要向Redis发送5次请求,而使用脚本功能完成同样的操作只需要发送一个请求即可,减少了网络往返时延。

(2) 原子操作: Redis会将整个脚本作为一个整体执行,中间不会被其他命令插入。换句话说在编写脚本的过程中无需担心会出现竞态条件,也就无需使用事务。事务可以完成的所有功能都可以用脚本来实现。

(3) 复用: 客户端发送的脚本会永久存储在Redis中,这就意味着其他客户端(可以是其他语言开发的项目)可以复用这一脚本而不需要使用代码完成同样的逻辑。

(4) 速度快:见 与其它语言的性能比较, 还有一个 JIT编译器可以显著地提高多数任务的性能; 对于那些仍然对性能不满意的人, 可以把关键部分使用C实现, 然后与其集成, 这样还可以享受其它方面的好处。


Redis+Lua:

 Redis 2.6.0 版本开始内置的 Lua 解释器来支持lua脚本,redis客户端可以使用lua脚本,直接在服务器原子地执行多个redis命令。

怎么使用:

1.调用Lua脚本的语法:

 $ redis-cli --eval path/to/redis.lua KEYS[1] KEYS[2] , ARGV[1] ARGV[2] ...

--eval,告诉redis-cli读取并运行后面的lua脚本

path/to/redis.lua,是lua脚本的位置

KEYS[1] KEYS[2],是要操作的键,可以指定多个,在lua脚本中通过KEYS[1], KEYS[2]获取

ARGV[1] ARGV[2],参数,在lua脚本中通过ARGV[1], ARGV[2]获取。

注意:KEYS和ARGV中间的 ',' 两边的空格,不能省略。

redis支持大部分Lua标准库

库名说明
Base提供一些基础函数
String提供用于字符串操作的函数
Table提供用于表操作的函数
Math提供数学计算函数
Debug提供用于调试的函数

2.在脚本中调用redis命令

在脚本中可以使用redis.call函数调用Redis命令

redis.call('set', 'foo', 'bar')
local value=redis.call('get', 'foo') --value的值为bar
redis.call函数的返回值就是Redis命令的执行结果

Redis命令的返回值有5种类型,redis.call函数会将这5种类型的回复转换成对应的Lua的数据类型,具体的对应规则如下(空结果比较特殊,其对应Lua的false)

redis返回值类型和Lua数据类型转换规则

redis返回值类型Lua数据类型
整数回复数字类型
字符串回复字符串类型
多行字符串回复table类型(数组形式)
状态回复table类型(只有一个ok字段存储状态信息)
错误回复table类型(只有一个err字段存储错误信息)

redis还提供了redis.pcall函数,功能与redis.call相同,唯一的区别是当命令执行出错时,redis.pcall会记录错误并继续执行,而redis.call会直接返回错误,不会继续执行。

在脚本中可以使用return语句将值返回给客户端,如果没有执行return语句则默认返回nil

Lua数据类型和redis返回值类型转换规则

Lua数据类型redis返回值类型
数字类型整数回复(Lua的数字类型会被自动转换成整数)
字符串类型字符串回复
table类型(数组形式)多行字符串回复
table类型(只有一个ok字段存储状态信息)状态回复
table类型(只有一个err字段存储错误信息)错误回复

3.脚本相关命令

EVAL语法:

eval script numkeys key [key …] arg [arg …]

通过key和arg这两类参数向脚本传递数据,它们的值在脚本中分别使用KEYS和ARGV两个表类型的全局变量访问。

script:是lua脚本

numkeys:表示有几个key,分别是KEYS[1],KEYS[2]…,如果有值,从第numkeys+1个开始就是参数值,剩余的参数就是ARGV[1],ARGV[2]…

注意:EVAL命令依据参数numkeys来将其后面的所有参数分别存入脚本中KEYS和ARGV两个table类型的全局变量。当脚本不需要任何参数时,也不能省略这个参数(设为0)192.168.127.128:6379>eval "return redis.call('set',KEYS[1],ARGV[1])" 1 name liulei
OK

192.168.127.128:6379>get name
"liulei" 

4.SCRIPT LOAD SCRIPT

将脚本加入缓存,返回值就是SHA1摘要

5.EVALSHA命令

eval sha1id numkeys key [key …] arg [arg …]

在脚本比较长的情况下,如果每次调用脚本都需要将整个脚本传给Redis会占用较多的带宽。

为了解决这个问题,Redis提供了EVALSHA命令,允许开发者通过脚本内容的SHA1摘要来执行脚本,该命令的用法和EVAL一样,只不过是将脚本内容替换成脚本内容的SHA1摘要。

Redis在执行EVAL命令时会计算脚本的SHA1摘要并记录在脚本缓存中,执行EVALSHA命令时Redis会根据提供的摘要从脚本缓存中查找对应的脚本内容,如果找到了则执行脚本,否则会返回错误:"NOSCRIPT No matching script. Please use EVAL."

在程序中使用EVALSHA命令的一般流程如下。

1)先计算脚本的SHA1摘要,并使用EVALSHA命令执行脚本。

2)获得返回值,如果返回“NOSCRIPT”错误则使用EVAL命令重新执行脚本。

虽然这一流程略显麻烦,但值得庆幸的是很多编程语言的Redis客户端都会代替开发者完成这一流程。执行EVAL命令时,先尝试执行EVALSHA命令,如果失败了才会执行EVAL命令。

SCRIPTLOAD "lua-script" 将脚本加入缓存,但不执行, 返回:脚本的SHA1摘要
SCRIPT EXISTS lua-script-sha1 判断脚本是否已被缓存

6.SCRIPT FLUSH(该命令不区分大小写)

清空脚本缓存,redis将脚本的SHA1摘要加入到脚本缓存后会永久保留,不会删除,但可以手动使用SCRIPT FLUSH命令情况脚本缓存。

192.168.127.128:6379>script flush
OK

192.168.127.128:6379>SCRIPT FLUSH
OK 

7.SCRIPT KILL(该命令不区分大小写)

强制终止当前脚本的执行。

但是,如果当前执行的脚步对redis的数据进行了写操作,则SCRIPT KILL命令不会终止脚本的运行,以防止脚本只执行了一部分。脚本中的所有命令,要么都执行,要么都不执行。

192.168.127.128:6379>script kill
(error)NOTBUSY No scripts in execution right now

192.168.127.128:6379>SCRIPT KILL
(error)NOTBUSY No scripts in execution right now

//这是当前没有脚本在执行,所以提示该错误

8.lua-time-limit 5000(redis.conf配置文件中)

为了防止某个脚本执行时间过长导致Redis无法提供服务(比如陷入死循环),Redis提供了lua-time-limit参数限制脚本的最长运行时间,默认为5秒钟。当脚本运行时间超过这一限制后,Redis将开始接受其他命令但不会执行(以确保脚本的原子性,因为此时脚本并没有被终止),而是会返回“BUSY”错误。

安装和使用Lua脚本

1.安装Lua类库环境

yum install -y readline

yum install -y readline-devel

2.下载Lua最新版本并安装

去官网下载lua,可以直接通过wget下载,地址如下:http://www.lua.org/download.html

[root@lunux~]# wget http://www.lua.org/ftp/lua-5.3.4.tar.gz /root/software/download/lua/ 

通过ssh SSH Secure File Transfer Client工具,把软件包上传到Linux服务器上。

目录是:/root/software/download/lua/

[root@linux~]# cd ./software/download/lua/

[root@linux lua]# tar zxvf lua-5.3.4.tar.gz

进入到已经解压的目录lua-5.3.4,准备安装文件。

[root@linux lua]# ls

[root@linux lua]# lua-5.3.4 lua-5.3.4.tar.gz

[root@linux lua]# cd lua-5.3.4

[root@linux lua-5.3.4]#

准备安装环境

使用make linux命令,当前也是需要gcc命令的支持,事先必须安装

安装gcc命令:yum install gcc。

[root@linux lua-5.3.4]# make linux

开始安装lua软件包

使用make install命令

[root@linux lua-5.3.4]# make install

最后进行测试,进到Linux的命令行,然后输入lua命令,开始测试。

[root@linux lua-5.3.4]# lua

>print('lua')
lua

按Ctrl+C退出lua命令模式。

lua脚本文件名必须以.lua后缀名,如果在Linux命令行执行lua脚本,直接lua 脚本名称。

[root@linux lua-5.3.4]# cd /root/application/program/   //执行文件都在这个目录里面

[root@linux program]# mkdir luascript  //创建luaScript脚本目录,存放lua脚本文件

[root@linux program]# cd luascript  

[root@linux luascript]# lua 01.lua    //执行01.lua脚本文件 

redis与lua脚本结合使用,如果在lua脚本里使用了 redis.call命令来操作Redis,执行lua脚步如下面:

//redis-cli和lua脚本的路径可以是相对路径,也可以是绝对路径

//以下代码就是通过绝对地址来执行

//绝对地址:
[root@linux ~]# /root/application/program/redis-tool/redis-cli -h 192.168.127.128 -p 6379 --eval /root/application/program/luascript/02.lua

//相对地址:
//当前目录
192.168.127.128:6379>pwd
[root@linux redis-tool]/root/application/program/redis-tool/

[root@linux redis-tool]# redis-cli -h 192.168.127.128 -p 6379 --eval /root/application/program/luascript/02.lua

Redis客户端执行带有参数的lua脚本,脚本文件的名称是:03.lua。使用脚本文件执行时,用逗号分隔key和arg参数值

//当前redis 数据库中只有name和age两个key,其他数据已经清空。

//当前所在目录
192.168.127.128:6379>keys *
1)"name"
2)"age"

192.168.127.128:6379>get name
"liulei"

192.168.127.128:6379>get age
"15"

//03.lua脚本代码如下:

local name=redis.call("get",KEYS[1])

local age=redis.call("get",KEYS[2])

if name=="LLL" then

redis.call("set",KEYS[1],ARGV[1])

redis.call("incr",KEYS[2])
end

//执行改脚本的命令,必须在Linux的命令行,不是在Redis的命令行

[root@linux ~]# /root/application/program/redis-tool/redis-cli -h 192.168.127.128 -p 6379 --eval /root/application/program/luascript/03.lua name age , patrickLiu

//执行脚本命令后
192.168.127.128:6379>keys *
1)"name"
2)"age"

192.168.127.128:6379>get name
"patrickLiu"

192.168.127.128:6379>get age
"16"

//说明带参数的执行Lua脚本成功 

Redis客户端执行有参数lua,并返回lua的表类型

//04.lua文件的源码

local b1=redis.call("hgetall",KEYS[1])
return b1

//代码很简单,话不多说

//清空当前数据库
192.168.127.128:6379>flushdb

192.168.127.128:6379>keys *
(empty list or set)

192.168.127.128:6379>hmset myhash name zhangsan sex nan address hebeibaoding school laiyuanyizhong
OK

192.168.127.128:6379>hmget myhash name sex address school
1)"zhangsan"
2)"nan"
3)"hebeibaoding"
4)"laiyuanyizhong"

//我们通过redis客户端获取myhash的结果,进入到redis客户端的当前目录

[root@linux redis-tool]# redis-cli -h 192.168.127.128 -p 6379 --eval ../luascript/04.lua myhash
1)"name"
2)"zhangsan"
3)"sex"
4)"nan"
5)"address"
6)"hebeibaoding"
7)"school"
8)"laiyuanyizhong"
//成功获取myhash的列表

多key命令hashtag解决方案:

当如下操作时:
EVAL "redis.call('mset',KEYS[1],ARGV[1],KEYS[2],ARGV[2])" 2 key1 key2 value1 value2
出现以下报错:
ERR 'key1' and 'key2' not in the same slot
通过hashtag进行解决:
EVAL "redis.call('mset',KEYS[1],ARGV[1],KEYS[2],ARGV[2])" 2 {user}key1 {user}key2 value1 value2
hashtag原理:
在设置了hashtag的情况下,集群会根据hashtag决定key分配到的slot, 两个key拥有相同的hashtag:{user}, 它们会被分配到同一个slot,详细见Redis官方文档:https://redis.io/topics/cluster-spec?spm=a2c4g.11186623.0.0.335e2e8bvr61sh

 类似资料: