lua脚本保证Redis多条命令原子性

迷南。 2023-06-08 04:47 141阅读 0赞

# lua脚本保证Redis多条命令原子性 #

> Redis能执行lua脚本，一段lua脚本可以作为一个整体，这样将多条Redis命令写入lua，即可以实现事务的原子性，下面演示了jedis和redisTemplate是如何调用lua脚本的

### Jedis调用lua脚本 ###

# 完成的功能是 set my_key1 my_value1 EX 15 NX, LUA脚本很神奇不需要加分号和换行符
    # 其实该功能在Jedis的API里实现了,但是在spring-data-redis的API里未实现
    # Jedis的API  public String set(final String key, final String value, final String nxxx, final String expx,final long time)  , 这个nxxx即NX或XX, expx即EX或PX
    Jedis j = getJedis();
    String script = "local key = KEYS[1]" + 
        "local value = ARGV[1]" + 
        "local expireInSecs = ARGV[2]" + 
        "return redis.call(\"set\",key,value,\"EX\",expireInSecs,\"NX\")";
    List<String> keys = Arrays.asList("my_key1");
    List<String> args = Arrays.asList("my_value1", "15");
    Object ret = j.eval(script, keys, args);
    System.out.println(ret);

### spring-data-redis(redisTemplate)方式调用lua脚本 ###

# 目前该代码存在一个问题, 不知为何拿不到返回值
    # 可以传入DefaultRedisScript,其是 RedisScript 的实现,无需自己实现RedisScript的所有接口
    List<String> keys = Arrays.asList("my_key1");
    String[] args = { "my_value1_中文", "15"};
    String scriptText = "local key = KEYS[1]" + 
    		"local value = ARGV[1]" + 
    		"local expireInSecs = ARGV[2]" + 
    		"return redis.call(\"set\",key,value,\"EX\",expireInSecs,\"NX\")";
    //ScriptSource scriptSource = new ResourceScriptSource(new ClassPathResource("/lua/lock1.lua"));
    DefaultRedisScript<String>  script = new DefaultRedisScript<String>();
    script.setResultType(String.class);// 【注意】一定要设置,否则返回值都是null
    script.setScriptText(scriptText);
    String s = jedisUtil.redisTemplate.execute(script, keys, args);
    System.out.println(s);
    System.out.println(jedisUtil.string.get("my_key1"));// 中文正确
    return responseResult(s);