MyBatis使用雪花ID的實現(xiàn)

更新時間：2022年04月07日 12:02:30 作者：陳琰AC

本文主要介紹了MyBatis使用雪花ID的實現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

一、實現(xiàn)MyBatis ID構(gòu)建接口

@Slf4j
@Component
public class CustomIdGenerator implements IdentifierGenerator {

    @Override
    public Long nextId(Object entity) {
        //生成ID
        long id = SnowFlakeUtils.nextId();
        log.info("生成ID: " + id);
        return id;
    }
}

二、雪花ID生成工具類

@Slf4j
public class SnowFlakeUtils {

    /** 初始偏移時間戳 */
    private static final long OFFSET = 1546300800L;

    /** 機(jī)器id (0~15 保留 16~31作為備份機(jī)器) */
    private static final long WORKER_ID;
    /** 機(jī)器id所占位數(shù) (5bit, 支持最大機(jī)器數(shù) 2^5 = 32)*/
    private static final long WORKER_ID_BITS = 5L;
    /** 自增序列所占位數(shù) (16bit, 支持最大每秒生成 2^16 = 65536) */
    private static final long SEQUENCE_ID_BITS = 16L;
    /** 機(jī)器id偏移位數(shù) */
    private static final long WORKER_SHIFT_BITS = SEQUENCE_ID_BITS;
    /** 自增序列偏移位數(shù) */
    private static final long OFFSET_SHIFT_BITS = SEQUENCE_ID_BITS + WORKER_ID_BITS;
    /** 機(jī)器標(biāo)識最大值 (2^5 / 2 - 1 = 15) */
    private static final long WORKER_ID_MAX = ((1 << WORKER_ID_BITS) - 1) >> 1;
    /** 備份機(jī)器ID開始位置 (2^5 / 2 = 16) */
    private static final long BACK_WORKER_ID_BEGIN = (1 << WORKER_ID_BITS) >> 1;
    /** 自增序列最大值 (2^16 - 1 = 65535) */
    private static final long SEQUENCE_MAX = (1 << SEQUENCE_ID_BITS) - 1;
    /** 發(fā)生時間回?fù)軙r容忍的最大回?fù)軙r間 (秒) */
    private static final long BACK_TIME_MAX = 1000L;

    /** 上次生成ID的時間戳 (秒) */
    private static long lastTimestamp = 0L;
    /** 當(dāng)前秒內(nèi)序列 (2^16)*/
    private static long sequence = 0L;
    /** 備份機(jī)器上次生成ID的時間戳 (秒) */
    private static long lastTimestampBak = 0L;
    /** 備份機(jī)器當(dāng)前秒內(nèi)序列 (2^16)*/
    private static long sequenceBak = 0L;

    static {
        // 初始化機(jī)器ID
        long workerId = getWorkId();
        if (workerId < 0 || workerId > WORKER_ID_MAX) {
            throw new IllegalArgumentException(String.format("cmallshop.workerId范圍: 0 ~ %d 目前: %d", WORKER_ID_MAX, workerId));
        }
        WORKER_ID = workerId;
    }

    private static Long getWorkId(){
        try {
            String hostAddress = Inet4Address.getLocalHost().getHostAddress();
            int[] ints = StringUtils.toCodePoints(hostAddress);
            int sums = 0;
            for(int b : ints){
                sums += b;
            }
            return (long)(sums % WORKER_ID_MAX);
        } catch (UnknownHostException e) {
            // 如果獲取失敗，則使用隨機(jī)數(shù)備用
            return RandomUtils.nextLong(0,WORKER_ID_MAX-1);
        }
    }


    /** 私有構(gòu)造函數(shù)禁止外部訪問 */
    private SnowFlakeUtils() {}

    /**
     * 獲取自增序列
     * @return long
     */
    public static long nextId() {
        return nextId(SystemClock.now() / 1000);
    }

    /**
     * 主機(jī)器自增序列
     * @param timestamp 當(dāng)前Unix時間戳
     * @return long
     */
    private static synchronized long nextId(long timestamp) {
        // 時鐘回?fù)軝z查
        if (timestamp < lastTimestamp) {
            // 發(fā)生時鐘回?fù)?
            log.warn("時鐘回?fù)? 啟用備份機(jī)器ID: now: [{}] last: [{}]", timestamp, lastTimestamp);
            return nextIdBackup(timestamp);
        }

        // 開始下一秒
        if (timestamp != lastTimestamp) {
            lastTimestamp = timestamp;
            sequence = 0L;
        }
        if (0L == (++sequence & SEQUENCE_MAX)) {
            // 秒內(nèi)序列用盡
//            log.warn("秒內(nèi)[{}]序列用盡, 啟用備份機(jī)器ID序列", timestamp);
            sequence--;
            return nextIdBackup(timestamp);
        }

        return ((timestamp - OFFSET) << OFFSET_SHIFT_BITS) | (WORKER_ID << WORKER_SHIFT_BITS) | sequence;
    }

    /**
     * 備份機(jī)器自增序列
     * @param timestamp timestamp 當(dāng)前Unix時間戳
     * @return long
     */
    private static long nextIdBackup(long timestamp) {
        if (timestamp < lastTimestampBak) {
            if (lastTimestampBak - SystemClock.now() / 1000 <= BACK_TIME_MAX) {
                timestamp = lastTimestampBak;
            } else {
                throw new RuntimeException(String.format("時鐘回?fù)? now: [%d] last: [%d]", timestamp, lastTimestampBak));
            }
        }

        if (timestamp != lastTimestampBak) {
            lastTimestampBak = timestamp;
            sequenceBak = 0L;
        }

        if (0L == (++sequenceBak & SEQUENCE_MAX)) {
            // 秒內(nèi)序列用盡
//            logger.warn("秒內(nèi)[{}]序列用盡, 備份機(jī)器ID借取下一秒序列", timestamp);
            return nextIdBackup(timestamp + 1);
        }

        return ((timestamp - OFFSET) << OFFSET_SHIFT_BITS) | ((WORKER_ID ^ BACK_WORKER_ID_BEGIN) << WORKER_SHIFT_BITS) | sequenceBak;
    }


    /**
     * 并發(fā)數(shù)
     */
    private static final int THREAD_NUM = 30000;
    private static volatile CountDownLatch countDownLatch = new CountDownLatch(THREAD_NUM);

    public static void main(String[] args) {
        ConcurrentHashMap<Long,Long> map = new ConcurrentHashMap<>(THREAD_NUM);
        List<Long> list = Collections.synchronizedList(new LinkedList<>());

        for (int i = 0; i < THREAD_NUM; i++) {
            Thread thread = new Thread(() -> {
                // 所有的線程在這里等待
                try {
                    countDownLatch.await();

                    Long id = SnowFlakeUtils.nextId();
                    list.add(id);
                    map.put(id,1L);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            });

            thread.start();
            // 啟動后，倒計時計數(shù)器減一，代表有一個線程準(zhǔn)備就緒了
            countDownLatch.countDown();
        }

        try{
            Thread.sleep(50000);
        }catch (Exception e){
            e.printStackTrace();
        }

        System.out.println("listSize:"+list.size());
        System.out.println("mapSize:"+map.size());
        System.out.println(map.size() == THREAD_NUM);
    }
}

到此這篇關(guān)于MyBatis使用雪花ID的實現(xiàn)的文章就介紹到這了,更多相關(guān)MyBatis 雪花ID內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: