gaoluyang
6 天以前 8168aef3c7557c9e1361892270a1ebe35dbddba5
yudao-module-erp/src/main/java/cn/iocoder/yudao/module/erp/dal/redis/no/ErpNoRedisDAO.java
@@ -9,6 +9,7 @@
import java.time.Duration;
import java.time.LocalDateTime;
import java.util.function.Predicate;
/**
@@ -20,53 +21,24 @@
public class ErpNoRedisDAO {
    /**
     * 其它入库 {@link cn.iocoder.yudao.module.erp.dal.dataobject.stock.ErpStockInDO}
     */
    public static final String STOCK_IN_NO_PREFIX = "QTRK";
    /**
     * 其它出库 {@link cn.iocoder.yudao.module.erp.dal.dataobject.stock.ErpStockOutDO}
     */
    public static final String STOCK_OUT_NO_PREFIX = "QCKD";
    /**
     * 库存调拨 {@link cn.iocoder.yudao.module.erp.dal.dataobject.stock.ErpStockMoveDO}
     */
    public static final String STOCK_MOVE_NO_PREFIX = "QCDB";
    /**
     * 库存盘点 {@link cn.iocoder.yudao.module.erp.dal.dataobject.stock.ErpStockCheckDO}
     */
    public static final String STOCK_CHECK_NO_PREFIX = "QCPD";
    /**
     * 销售订单 {@link cn.iocoder.yudao.module.erp.dal.dataobject.sale.ErpSaleOrderDO}
     */
    public static final String SALE_ORDER_NO_PREFIX = "XSDD";
    /**
     * 销售出库 {@link cn.iocoder.yudao.module.erp.dal.dataobject.sale.ErpSaleOutDO}
     */
    public static final String SALE_OUT_NO_PREFIX = "XSCK";
    /**
     * 销售退货 {@link cn.iocoder.yudao.module.erp.dal.dataobject.sale.ErpSaleReturnDO}
     */
    public static final String SALE_RETURN_NO_PREFIX = "XSTH";
    /**
     * 采购订单 {@link cn.iocoder.yudao.module.erp.dal.dataobject.purchase.ErpPurchaseOrderDO}
     */
    public static final String PURCHASE_ORDER_NO_PREFIX = "CGDD";
    /**
     * 采购申请 {@link cn.iocoder.yudao.module.erp.dal.dataobject.purchase.ErpPurchaseRequestDO}
     */
    public static final String PURCHASE_REQUEST_NO_PREFIX = "CGSQ";
    /**
     * 采购入库 {@link cn.iocoder.yudao.module.erp.dal.dataobject.purchase.ErpPurchaseInDO}
     * 采购来票 {@link cn.iocoder.yudao.module.erp.dal.dataobject.purchase.ErpPurchaseInvoiceDO}
     */
    public static final String PURCHASE_IN_NO_PREFIX = "CGRK";
    /**
     * 采购退货 {@link cn.iocoder.yudao.module.erp.dal.dataobject.purchase.ErpPurchaseReturnDO}
     */
    public static final String PURCHASE_RETURN_NO_PREFIX = "CGTH";
    public static final String PURCHASE_INVOICE_NO_PREFIX = "FPD";
    /**
     * 付款单 {@link cn.iocoder.yudao.module.erp.dal.dataobject.finance.ErpFinancePaymentDO}
@@ -82,7 +54,7 @@
    /**
     * 生成序号,使用当前日期,格式为 {PREFIX} + yyyyMMdd + 6 位自增
     * 例如说:QTRK 202109 000001 (没有中间空格)
     * 例如说:CGDD 202109 000001 (没有中间空格)
     *
     * @param prefix 前缀
     * @return 序号
@@ -97,4 +69,26 @@
        return noPrefix + String.format("%06d", no);
    }
    /**
     * 生成序号,并跳过已存在的单号。
     *
     * Redis 中的自增计数一旦丢失(如 Redis 重启未持久化、手工清库),计数会从 1 重新开始,
     * 而当天数据库中可能已有同日期后缀的单号,此时逐个递增重试直到取到未占用的序号。
     *
     * @param prefix  前缀
     * @param isNoExists 单号是否已存在
     * @return 未占用的序号
     */
    public String generate(String prefix, Predicate<String> isNoExists) {
        for (int i = 0; i < MAX_GENERATE_ATTEMPTS; i++) {
            String no = generate(prefix);
            if (!isNoExists.test(no)) {
                return no;
            }
        }
        throw new IllegalStateException("生成单号失败:连续 " + MAX_GENERATE_ATTEMPTS + " 次尝试均已被占用,请稍后重试");
    }
    private static final int MAX_GENERATE_ATTEMPTS = 10000;
}