LUA批量原子性操作redis

我就是我 2023-07-24 13:13 19阅读 0赞

## LUA执行redis指令 ##

*  eval命令

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2MDQyOTM4_size_16_color_FFFFFF_t_70][]

*  redisTemplate的execute方法：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2MDQyOTM4_size_16_color_FFFFFF_t_70 1][]

## 使用stringRedisTemplate执行LUA ##

注意：lua脚本的下标是从1开始的

//批量执行redis脚本拼接
    StringBuffer script = new StringBuffer();
    for (int i = 1; i <= redisKeyArr.length; i++) {
         script.append("redis.call('SET', KEYS[" + i + "], " + "'" + JSONObject.toJSONString(valueList.get(i - 1)) + "'" + ") ");
         script.append("redis.call('EXPIRE', KEYS["+ i +"], " + EquityConstant.SECONDS_OF_DAY + ") ");
    }
    DefaultRedisScript redisScript = new DefaultRedisScript();
    redisScript.setScriptText(script.toString());
    // 批量执行
    stringRedisTemplate.execute(redisScript, keyList);

##  使用JedisCluster执行LUA ##

/**
         * lua redis 脚本回滚
         * @param redisMap
         */
        public void redisLuaRollback(Map<String, Long> redisMap) {
            if (MapUtils.isEmpty(redisMap)) {
                return;
            }
            //从map中获取redis 中需要回滚的key
            List<String> keyList = new ArrayList<>(redisMap.keySet());
            //list转数组为了传多个KEY
            String[] redisKeyArr = keyList.toArray(new String[keyList.size()]);
            //回滚脚本拼接
            StringBuffer script = new StringBuffer();
            script.append("local result = {} ");
            for (int i = 1; i <= redisKeyArr.length; i++) {
                script.append("result[" + i + "]=redis.call('DECRBY', KEYS[" + i + "], " + redisMap.get(redisKeyArr[i - 1]) + ") ");
            }
            script.append("return result");
            log.info("Lua脚本批量回滚key:{},结果={}", keyList, jedisCluster.eval(script.toString(), redisKeyArr.length, redisKeyArr));
    
        }

# #

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2MDQyOTM4_size_16_color_FFFFFF_t_70]: /images/20230528/f222bfe091214590ba2b8a00cdd7e78e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM2MDQyOTM4_size_16_color_FFFFFF_t_70 1]: /images/20230528/334dbfa897774c2eb62d28e9cb250037.png