技术整理:SpringBoot+Redis+lua脚本防止超卖

SpringBoot+redis+lua 防止超卖

一、背景

工作中遇到了有人用 RedisTemplateincrement去做总库存的加减,但是这种方式是保证不了原子性的还是会超卖。

  • redis 是可以保证原子性,但是 RedisTemplate 里面的方法去调用redis是不能保证原子性

二、优化方案

使用 lua 脚本,去执行 加减操作,执行 redis 的命令,来保证原子性

三、重点代码

RedisTemplate 注入

java 复制代码
@Configuration
public class RedisConfig {

    @SuppressWarnings({"rawtypes", "unchecked"})
    @Bean(name = "redisTemplate")
    public RedisTemplate<String, Object> redisTemplate(LettuceConnectionFactory lettuceConnectionFactory) {
        RedisTemplate<String, Object> redisTemplate = new RedisTemplate<>();
        redisTemplate.setConnectionFactory(lettuceConnectionFactory);
        Jackson2JsonRedisSerializer jackson2JsonRedisSerializer = new Jackson2JsonRedisSerializer(Object.class);
        ObjectMapper objectMapper = new ObjectMapper();
        objectMapper.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY);
        objectMapper.enableDefaultTyping(ObjectMapper.DefaultTyping.NON_FINAL);
        jackson2JsonRedisSerializer.setObjectMapper(objectMapper);
        redisTemplate.setValueSerializer(jackson2JsonRedisSerializer);
        redisTemplate.setKeySerializer(new StringRedisSerializer());
        redisTemplate.afterPropertiesSet();
        return redisTemplate;
    }
抢购帮助类并支持集群模式
java 复制代码
/**
 * @Author liyue
 * @date 2024/3/22 14:31
 **/
@Component
public class SecKillProvider {

    private static final String PRODUCT_KEY = "{cluster:}productstock";
    private static final String SECKILL_SCRIPT = "lua/seckill.lua";


    @Resource
    private RedisTemplate<String, Object> redisTemplate;

    public void initStock(int stock) {
        //24小时过期
        RedisUtils.setIfAbsentTimeout(PRODUCT_KEY, stock, 86400);
    }

    //+ DateUtil.format(new Date(), DatePattern.NORM_DATE_PATTERN))
    public boolean seckill(String userId) {
        //调用lua脚本并执行
        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
        redisScript.setResultType(Long.class);//返回类型是Long
        //lua文件存放在resources目录下的redis文件夹内
        redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource(SECKILL_SCRIPT)));

        Integer result = (Integer) redisTemplate.opsForValue().get(PRODUCT_KEY);


        System.out.println(result);
        System.out.println(PRODUCT_KEY);
        Long stock = redisTemplate.execute(redisScript, Arrays.asList(PRODUCT_KEY));

        System.out.println("执行完成--=--stock=" + stock);

        if (stock >= 0) {
            // 抢购成功,可以继续处理订单等逻辑
            System.out.println("User " + userId + " seckill success!");
            return true;
        } else {
            // 抢购失败,库存不足
            System.out.println("User " + userId + " seckill failed!");
            return false;
        }
    }
}
lua 脚本
java 复制代码
local stock = tonumber(redis.call('get', KEYS[1]))
if stock and stock > 0 then
    redis.call('decr', KEYS[1])
    return stock - 1
else
    return -1
end
并发测试类
java 复制代码
/**
 * @Author liyue
 * @date 2024/3/22 15:14
 **/
@RunWith(SpringRunner.class)
@SpringBootTest
@Slf4j
public class SecKillProviderTest {

    @Resource
    private SecKillProvider secKillProvider;


    @Test
    public void t() throws InterruptedException {

        secKillProvider.initStock(10);

        for (int i = 0; i < 2000; i++) {
            new Thread(new Mythread(i)).start();
        }

        Thread.sleep(10000);

    }

    class Mythread implements Runnable {

        private int num;

        Mythread(int num) {
            this.num = num;
        }

        @SuppressWarnings("unchecked")
       SecKillProvider secKillProvider = SpringUtil.getBean("secKillProvider");

        @Override
        public void run() {
            secKillProvider.seckill(String.valueOf(num));
        }
    }
}

总结

使用这种方式,读取文件可以优化成sha的方式去去读取。后面再进行调整,测试类可以模拟并发情况。测试没有什么问题。

本文由mdnice多平台发布

相关推荐
2401_857610031 分钟前
Spring Boot OA:企业办公自动化的创新之路
spring boot·后端·mfc
2401_8543910826 分钟前
Spring Boot OA:企业数字化转型的利器
java·spring boot·后端
山山而川粤33 分钟前
废品买卖回收管理系统|Java|SSM|Vue| 前后端分离
java·开发语言·后端·学习·mysql
2301_8112743136 分钟前
基于Spring Boot的同城宠物照看系统的设计与实现
spring boot·后端·宠物
2301_811274311 小时前
springboot嗨玩旅游网站
spring boot·后端·旅游
mit6.8242 小时前
[Redis#4] string | 常用命令 | + mysql use:cache | session
数据库·redis·后端·缓存
疯狂学习GIS2 小时前
创建第一个IDEA的Java项目的方法
java·后端·intellij idea
捂月3 小时前
Spring Boot 核心逻辑与工作原理详解
java·spring boot·后端
Nightselfhurt3 小时前
RPC学习
java·spring boot·后端·spring·rpc
Estar.Lee10 小时前
查手机号归属地免费API接口教程
android·网络·后端·网络协议·tcp/ip·oneapi