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

Posted 2022-07-29 香菜聊游戏

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了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