/**
 * 單號生成常量
 *
 * @author mq
 */
public class FormNoConstants {
    /**
     * 單號流水號緩存Key前綴
     */
    public static final String SERIAL_CACHE_PREFIX = "FORM_NO_CACHE_";

    /**
     * 單號流水號yyMMdd前綴
     */
    public static final String SERIAL_YYMMDD_PREFIX = "yyMMdd";

    /**
     * 單號流水號yyyyMMdd前綴
     */
    public static final String SERIAL_YYYYMMDD_PREFIX = "yyyyMMdd";
    
    /**
     * 默認緩存天數(shù)
     */
    public static final int DEFAULT_CACHE_DAYS = 7;
}

/**
 * 單號生成類型枚舉
 *
 * @author mq
 * 注：隨機號位于流水號之后,流水號使用redis計數(shù)據(jù)，每天都是一個新的key,長度不足時則自動補0
 * <p>
 * 生成規(guī)則 =固定前綴+當(dāng)天日期串+流水號(redis自增，不足長度則補0)+隨機數(shù)
 */
public enum FormNoTypeEnum {

    /**
     * 應(yīng)付單單號：
     * 固定前綴：YF
     * 時間格式：yyyyMMdd
     * 流水號長度：7(當(dāng)單日單據(jù)較多時可根據(jù)業(yè)務(wù)適當(dāng)增加流水號長度)
     * 隨機數(shù)長度：3
     * 總長度：20
     */
    YF_ORDER("YF", FormNoConstants.SERIAL_YYYYMMDD_PREFIX, 7, 3, 20),

    /**
     * 付款單單號：
     * 固定前綴：FK
     * 時間格式：yyyyMMdd
     * 流水號長度：7
     * 隨機數(shù)長度：3
     * 總長度：20
     */
    FK_ORDER("FK", FormNoConstants.SERIAL_YYYYMMDD_PREFIX, 7, 3, 20),

    /**
     * 測試單單號：
     * 固定前綴：""
     * 時間格式：yyyyMMdd
     * 流水號長度：10
     * 隨機數(shù)長度：0
     * 總長度：20
     */
    TEST_ORDER("te", FormNoConstants.SERIAL_YYYYMMDD_PREFIX, 10, 0, 20),
    ;

    /**
     * 單號前綴
     * 為空時填""
     */
    private String prefix;

    /**
     * 時間格式表達式
     * 例如：yyyyMMdd
     */
    private String datePattern;

    /**
     * 流水號長度
     */
    private Integer serialLength;
    /**
     * 隨機數(shù)長度
     */
    private Integer randomLength;

    /**
     * 總長度
     */
    private Integer totalLength;


    FormNoTypeEnum(String prefix, String datePattern, Integer serialLength, Integer randomLength, Integer totalLength) {
        this.prefix = prefix;
        this.datePattern = datePattern;
        this.serialLength = serialLength;
        this.randomLength = randomLength;
        this.totalLength = totalLength;
    }
    //省略 get 方法
}

/**
 * 單號生成工具類
 *
 * @author mq
 */
public class FormNoSerialUtil {

    /**
     * 生成單號前綴
     */
    public static String getFormNoPrefix(FormNoTypeEnum formNoTypeEnum) {
        //格式化時間
        DateTimeFormatter formatter = DateTimeFormatter.ofPattern(formNoTypeEnum.getDatePattern());
        StringBuffer sb = new StringBuffer();
        sb.append(formNoTypeEnum.getPrefix());
        sb.append(formatter.format(LocalDateTime.now()));
        return sb.toString();
    }
    
    /**
     * 構(gòu)建流水號緩存Key
     *
     * @param serialPrefix 流水號前綴
     * @return 流水號緩存Key
     */
    public static String getCacheKey(String serialPrefix) {
        return FormNoConstants.SERIAL_CACHE_PREFIX.concat(serialPrefix);
    }
    
    /**
     * 補全流水號
     *
     * @param serialPrefix      單號前綴
     * @param incrementalSerial 當(dāng)天自增流水號
     * @author mengqiang
     * @date 2019/1/1
     */
    public static String completionSerial(String serialPrefix, Long incrementalSerial,
                                          FormNoTypeEnum formNoTypeEnum) {
        StringBuffer sb = new StringBuffer(serialPrefix);

        //需要補0的長度=流水號長度 -當(dāng)日自增計數(shù)長度
        int length = formNoTypeEnum.getSerialLength() - String.valueOf(incrementalSerial).length();
        //補零
        for (int i = 0; i < length; i++) {
            sb.append("0");
        }
        //redis當(dāng)日自增數(shù)
        sb.append(incrementalSerial);
        return sb.toString();
    }


    /**
     * 補全隨機數(shù)
     *
     * @param serialWithPrefix 當(dāng)前單號
     * @param formNoTypeEnum   單號生成枚舉
     * @author mengqiang
     * @date 2019/1/1
     */
    public static String completionRandom(String serialWithPrefix, FormNoTypeEnum formNoTypeEnum) {
        StringBuffer sb = new StringBuffer(serialWithPrefix);
        //隨機數(shù)長度
        int length = formNoTypeEnum.getRandomLength();
        if (length > 0) {
            Random random = new Random();
            for (int i = 0; i < length; i++) {
                //十以內(nèi)隨機數(shù)補全
                sb.append(random.nextInt(10));
            }
        }
        return sb.toString();
    }
}

/**
 * 單號生成接口
 *
 * @author mq
 */
public interface FormNoGenerateService {

    /**
     * 根據(jù)單據(jù)編號類型 生成單據(jù)編號
     *
     * @param formNoTypeEnum 單據(jù)編號類型
     * @author mengqiang
     * @date 2019/1/1
     */
    String generateFormNo(FormNoTypeEnum formNoTypeEnum);
}

/**
 * 單號生成接口實現(xiàn)
 *
 * @author mengqiang
 * @version FormNoGenerateServiceImpl.java, v 1.0 2019-01-01 18:10
 */
@Service
public class FormNoGenerateServiceImpl implements FormNoGenerateService {
    /**
     * redis 服務(wù)
     * demo 項目沒有加redis相關(guān)，若有需要請參考，redis的博客
     */
    @Autowired
    private RedisCache redisCache;
    /**
     * 根據(jù)單據(jù)編號類型 生成單據(jù)編號
     *
     * @param formNoTypeEnum 單據(jù)編號類型
     * @author mengqiang
     * @date 2019/1/1
     */
    @Override
    public String generateFormNo(FormNoTypeEnum formNoTypeEnum) {
        //獲得單號前綴
        //格式 固定前綴 +時間前綴 示例 ：YF20190101
        String formNoPrefix = FormNoSerialUtil.getFormNoPrefix(formNoTypeEnum);
        //獲得緩存key
        String cacheKey = FormNoSerialUtil.getCacheKey(formNoPrefix);
        //獲得當(dāng)日自增數(shù)
        Long incrementalSerial = redisCache.incr(cacheKey);
        //設(shè)置失效時間 7天
        redisCache.expire(cacheKey, FormNoConstants.DEFAULT_CACHE_DAYS, TimeUnit.DAYS);
        //組合單號并補全流水號
        String serialWithPrefix = FormNoSerialUtil
                .completionSerial(formNoPrefix, incrementalSerial, formNoTypeEnum);
        //補全隨機數(shù)
        return FormNoSerialUtil.completionRandom(serialWithPrefix, formNoTypeEnum);
    }
}