SpringBoot + Redis 執行lua腳本

一、背景

有時候，咱們須要一次性操做多個 Redis 命令，可是 這樣的多個操做不具有原子性，並且 Redis 的事務也不夠強大，不支持事務的回滾，還沒法實現命令之間的邏輯關係計算。因此，通常在開發中，咱們會利用 lua 腳原本實現 Redis 的事務。

二、lua 腳本

Redis 中使用 lua 腳本，咱們須要注意的是，從 Redis 2.6.0後才支持 lua 腳本的執行。
使用 lua 腳本的好處：

• 原子操做：lua腳本是做爲一個總體執行的，因此中間不會被其餘命令插入。
• 減小網絡開銷：能夠將多個請求經過腳本的形式一次發送，減小網絡時延。
• 複用性：lua腳本能夠常駐在redis內存中，因此在使用的時候，能夠直接拿來複用，也減小了代碼量。

三、Redis 中執行 lua 腳本

一、命令格式：

EVAL script numkeys key [key ...] arg [arg ...]

說明：

• script是第一個參數，爲Lua 5.1腳本(字符串)。
• 第二個參數numkeys指定後續參數有幾個key。
• key [key ...]，被操做的key，能夠多個，在lua腳本中經過KEYS[1], KEYS[2]獲取
• arg [arg ...]，參數，能夠多個，在lua腳本中經過ARGV[1], ARGV[2]獲取。

二、若是直接使用 redis-cli命令：

redis-cli --eval lua_file key1 key2 , arg1 arg2 arg3

說明：

• eval 命令後再也不是 lua 腳本的字符串形式，而是一個 lua 腳本文件。後綴爲.lua
• 再也不須要numkeys參數，而是用 , 隔開多個key和多個arg

四、使用 RedisTemplate 執行 lua 腳本

例子：刪除 Redis 分佈式鎖
引入依賴：此依賴爲咱們整合了 Redis ，而且提供了很是好用的 RedisTemplate。

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>

方式一：lua 腳本文件
一、新建 lua 腳本文件：

if redis.call("get",KEYS[1]) == ARGV[1] then
    return redis.call("del",KEYS[1])
else
    return 0
end

說明：先獲取指定key的值，而後和傳入的arg比較是否相等，相等值刪除key，不然直接返回0。

二、代碼測試：

/**
 * @author Howinfun
 * @desc lua 測試
 * @date 2019/11/5
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ActiveProfiles("test")
@SpringBootTest(classes = ThirdPartyServerApplication.class)
public class RedisTest {

    @Autowired
    private StringRedisTemplate redisTemplate;
    @Test
    public void contextLoads() {
        String lockKey = "123";
        String UUID = cn.hutool.core.lang.UUID.fastUUID().toString();
        boolean success = redisTemplate.opsForValue().setIfAbsent(lockKey,UUID,3, TimeUnit.MINUTES);
        if (!success){
            System.out.println("鎖已存在");
        }
        // 執行 lua 腳本
        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
        // 指定 lua 腳本
        redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("redis/DelKey.lua")));
        // 指定返回類型
        redisScript.setResultType(Long.class);
        // 參數一：redisScript，參數二：key列表，參數三：arg（可多個）
        Long result = redisTemplate.execute(redisScript, Collections.singletonList(lockKey),UUID);
        System.out.println(result);
    }
}

方式二：直接編寫 lua 腳本（字符串）
一、代碼測試：

/**
 * @author Howinfun
 * @desc lua 腳本測試
 * @date 2019/11/5
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ActiveProfiles("test")
@SpringBootTest(classes = ThirdPartyServerApplication.class)
public class RedisTest {

    /** 釋放鎖lua腳本 */
    private static final String RELEASE_LOCK_LUA_SCRIPT = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";

    @Autowired
    private StringRedisTemplate redisTemplate;
    @Test
    public void contextLoads() {
        String lockKey = "123";
        String UUID = cn.hutool.core.lang.UUID.fastUUID().toString();
        boolean success = redisTemplate.opsForValue().setIfAbsent(lockKey,UUID,3, TimeUnit.MINUTES);
        if (!success){
            System.out.println("鎖已存在");
        }
        // 指定 lua 腳本，而且指定返回值類型
        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>(RELEASE_LOCK_LUA_SCRIPT,Long.class);
        // 參數一：redisScript，參數二：key列表，參數三：arg（可多個）
        Long result = redisTemplate.execute(redisScript, Collections.singletonList(lockKey),UUID);
        System.out.println(result);
    }
}

注意：可能會有同窗發現，爲何返回值不用 Integer 接收而是用 Long。這裏是由於 spring-boot-starter-data-redis 提供的返回類型裏面不支持 Integer。
你們能夠看看源碼：

/**
 * Represents a data type returned from Redis, currently used to denote the expected return type of Redis scripting
 * commands
 *
 * @author Jennifer Hickey
 * @author Christoph Strobl
 */
public enum ReturnType {

    /**
     * Returned as Boolean
     */
    BOOLEAN,

    /**
     * Returned as {@link Long}
     */
    INTEGER,

    /**
     * Returned as {@link List<Object>}
     */
    MULTI,

    /**
     * Returned as {@literal byte[]}
     */
    STATUS,

    /**
     * Returned as {@literal byte[]}
     */
    VALUE;

    /**
     * @param javaType can be {@literal null} which translates to {@link ReturnType#STATUS}.
     * @return never {@literal null}.
     */
    public static ReturnType fromJavaType(@Nullable Class<?> javaType) {

        if (javaType == null) {
            return ReturnType.STATUS;
        }
        if (javaType.isAssignableFrom(List.class)) {
            return ReturnType.MULTI;
        }
        if (javaType.isAssignableFrom(Boolean.class)) {
            return ReturnType.BOOLEAN;
        }
        if (javaType.isAssignableFrom(Long.class)) {
            return ReturnType.INTEGER;
        }
        return ReturnType.VALUE;
    }
}

因此當咱們使用 Integer 做爲返回值的時候，是報如下錯誤：

org.springframework.data.redis.RedisSystemException: Redis exception; nested exception is io.lettuce.core.RedisException: java.lang.IllegalStateException

    at org.springframework.data.redis.connection.lettuce.LettuceExceptionConverter.convert(LettuceExceptionConverter.java:74)
    at org.springframework.data.redis.connection.lettuce.LettuceExceptionConverter.convert(LettuceExceptionConverter.java:41)