ID生成工具类

2021-12-17 22:43:56 浏览数 (1)
代码语言：txt复制
public static void main(String[] args) {
代码语言：txt复制
    IDUtil idUtil = new IDUtil("yyyyMMddHHmmssSSS");//yyyy-MM-dd HH:mm:ss：SSS
代码语言：txt复制
    System.out.println(idUtil.generatedNoByFormatDateAndRandom(5,"5"));
代码语言：txt复制
}

时间格式在创建对象时传入。在集群中使用应传入机器码或唯一标识作为唯一ID的根据。

更新

添加了多个可选的构造方法，来适应不同的生成规则。

IDUtil:

代码语言：txt复制
public class IDUtil {
代码语言：txt复制
    private final Set<Integer> randomSet = new HashSet<>();
代码语言：txt复制
    private Long dateFlag;
代码语言：txt复制
    private int sequence;
代码语言：txt复制
    private final int maxDigits;
代码语言：txt复制
    private static final ThreadLocal<SimpleDateFormat> LOCAL_DATEFORMAT = new ThreadLocal<>();
代码语言：txt复制
    /**
代码语言：txt复制
     *   初始化  默认
     * @author Surpass
     * @date 2021/6/15 16:45
     */
    public IDUtil() {
        LOCAL_DATEFORMAT.set(new SimpleDateFormat("yyyyMMddHHmmssSSS"));
        dateFlag = System.currentTimeMillis();
        this.sequence = 0;
        this.maxDigits = 5;
    }
代码语言：txt复制
    /**
代码语言：txt复制
     *   初始化
     * @param format 时间格式   yyyyMMddHHmmssSSS
     * @author Surpass
     * @date 2021/6/15 16:45
     */
    public IDUtil(String format) {
        LOCAL_DATEFORMAT.set(new SimpleDateFormat(format));
        this.dateFlag = System.currentTimeMillis();
        this.sequence = 0;
        this.maxDigits = 5;
    }
代码语言：txt复制
    /**
代码语言：txt复制
     *   初始化
     * @param sequence 开始累加数字
     * @author Surpass
     * @date 2021/6/15 16:45
     */
    public IDUtil(int sequence) {
        LOCAL_DATEFORMAT.set(new SimpleDateFormat("yyyyMMddHHmmssSSS"));
        dateFlag = System.currentTimeMillis();
        this.sequence = sequence;
        this.maxDigits = 5;
    }
代码语言：txt复制
    public String generatedNoByFormatDate(){
代码语言：txt复制
        return this.generatedNoByFormatDate(null);
代码语言：txt复制
    }
代码语言：txt复制
    public String generatedNoByFormatDateAndNum(int digits){
代码语言：txt复制
        return this.generatedNoByFormatDateAndNum(digits, null);
代码语言：txt复制
    }
代码语言：txt复制
    public String generatedNoByFormatDateAndRandom(int randomDigits){
代码语言：txt复制
        return this.generatedNoByFormatDateAndRandom(randomDigits, null);
代码语言：txt复制
    }
代码语言：txt复制
    public String generatedNoByDateAndPrefix(String prefix){
代码语言：txt复制
        return this.generatedNoByDateAndPrefix(prefix, null);
代码语言：txt复制
    }
代码语言：txt复制
    public String generatedNoByDateAndSuffix(String suffix){
代码语言：txt复制
        return this.generatedNoByDateAndSuffix(suffix, null);
代码语言：txt复制
    }
代码语言：txt复制
    /**
代码语言：txt复制
     *   根据时间获取编号
     * @param machine 机器码
     * @return java.lang.String
     * @author Surpass
     * @date 2021/6/15 16:46
     */
    public String generatedNoByFormatDate(String machine){
        String formatDate = LOCAL_DATEFORMAT.get().format(new Date());
        if (Objects.nonNull(machine)){
            return formatDate   machine;
        }
        LOCAL_DATEFORMAT.remove();
        return formatDate;
    }
代码语言：txt复制
    /**
代码语言：txt复制
     *   根据时间 机器码 后面累加的位数组成的编号
     * @param machine 机器码
     * @param digits 累加值的位数
     * @return java.lang.String
     * @author Surpass
     * @date 2021/6/15 16:46
     */
    public String generatedNoByFormatDateAndNum(int digits, String machine){
        String no = "";
        long l = System.currentTimeMillis();
        Calendar c = Calendar.getInstance();
        c.setTimeInMillis(l);
        Date date=c.getTime();
        String formatDate = LOCAL_DATEFORMAT.get().format(date);
        if (this.dateFlag.equals(l)){
            this.sequence   ;
            StringBuilder sb = new StringBuilder("9");
            for (int i = 1; i < maxDigits; i  ) {
                sb.append("9");
            }
            throw new RuntimeException("sequence out of MAX: "   sb.toString());
        }else {
            this.sequence = 0;
        }
代码语言：txt复制
        if (Objects.nonNull(machine)){
代码语言：txt复制
            no = formatDate   machine   String.format("%0" digits "d", this.sequence);
代码语言：txt复制
        }else {
代码语言：txt复制
            no = formatDate   String.format("%0" digits "d", this.sequence);
代码语言：txt复制
        }
代码语言：txt复制
        LOCAL_DATEFORMAT.remove();
代码语言：txt复制
        return no;
代码语言：txt复制
    }
代码语言：txt复制
    /**
代码语言：txt复制
     *   根据时间 机器码 后面随机数组成的编号
     * @param machine 机器码
     * @param randomDigits 随机数的位数
     * @return java.lang.String
     * @author Surpass
     * @date 2021/6/15 16:46
     */
    public String generatedNoByFormatDateAndRandom(int randomDigits, String machine){
        String no = "";
        long l = System.currentTimeMillis();
        Calendar c = Calendar.getInstance();
        c.setTimeInMillis(l);
        Date date=c.getTime();
        String formatDate = LOCAL_DATEFORMAT.get().format(date);
        StringBuilder sb = new StringBuilder("9");
        for (int i = 1; i < randomDigits; i  ) {
            sb.append("9");
        }
        Random random = new Random();
代码语言：txt复制
        //判断是不是在同一个时间段  不是则修改时间标签和清空随机数Set
代码语言：txt复制
        if (!this.dateFlag.equals(l)){
代码语言：txt复制
            this.dateFlag = l;
代码语言：txt复制
            randomSet.clear();
代码语言：txt复制
        }
代码语言：txt复制
        //判断生成的随机数是不是已经呗用过，如果是则重新生成
代码语言：txt复制
        int randomNum = 0;
代码语言：txt复制
        do {
代码语言：txt复制
            randomNum = random.nextInt(Integer.parseInt(sb.toString()));
代码语言：txt复制
        }while (this.randomSet.contains(randomNum));
代码语言：txt复制
        randomSet.add(randomNum);
代码语言：txt复制
        if (randomSet.size() > Integer.parseInt(sb.toString())) {
代码语言：txt复制
            throw new RuntimeException("sequence out of MAX: "   sb.toString());
代码语言：txt复制
        }
代码语言：txt复制
        if (Objects.nonNull(machine)){
代码语言：txt复制
            no = formatDate   machine   String.format("%0" randomDigits "d", randomNum);
代码语言：txt复制
        }else {
代码语言：txt复制
            no = formatDate   String.format("%0" randomDigits "d", randomNum);
代码语言：txt复制
        }
代码语言：txt复制
        LOCAL_DATEFORMAT.remove();
代码语言：txt复制
        return no;
代码语言：txt复制
    }
代码语言：txt复制
    /**
代码语言：txt复制
     *   根据前缀 时间
     * @param prefix 后缀
     * @param machine 机器码(可省略) null
     * @return java.lang.String
     * @author Surpass
     * @date 2021/6/15 16:46
     */
    public String generatedNoByDateAndPrefix(String prefix, String machine){
        String formatDate = LOCAL_DATEFORMAT.get().format(new Date());
        String no = "";
代码语言：txt复制
        if (Objects.nonNull(machine)){
代码语言：txt复制
            no = prefix   formatDate   machine;
代码语言：txt复制
        }else {
代码语言：txt复制
            no = prefix   formatDate;
代码语言：txt复制
        }
代码语言：txt复制
        LOCAL_DATEFORMAT.remove();
代码语言：txt复制
        return no;
代码语言：txt复制
    }
代码语言：txt复制
    /**
代码语言：txt复制
     *   根据后缀 时间
     * @param suffix 后缀
     * @param machine 机器码(可省略) null
     * @return java.lang.String
     * @author Surpass
     * @date 2021/6/15 16:46
     */
    public String generatedNoByDateAndSuffix(String suffix, String machine){
        String formatDate = LOCAL_DATEFORMAT.get().format(new Date());
        String no = "";
代码语言：txt复制
        if (Objects.nonNull(machine)){
代码语言：txt复制
            no = formatDate   machine   suffix;
代码语言：txt复制
        }else {
代码语言：txt复制
            no = formatDate   suffix;
代码语言：txt复制
        }
代码语言：txt复制
        LOCAL_DATEFORMAT.remove();
代码语言：txt复制
        return no;
代码语言：txt复制
    }
代码语言：txt复制
}
集群

0 人点赞