springboot系列：通过redis+lua脚本实现无锁库存扣减

前言：

在所有的营销活动中，都涉及到库存的处理，比如红包的发放，能量的发放等都需要做到既不能多扣，也不能少发。通常都是通过对扣减接口加锁来实现，但是遇到大规模并发时，对活动库存加锁会影响到并发效率，因为需要每个线程处理完成之后，才会轮到下一个线程。redis提供了lua脚本的形式，来实现命令的原子操作，可以通过将需要扣减的全部逻辑放置在脚本中，既保证并发，又不会影响到库存扣减的数量。

• 1

AI写代码

mavan依赖

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

• 1
• 2
• 3
• 4

AI写代码

这里用的springboot版本为2.0以上。
springboot 关于redis的连接配置就不在这里赘述了。直接看关于lua脚本的配置内容。

luaConfig配置

@Configuration
public class LuaConfig {

    @Bean
    public DefaultRedisScript<Long> stockLua() {
        DefaultRedisScript<Long> defaultRedisScript = new DefaultRedisScript<>();
        defaultRedisScript.setScriptSource(new ResourceScriptSource(
                new ClassPathResource("/lua/")));
        defaultRedisScript.setResultType(Long.class);
        return defaultRedisScript;
    }
}

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12

AI写代码

stock- 脚本文件

--成功返回1,失败返回0
if (redis.call('exists', KEYS[1]) == 1)
then
    local cc = redis.call('get', KEYS[1]);
    local stock = tonumber(cc);
    local num = tonumber(ARGV[1]);
    if (stock < num) then
        return -1;
    end ;
    if (stock >= num) then
        return redis.call('INCRBYFLOAT', KEYS[1], 0 - num);
    end ;
end ;
return -1;

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13
• 14

AI写代码

如何使用

    @Autowired
    private DefaultRedisScript stockLua;

    @Autowired
    private StringRedisTemplate stringRedisTemplate;

• 1
• 2
• 3
• 4
• 5

AI写代码

将脚本的bean 和操作redis服务器的template注入到服务类中。

        if (SystemConstant.MONEY_REWARDS.equals(reqVO.getRightType())) {
            redisKeyList.add(activityStockKey);
            List<String> redisAvgList = new ArrayList<>();
            redisAvgList.add(reqVO.getQuota().toString());
            Object aa = stringRedisTemplate.execute(stockLua, redisKeyList, redisAvgList.toArray());
            BigDecimal leftStock = new BigDecimal(aa.toString());
            if (leftStock.compareTo(new BigDecimal("-1")) <= 0) {
                throw new BusinessException(ExceptionEnum.ACTIVITY_ACCOUNT_RIGHT_NOT_ENOUGH);
            }
        }

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10

AI写代码

通过调用Template的execute方法，执行脚本，扣减对应的库存。
无需加锁操作。