Redis使用Lua脚本保证zset的删除和插入的原子性

我会带着你远行 2022-12-21 03:10 179阅读 0赞

> 欢迎大家关注我的公众号【老周聊架构】，Java后端主流技术栈的原理、源码分析、架构以及各种互联网高并发、高性能、高可用的解决方案。

话不多说 直接上代码！！！

@Autowired
    private StringRedisTemplate stringRedisTemplate;
    
    private static final String DELANDZADDSCRIPT =
            "if redis.call('zcard', KEYS[1]) > 0 then\n" +
            " redis.call('del', KEYS[1])\n" +
            " for i, v in pairs(ARGV) do\n" +
            " if i > (table.getn(ARGV)) / 2 then\n" +
            " break\n" +
            " end\n" +
            " redis.call('zadd', KEYS[1], ARGV[2*i - 1], ARGV[2*i])\n" +
            " end\n" +
            " return 1\n" +
            "else\n" +
            " for i, v in pairs(ARGV) do\n" +
            " if i > (table.getn(ARGV)) / 2 then\n" +
            " break\n" +
            " end\n" +
            " redis.call('zadd',KEYS[1], ARGV[2*i - 1], ARGV[2*i])\n" +
            " end\n" +
            " return 1\n" +
            "end";
    
    private RedisScript<Long> redisDelAndZaddScript = new DefaultRedisScript<>(DELANDZADDSCRIPT, Long.class);
    
    /** * 刪除及插入 * @param key 键 * @param val 批量值 */
    public void delAndZaddExec(String key, Set<ZSetOperations.TypedTuple<String>> val) { 
        if (StringUtils.isEmpty(key)) { 
            throw new IllegalArgumentException();
        }
        Object[] args = new Object[val.size()*2];
        int i= 0;
        for (ZSetOperations.TypedTuple<String> it : val ) { 
            args[2*i] = String.valueOf(it.getScore());
            args[2*i + 1] = it.getValue();
            i++;
        }
        stringRedisTemplate.execute(redisDelAndZaddScript, Collections.singletonList(key), args);
    }