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

    /**
     * 單號(hào)流水號(hào)yyMMdd前綴
     */
    public static final String SERIAL_YYMMDD_PREFIX = "yyMMdd";

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

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

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

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

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

    /**
     * 單號(hào)前綴
     * 為空時(shí)填""
     */
    private String prefix;

    /**
     * 時(shí)間格式表達(dá)式
     * 例如：yyyyMMdd
     */
    private String datePattern;

    /**
     * 流水號(hào)長度
     */
    private Integer serialLength;
    /**
     * 隨機(jī)數(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 方法
}

/**
 * 單號(hào)生成工具類
 *
 * @author mq
 */
public class FormNoSerialUtil {

    /**
     * 生成單號(hào)前綴
     */
    public static String getFormNoPrefix(FormNoTypeEnum formNoTypeEnum) {
        //格式化時(shí)間
        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)建流水號(hào)緩存Key
     *
     * @param serialPrefix 流水號(hào)前綴
     * @return 流水號(hào)緩存Key
     */
    public static String getCacheKey(String serialPrefix) {
        return FormNoConstants.SERIAL_CACHE_PREFIX.concat(serialPrefix);
    }
    
    /**
     * 補(bǔ)全流水號(hào)
     *
     * @param serialPrefix      單號(hào)前綴
     * @param incrementalSerial 當(dāng)天自增流水號(hào)
     * @author mengqiang
     * @date 2019/1/1
     */
    public static String completionSerial(String serialPrefix, Long incrementalSerial,
                                          FormNoTypeEnum formNoTypeEnum) {
        StringBuffer sb = new StringBuffer(serialPrefix);

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


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

/**
 * 單號(hào)生成接口
 *
 * @author mq
 */
public interface FormNoGenerateService {

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

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