基于mysql的单据号生成（前缀+日期+自增id+后缀）

Posted 2020-11-25 珍惜当下 不负此生

介绍

本次采用mysql处理，性能不是很好，对于高并发有要求的建议不要采用
公司一个小项目，需要生成一个单据号，格式为： 日期 + 每日重新自增号，自己考虑了一下每日自增需要考虑并发和持久问题，两种数据库redis和mysql由于项目较小，所以没有redis因为这个增加一个redis好像有点不值得，所以采用mysql作为持久化处理，一下思路也是借鉴了网上的许多想法

源码

源码查看规则
源码位置： blog-study:module-utils:work-no

欢迎大家随时批评指正

思路

1. 首先根据需求，需要这样一个格式： 20200101 + 00001 的单据号
2. 分析日期只需要获取当前时间格式化即可简单，
   自增号：
   1）需要考虑每日重新开始
   2）宕机数据可以持久化（存库即可）
   3）并发获取单据号唯一
   以上四个问题只需要考虑 每日重新开始计数和并发问题
3. 每日重新开始，在数据库中加一个时间标志，每次获取都与当前时间做对比然后做相应处理， 并发问题我能想到的就是加锁（编程锁或者数据库锁，我这里采用lock锁）
4. 以上基本可以解决项目问题了，然后思考是否可以用于其它业务号的生成
   1）于是想可以增加业务标识、前缀和后缀来满足不同业务，其中业务标识不参与单据号组成只是标识某一类单据号，这样可以保证不同业务可以使用相同的单据号，前缀是根据业务需要标识业务单据号的参与单据号生成， 后缀的作用是干扰随机数，可以尽可能防止他人直接看出来业务单据每日的生成数量
   格式： GX + 20200101 + 00001 + 23
   2） 每日重新开始计数，是否可以改为可选择的，每月重新计数，每年重新计数，或者一直不需要重新计数
5. 思考是否可以封装为组件，引入依赖直接就可以使用，想了一下自己目前还不具备这种能力吧，以后还要多学多了解，也希望大家可以多多指点，小子将不胜感激。

功能介绍

1. 数据库结构（使用实体类代替）

/**
     * 业务唯一标识(只是作为唯一标识，并不参与单据号的生成)
     */
    private String workNo;
    /**
     * 业务码前缀
     */
    private String prefix;
    /**
     * 序号
     */
    private Integer serialNum;
    /**
     * 序号长度
     */
    private Integer serialLength;
    /**
     * 序号之后的随机数长度（干扰串）
     */
    private Integer randomLength;
    /**
     * 重置模式（0：无需重置，1： 按天重置， 2： 按月重置， 3： 按年重置）
     */
    private Integer restType;
    /**
     * 重置时间（检测是否需要重置使用）
     */
    private LocalDate restTime;
    /**
     * 创建时间
     */
    private LocalDateTime createTime;
    /**
     * 修改时间
     */
    private LocalDateTime updateTime;
    /**
     * 备注
     */
    private String remark;

2. 业务号格式使用枚举规定好，每次启动项目自动识别处理

/**
     * 为了保证每次生成的单号一致性，所以初始化作为第一次初始化成功后不再修改
     */
    ORDER_NO("XG", "XG", 5, 2, 1, "小郭测试单据号");



    /**
     * 业务唯一标识(只是作为唯一标识，并不参与单据号的生成)
     */
    private String workNo;
    /**
     * 业务码前缀
     */
    private String prefix;
    /**
     * 序号长度
     */
    private Integer serialLength;
    /**
     * 序号之后的随机数长度（干扰串）0代表没有干扰
     */
    private Integer randomLength;
    /**
     * 重置模式（0：无需重置，1： 按天重置， 2： 按月重置， 3： 按年重置）
     */
    private Integer restType;
    /**
     * 备注
     */
    private String remark;

项目启动初始化：

@Configuration
public class WorkNoInit implements ApplicationRunner {
    @Autowired
    private WorkNoDao workNoDao;

    /**
     * 单据号初始化
     *
     * @param args
     * @throws Exception
     */
    @Override
    public void run(ApplicationArguments args) throws Exception {
        WorkNoInfo workNoInfo = null;
        for (WorkInfoEnum workInfoEnum : WorkInfoEnum.values()) {
            workNoInfo = new WorkNoInfo(workInfoEnum);
            //查询是否已有该业务单据号（没有初始化）
            if (workNoDao.isHaveWorkNo(workNoInfo.getWorkNo()) == 0) {
                workNoDao.addOrderNumInfo(workNoInfo);
            }
        }
    }
}

3. 获取单据号：

@Component
public class WorkNoService {

    @Autowired
    private WorkNoDao workNoDao;

    private static Lock lock = new ReentrantLock();

    /**
     * 获取业务序列码
     *
     * 流程：
     * 1. 加锁保证线程安全
     * 2. 查询数据库中的业务信息
     * 3. 判断重置模式修改数据库中相应的数据
     * 4. 返回业务码
     *
     * @param workInfoEnum 枚举
     * @return 业务序列码
     */
    public String getOrderNo(WorkInfoEnum workInfoEnum) {
        lock.lock();
        LocalDateTime localDateTime = LocalDateTime.now();
        try {
            WorkNoInfo workNoInfo = workNoDao.queryByWorkNo(workInfoEnum.getWorkNo());
            if (workNoInfo == null) {
                throw new ErrorCodeException(500, "查询生成业务码基础数据失败！");
            }
            //序号
            Integer serialNum = workNoInfo.getSerialNum();
            //判断是否需要重置序号
            if (workNoInfo.getRestType() == 1) {
                String dayTime = localDateTime.format(DateTimeFormatter.ofPattern("yyyyMMdd"));
                String dataBaseTime = workNoInfo.getRestTime().format(DateTimeFormatter.ofPattern("yyyyMMdd"));
                if (!dataBaseTime.equals(dayTime)) {
                    serialNum = 0;
                }
            } else if (workNoInfo.getRestType() == 2) {
                String monthTime = localDateTime.format(DateTimeFormatter.ofPattern("yyyyMM"));
                String dataBaseTime = workNoInfo.getRestTime().format(DateTimeFormatter.ofPattern("yyyyMM"));
                if (!dataBaseTime.equals(monthTime)) {
                    serialNum = 0;
                }
            } else if (workNoInfo.getRestType() == 3) {
                String yearTime = localDateTime.format(DateTimeFormatter.ofPattern("yyyy"));
                String dataBaseTime = workNoInfo.getRestTime().format(DateTimeFormatter.ofPattern("yyyy"));
                if (!dataBaseTime.equals(yearTime)) {
                    serialNum = 0;
                }
            }
            workNoInfo.setSerialNum(serialNum + 1);
            workNoInfo.setRestTime(localDateTime.toLocalDate());
            //更新数据库
            workNoDao.updateOrderNo(workNoInfo);
            return workNoInfo.getNo();
        } catch (Exception e) {
           throw new ErrorCodeException(500, "生成业务码失败" + e.getMessage());
        } finally {
            lock.unlock();
        }
    }

}