springboot中使用redis并且执行调试lua脚本

今天有个项目需要使用redis，并且有使用脚本的需求。但是因为之前没有写过，所以还有一点点不熟悉，今天记录一下。

原因：

原子操作，redis会将整个脚本作为一个整体执行，中间不会被其他命令插入。

1、创建一个基本的web项目

文件 ->新建 -> 项目，选择spring initializr ，勾选spring web 方便测试，最主要勾选 spring data redis，和下图一样

2、配置redis

因为我是为了测试redis，所以直接使用的本地的redis，你可以替换成application.yaml，或者使用环境变量替换。

#Redis服务器ip
spring.redis.host=127.0.0.1
#Redis服务器连接端口
spring.redis.port=6379

注： 你不配置的话默认值就是上面的

3、测试redis 的lua脚本

先写个能方便测试的接口，因为我为了测试lua 的脚本执行，所以就没讲什么设计，直接验证脚本

这里要做的就是删除过期的key，同时移除hash中的key

package com.pdool.main;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.ClassPathResource;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.scripting.support.ResourceScriptSource;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.*;
import java.util.stream.Collectors;

@RestController
public class TestController {
   @Autowired
   private StringRedisTemplate redisTemplate;

@RequestMapping("test")
   public String test() {
       System.out.println("xxxxxxxxxxxxxx");
       try {
           //调用lua脚本并执行
           DefaultRedisScript<Void> redisScript = new DefaultRedisScript<>();
           redisScript.setResultType(Void.class);//返回类型是Long
           //lua文件存放在resources目录下的redis文件夹内
           Set<String> webApiRequestSet = new HashSet<>();
           webApiRequestSet.add("aa");
           webApiRequestSet.add("bb");

Set<String> webWebsocketRequestSet = new HashSet<>();

String shortHashKey = "abc";
           String longHashKey = "def";
           String delShortKeys = webApiRequestSet.stream().map((requestId) -> shortHashKey + ":" + requestId).collect(Collectors.joining(";"));
           String delLongKeys = webWebsocketRequestSet.stream().map((requestId) -> longHashKey + ":" + requestId).collect(Collectors.joining(";"));
           List<String> keys = Arrays.asList(shortHashKey, longHashKey);

redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("redis/clear-local-key.lua")));
           redisTemplate.execute(redisScript, keys, delShortKeys, delLongKeys);
       } catch (Exception e) {
           e.printStackTrace();
       }
       return "sssssssss";
   }
}

再来看下lua脚本，这东西花了我不少的时间

local short_hash_key = KEYS[1];
local long_hash_key = KEYS[2];
local del_short_hash_keys = ARGV[1];
local del_long_hash_keys = ARGV[2];

local function tt_split(str,reps )
         local resultStrList = {}
         string.gsub(str,'[^'..reps..']+',function ( w )
             table.insert(resultStrList,w)
         end)
         return resultStrList
     end
local del_short_key_list= tt_split(del_short_hash_keys,";")
for i = 1, #del_short_key_list do
   local del_key = del_short_key_list[i];
   redis.call("DEL", del_key)
   redis.call("HDEL", short_hash_key, del_key)
end

local del_long_key_list = tt_split(del_long_hash_keys,";")
for i = 1, #del_long_key_list do
   local del_key = del_long_key_list[i];
   redis.call("DEL", del_key)
   redis.call("HDEL", long_hash_key, del_key)
end

4、技术点

1、redis 传参可以有两个全局变量，一个KEYS，一个是ARGV

2、redis执行的lua 不可以有全局变量，因为会污染环境，所以这里的function 是local

3、lua没有线程的字符串拆分函数，上面的函数是我找些unity的同学从项目中扒出来的

4、lua 列表的下标从1 开始的

5、redis中执行的lua 是 事务性的

6、lua 会阻塞线程，如果脚本太耗时会卡主服务器

5、调试方式

调试lua脚本是真的费劲，因为在redis desktop manage中不太好测试，下面说下怎么测试，如果你本机安装了redis。

1、进入服务关闭关闭正在运行的服务器

2、从命令行启动redis

找到redis 在本机的安装路径，我的路径是 C:\Program Files\Redis

打开命令行，输入下面的命令就能启动redis服务器了

redis-server.exe redis.windows.conf

3、在lua脚本中增加打印

redis.log(redis.LOG_WARNING, "last_tokens " .. last_tokens)

4、运行代码

调用lua脚本，就可以看到下面的输出了，如果你不想看了，就直接从服务启动redis就好了

注意：正式应用的时候把log 注释掉。

6、总结

今天主要的时间耗费在lua 函数的定义，一直没有搞懂怎么调用，点有背，碰了好多次都没成功。

来源：https://blog.csdn.net/perfect2011/article/details/124392225