redis执行一对一和一对多形式Lua脚本

最新推荐文章于 2024-11-03 08:30:37 发布

原创最新推荐文章于 2024-11-03 08:30:37 发布 · 915 阅读

0 ·

CC 4.0 BY-SA版权

Java开发相关笔记专栏收录该内容

12 篇文章

订阅专栏

本文分享了Redis中一对一和一对多形式的Lua脚本。介绍了一对一查询命令，以及一对多场景下通过一个key查询其下随机10个内容的操作流程。还提到生产环境下的写法可提高性能，经1000次执行比较，脚本速度快于Java调用两次redisTemplate方法，并给出测试案例。

在上一篇文章说到java可以执行通过redisTemplate执行Lua脚本。目前从网上的结果来看，这种文章很多只能当成自己的一种笔记。

这次我分享的是，一对一和一对多形式的Lua脚本。

一对一查询场景

redis中存放的数数据结构
key : poem_verse_var ; value: PoemEntity
从结构可知道，一对一查询命令如下，同理也可以交给java执行
eval "local setCoo = redis.call('get', KEYS[1]) return setCoo" 1 poem_verse_瞿唐石城草萧瑟

一对多查询场景

key: poetry对应的value是个id集合
后面还有poem_id加上id这个变量，形如 poem_id_5aed7a5d886bae18ab743ca8 这样key对应value的PoemEntity。

场景需求：通过一个poetry查询他下面随机10首PoemEntity出来。

local id_set_collection = {} 
local id_set = redis.call('SRANDMEMBER', KEYS[1], '10') 
for i=1, #id_set do 
table.insert(id_set_collection, i, redis.call('get', 'poem_id_'..id_set[i])) 
end 
return id_set_collection

这里的操作流程是，通过key随机查询当前下面10个的id集合。
然后通过id集合遍历，每个id查询库中的内容，再次塞到id_set_collection集合里去。
需要注意
#id_set是求数组的大小，Lua脚本不叫数组，叫table

我的生产环境下的写法是，之所以这样写，是因为static块在类加载时只执行一次，并且生命周期与类一样。所以提高了性能。

通过1000次的执行比较发现，脚本的速度明显快于Java调用两次的redisTemplate方法。

final static String one2ManyLuaScript = "Lua Script内容";
private static DefaultRedisScript<List> one2ManyRs = 
	new DefaultRedisScript<List>();

static {
   one2ManyRs.setScriptText(one2ManyLuaScript);
   one2ManyRs.setResultType(List.class);
}

测试案例

测试代码如下，脚本调用的初始化就是上面的静态代码初始化方法。
redis初始化连接已经忽略。

//----------脚本执行块---------//
String key1 = "poem_poet_唐彦谦";
List<String> keyList = new ArrayList<String>();
keyList.add(key1);
long startTime = System.currentTimeMillis();
int len = 1000;
for(int i=0; i<len; i++) {
    redisTemplate.execute(rs, keyList);
}
long time = (System.currentTimeMillis() - startTime);
System.out.println("脚本1000次->" + time);


//----------代码执行块---------//
long javaStartTime = System.currentTimeMillis();
int resultCount = 10;
for (int i=0; i<len; i++) {
    List<String> id = stringRedisTemplate.opsForSet().randomMembers(key1, resultCount);
    List<String> newIds = new ArrayList<>();
    for(int j=0; j<resultCount; j++) {
        newIds.add("poem_id_" + id.get(j));
    }
    redisTemplate.opsForValue().multiGet(newIds);
}
long result = (System.currentTimeMillis() - javaStartTime);
System.out.println("java代码1000次->" + result);