SpringBoot中使用Lua脚本实现脚本缓存

lua 脚本

Redis 中使用 lua 脚本，我们需要注意的是，从 Redis 2.6.0后才支持 lua 脚本的执行。

使用 lua 脚本的好处：

原子操作：lua脚本是作为一个整体执行的，所以中间不会被其他命令插入。
减少网络开销：可以将多个请求通过脚本的形式一次发送，减少网络时延。
复用性：lua脚本可以常驻在redis内存中，所以在使用的时候，可以直接拿来复用，也减少了代码量。

首先得引入spring-boot-starter-data-redis依赖，其次把lua脚本放在resources目录下。

1、编写.lua文件 

--库存key
local key1 = KEYS[1]
--事务回查key
local key2 = KEYS[2]
--扣减数量
local num = ARGV[1]
--操作redis 获取sku库存数stock
local stock = tonumber(redis.call('GET', key1))
local result = 0
--如果库存数大于等于减去的数量 则执行redis的decrby命令
--在lua中，除了nil和false，其他的值都为true，所以可用作判空
if(stock and tonumber(stock) >= tonumber(num)) then
  redis.call('DECRBY', key1, num)
  result = 1
end
--用于事务回查
redis.call('SET', key2, result)
return result

2、实例一个脚本执行器RedisScript

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.ClassPathResource;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.scripting.support.ResourceScriptSource;

@Configuration(proxyBeanMethods = false)
public class StockRedisAutoConfiguration {
    @Bean
    public DefaultRedisScript<Long> redisScript() {
        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
        //resource目录下的scripts文件下的.lua文件
        redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("scripts/stock.lua")));
        redisScript.setResultType(Long.class);
        return redisScript;
    }
}

或者

@Slf4j
@Configuration
public class RedisScriptConfig {
    @Resource
    private StringRedisTemplate stringRedisTemplate;

    @Bean
    public DefaultRedisScript<Long> deductStockRedisScript() {
        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
        //resource目录下的scripts文件下的.lua文件
        redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("scripts/stock.lua")));
        redisScript.setResultType(Long.class);
        loadRedisScript(redisScript, "stock.lua");
        return redisScript;
    }

    /**
     * 加载lua脚本到redis服务器
     * @param redisScript
     * @param luaName
     */
    private void loadRedisScript(DefaultRedisScript<Long> redisScript, String luaName) {
        try {
            List<Boolean> results = stringRedisTemplate.getConnectionFactory().getConnection().scriptExists(redisScript.getSha1());
            if (Boolean.FALSE.equals(results.get(0))) {
                String sha = stringRedisTemplate.getConnectionFactory().getConnection().scriptLoad(scriptBytes(redisScript));
                log.info("预加载lua脚本成功：{}, sha=[{}]", luaName, sha);
            }
        } catch (Exception e) {
            log.error("预加载lua脚本异常：{}", luaName, e);
        }
    }

    /**
     * 序列化lua脚本
     * @param script
     * @return
     */
    private byte[] scriptBytes(RedisScript<?> script) {
        return this.stringRedisTemplate.getStringSerializer().serialize(script.getScriptAsString());
    }
}

3、调用

        @Resource
        private StringRedisTemplate stringRedisTemplate;
        @Resource
        private RedisScript<Long> stockRedisScript;
        // 在方法里调用执行
        // 第一个参数 对 lua 脚本对象的调用
        // 第二个参数 是 lua 脚本中 Keys 参数
        // 第三个参数 是 lua 脚本中 ARGV 参数
        Long result = stringRedisTemplate.execute(stockRedisScript, skuKey, num);