多线程安全的编号生成器

package com.finance.common.core.utils;

import com.finance.common.core.constant.NoPrefixConstants;
import com.finance.common.core.text.UUID;
import cn.hutool.core.date.DateUtil;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * ID生成器工具类
 */
public class IdUtils {

    private static final AtomicInteger MARK = new AtomicInteger(0);
    private static final int MAX = 99999;

    /**
     * 获取自定义前缀的编号
     *
     * 【编号生成规则:SXSQ+YYYYMMDD(年月日)+5位流水号】
     */
    public static String getCusNo(String prefix) {
        String time = DateUtil.format(DateUtil.date(), "yyyyMMddHHmmss");
        // 截取时间后5位
        int last5 = Integer.parseInt(time.substring(time.length() - 6, time.length() - 1));
        int currentMark;

        do {
            currentMark = MARK.get();
            if (currentMark >= last5) {
                last5 = currentMark + 1;
            }
        } while (!MARK.compareAndSet(currentMark, last5));

        if (last5 > MAX) {
            throw new RuntimeException("编号已经达到最大值");
        }

        return " (Thread ID: " + Thread.currentThread().getId() + ")" + prefix + time.substring(0, time.length() - 6) + last5;
    }

    public static void main(String[] args) {
        Runnable runnable = () -> {
            for (int i = 0; i < 10; i++) {
                System.out.println(getCusNo(NoPrefixConstants.CREDIT_APPLY));
            }
        };

        // 创建多个线程执行getCusNo方法
        Thread thread1 = new Thread(runnable);
        Thread thread2 = new Thread(runnable);
        Thread thread3 = new Thread(runnable);

        // 启动线程
        thread1.start();
        thread2.start();
        thread3.start();
    }
}

测试结果:
在这里插入图片描述

public synchronized String nextId() { long timestamp = timeGen(); //获取当前毫秒数 //如果服务器时间有问题(时钟后退) 报错。 if (timestamp < lastTimestamp) { throw new RuntimeException(String.format( "Clock moved backwards. Refusing to generate id for %d milliseconds", lastTimestamp - timestamp)); } //如果上次生成时间和当前时间相同,在同一毫秒内 if (lastTimestamp == timestamp) { //sequence自增,因为sequence只有12bit,所以和sequenceMask相与一下,去掉高位 sequence = (sequence + 1) & sequenceMask; //判断是否溢出,也就是每毫秒内超过4095,当为4096时,与sequenceMask相与,sequence就等于0 if (sequence == 0) { timestamp = tilNextMillis(lastTimestamp); //自旋等待到下一毫秒 } } else { sequence = 0L; //如果和上次生成时间不同,重置sequence,就是下一毫秒开始,sequence计数重新从0开始累加 } lastTimestamp = timestamp; long suffix = (datacenterId << datacenterIdShift) | (workerId << workerIdShift) | sequence; String datePrefix = DateFormatUtils.format(timestamp, "yyyyMMddHHMMssSSS"); return datePrefix + suffix; } protected long tilNextMillis(long lastTimestamp) { long timestamp = timeGen(); while (timestamp <= lastTimestamp) { timestamp = timeGen(); } return timestamp; } protected long timeGen() { return System.currentTimeMillis(); } private byte getLastIP(){ byte lastip = 0; try{ InetAddress ip = InetAddress.getLocalHost(); byte[] ipByte = ip.getAddress(); lastip = ipByte[ipByte.length - 1]; } catch (UnknownHostException e) { e.printStackTrace(); } return lastip; }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值