如果你对Redis和Lua的关系不太清楚,请先阅读:Redis进阶之使用Lua脚本开发
1.RedisScript
首先你得引入spring-boot-starter-data-redis依赖,其次把lua脚本放在resources目录下。
@Bean
public DefaultRedisScript defaultRedisScript() {
DefaultRedisScript defaultRedisScript = new DefaultRedisScript<>();
defaultRedisScript.setResultType(List.class);
// defaultRedisScript.setScriptText("");
defaultRedisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("redis/demo.lua")));
return defaultRedisScript;
}
在Spring Boot2.0的时候,上述配置没有问题,但在Spring Boot1.5测试会出错,需要将List.class改为具体的返回类型(如Long.class)。
RedisScript的getSha1()方法可以获取脚本摘要。
2.调用脚本
/**
* List设置lua的KEYS
*/
List keyList = new ArrayList();
keyList.add("count");
keyList.add("rate.limiting:127.0.0.1");
/**
* 用Mpa设置Lua的ARGV[1]
*/
Map argvMap = new HashMap();
argvMap.put("expire", 10000);
argvMap.put("times", 10);
/**
* 调用脚本并执行
*/
List result = redisTemplate1.execute(redisScript, keyList, argvMap);
System.out.println(result);
若是出现序列化问题,可以指定序列化方式。
public T execute(RedisScript script, RedisSerializer> argsSerializer, RedisSerializer resultSerializer,
List keys, Object... args) {
return scriptExecutor.execute(script, argsSerializer, resultSerializer, keys, args);
}
3.Lua脚本
--获取KEY
local key1 = KEYS[1]
local key2 = KEYS[2]
-- 获取ARGV[1],这里对应到应用端是一个List
参考:https://blog.csdn.net/fsw4848438/article/details/81540495