未对线程安全进行校验(因为后面发现有方法可以直接调用,没有使用自己写的)
package com.tiansu.nts.cds.service.util;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import lombok.extern.slf4j.Slf4j;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.Set;
import java.util.concurrent.atomic.AtomicInteger;
@Slf4j
public class OrderUtils {
/**
* 单编码前缀
*/
private static final String ORDER_CODE_PREFIX = "YSGD";
/**
* redis key前缀
*/
private static final String KEY_PREFIX = "ORDER:NO";
/**
* 最大订单号
*/
private static final Integer MAX_NO = 9999;
/**
* 自增变量
*/
public static final AtomicInteger inc = new AtomicInteger(0);
/**
* 生成订单编码
*
* @return 订单编码
*/
public static String generateOrderCode() {
// 获取当前日期
String currentDate = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));
// 4位数字递增序列号
String orderNo;
// 校验当天是否已生成工单编码
String value = XXXRedisUtils.strRedisTemplate().opsForValue().get(genKey(currentDate));
if (StrUtil.isBlank(value)) {
// 当天还没有生成key
// 删除之前的key
Set<String> keySet = XXXRedisUtils.strRedisTemplate().keys(KEY_PREFIX);
if (CollUtil.isNotEmpty(keySet)) {
keySet.forEach(key -> XXXRedisUtils.strRedisTemplate().delete(genKey(key)));
}
// 生成当天的第一个订单编号
inc.set(1);
orderNo = genOrderNo(currentDate, inc.get());
} else {
// 当天已有订单编号生成
int num = inc.get();
if (num > MAX_NO) {
// 提示:已超过当天运送的负载,请明天再下单或者联系管理员下单
}
orderNo = genOrderNo(currentDate, num);
}
return ORDER_CODE_PREFIX + currentDate + orderNo;
}
/**
* 生成订单序号
*
* @param currentDate 当前日期
* @param num 订单号
* @return 订单序号
*/
private static String genOrderNo(String currentDate, Integer num) {
String orderNo = String.format("%04d", num);
XXXRedisUtils.strRedisTemplate().opsForValue().set(genKey(currentDate), num.toString());
// 订单序号自增
inc.incrementAndGet();
return orderNo;
}
/**
* 生成key
*
* @param currentDate 当前日期
* @return redis key
*/
public static String genKey(String currentDate) {
return XXXRedisUtils.generateKey(KEY_PREFIX, currentDate);
}
private OrderUtils() {
}
}