[一个简单的秒杀架构的演变]2. 使用乐观锁

地址

Github：https://github.com/dolyw/SeckillEvolution
Gitee(码云)：https://gitee.com/dolyw/SeckillEvolution

1. 思路介绍

这次我们引入乐观锁，这里可以先查看一篇文章: MySQL那些锁

主要改造是扣库存，每个线程在检查库存的时候会拿到当前商品的乐观锁版本号，然后在扣库存时，如果版本号不对，就会扣减失败，抛出异常结束，这样每个版本号就只能有一个线程操作成功，其他相同版本号的线程秒杀失败，就不会存在卖超问题了

2. 代码实现

在SeckillEvolutionController创建一个乐观锁下订单入口的方法

SeckillEvolutionController

/**
 * 使用乐观锁下订单
 *
 * @param id 商品ID
 * @return com.example.common.ResponseBean
 * @throws Exception
 * @author wliduo[i@dolyw.com]
 * @date 2019/11/22 14:24
 */
@PostMapping("/createOptimisticLockOrder/{id}")
public ResponseBean createOptimisticLockOrder(@PathVariable("id") Integer id) throws Exception {
    Integer orderCount = seckillEvolutionService.createOptimisticLockOrder(id);
    return new ResponseBean(HttpStatus.OK.value(), "购买成功", orderCount);
}

在Service添加方法

ISeckillEvolutionService

/**
 * 使用乐观锁创建订单(解决卖超问题)
 *
 * @param id
 * @return java.lang.Integer
 * @throws Exception
 * @author wliduo[i@dolyw.com]
 * @date 2019/11/22 14:21
 */
Integer createOptimisticLockOrder(Integer id) throws Exception;

ISeckillEvolutionService

/**
 * 乐观锁方式(名称注入)
 */
@Autowired
@Qualifier("seckillOptimisticLockService")
private ISeckillService seckillOptimisticLockServiceImpl;

@Override
@Transactional(rollbackFor = Exception.class)
public Integer createOptimisticLockOrder(Integer id) throws Exception {
    // 检查库存
    StockDto stockDto = seckillOptimisticLockServiceImpl.checkStock(id);
    Thread.sleep(10);
    // 扣库存
    Integer saleCount = seckillOptimisticLockServiceImpl.saleStock(stockDto);
    if (saleCount <= 0) {
        throw new CustomException("扣库存失败");
    }
    // 下订单
    Integer orderCount = seckillOptimisticLockServiceImpl.createOrder(stockDto);
    if (saleCount <= 0) {
        throw new CustomException("下订单失败");
    }
    return orderCount;
}

然后创建一个ISeckillService的使用乐观锁方式的实现类提供上面使用

SeckillOptimisticLockServiceImpl

package com.example.seckill.impl;

import ...;

/**
 * 使用乐观锁
 *
 * @author wliduo[i@dolyw.com]
 * @date 2019-11-20 18:03:33
 */
@Service("seckillOptimisticLockService")
public class SeckillOptimisticLockServiceImpl implements ISeckillService {

    @Autowired
    private StockDao stockDao;

    @Autowired
    private StockOrderDao stockOrderDao;

    @Override
    public StockDto checkStock(Integer id) {
        StockDto stockDto = stockDao.selectByPrimaryKey(id);
        if (stockDto.getCount() > 0) {
            return stockDto;
        }
        throw new CustomException("库存不足");
    }

    @Override
    public Integer saleStock(StockDto stockDto) {
        return stockDao.updateByOptimisticLock(stockDto);
    }

    @Override
    public Integer createOrder(StockDto stockDto) {
        StockOrderDto stockOrderDto = new StockOrderDto();
        stockOrderDto.setStockId(stockDto.getId());
        return stockOrderDao.insertSelective(stockOrderDto);
    }
}

乐观锁详细使用方法实现，这里SQL直接写注解上了，没写XML里了，偷个懒

StockDao

/**
 * 乐观锁更新扣减库存
 *
 * @param stockDto
 * @return int
 * @throws
 * @author wliduo[i@dolyw.com]
 * @date 2019/11/22 14:14
 */
@Update("UPDATE t_seckill_stock SET count = count - 1, sale = sale + 1, version = version + 1 " +
        "WHERE id = #{id, jdbcType = INTEGER} AND version = #{version, jdbcType = INTEGER}")
int updateByOptimisticLock(StockDto stockDto);