0
点赞
收藏
分享

微信扫一扫

使用redis 加锁的方式生成订单号


建一个订单对象

/**
* 订单号 对象
*/
@Data
public class OrderNumberVo implements Serializable {

/**
* 日期字符串 格式:"yyyyMMdd"
*/
private String meetDate;

/**
* 计数id
*/
private Integer id;

/**
* 订单号 格式:"yyyyMMddXXX1"
*/
private String orderNumber;

}

业务接口中定义缓存名称

public interface IReserveOrderService extends IService<ReserveOrder> {

public static String CACHE_ORDERNUMBER="ordernumber_cache";

String createOrderNumber();
}

业务类中实现创建订单号方法

@Service
@Slf4j
public class ReserveOrderServiceImpl extends ServiceImpl<ReserveOrderMapper, ReserveOrder> implements IReserveOrderService {

@Resource
RedisUtil redisUtil;

private static AtomicInteger atomicInteger=new AtomicInteger();

/**
* 生成订单号
*
* @return
*/
@Override
public String createOrderNumber() {
SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMdd");
String format = dateFormat.format(new Date());
Integer id = 0;
String orderNumber = "";
while (true) {
OrderNumberVo vo = this.getCache();
if (vo == null) {
//此时没有缓存,从数据库中读取当天的订单
ReserveOrder order = this.getOne(new LambdaQueryWrapper<ReserveOrder>().likeRight(ReserveOrder::getOrderNumber, format).last("limit 1"));
if (order != null) {
id = Integer.valueOf(order.getOrderNumber().substring(format.length()));
}
vo = new OrderNumberVo();
} else {
id = vo.getId();
}
if(atomicInteger.get()==0) {
atomicInteger.set(id++);
}
orderNumber = String.format("%s%04d", format, atomicInteger.incrementAndGet());
vo.setMeetDate(format);
vo.setOrderNumber(orderNumber);

OrderNumberVo cache = this.getCache();
if (cache==null || ( cache!=null && cache.getId() == vo.getId())) {
//如果缓存中的id 和 前边获取到的vo对象中id一致,就更新缓存,中断循环
vo.setId(atomicInteger.get());//更新id
this.setCache(vo);
break;
}
}

return orderNumber;
}

private OrderNumberVo getCache() {
OrderNumberVo vo = (OrderNumberVo) redisUtil.get(IReserveOrderService.CACHE_ORDERNUMBER);
return vo;
}

private void setCache(OrderNumberVo vo) {
redisUtil.set(IReserveOrderService.CACHE_ORDERNUMBER, vo);
}
}

单元测试

public void testOrderNumber(){
for(int i=0;i<20;i++){
new Thread(()->{
String orderNumber = orderService.createOrderNumber();
log.info("订单号 = {}",orderNumber);
}).start();
}
}

说明:在redis中保存业务订单对象,每次需要创建业务号时,先获取这个对象,在这个对象中使用原子操作AtomicInteger 生成新的业务号,写入缓存时,判断这个对象与缓存中保存的订单对象id是否一致,不一致时说明已经有另外的操作在生成业务号期间写入了新的业务号,这个时候需要再次重复操作,直到能正确写入缓存为止,表示生成的业务号是唯一的。

经几轮单元测试,每次都能正确生成不重复的业务号,暂时假定这个方法是正确的,还有待实际业务检验。数据表中要将业务号字段设置成唯一约束,避免万一情况。


举报

相关推荐

0 条评论