分布式雪花算法生成唯一ID

本文主要是介绍分布式雪花算法生成唯一ID，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

Hutool工具包

<dependency>
    <groupId>cn.hutool</groupId>
    <artifactId>hutool-all</artifactId>
    <version>5.7.16</version>
</dependency>

Snowflake需自行保证单例，否则多个对象生成ID会导致重复

代码

package com.lyods.base.utils;

import cn.hutool.core.lang.Snowflake;
import cn.hutool.core.net.NetUtil;

import javax.annotation.PostConstruct;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * 
 * @creator linhy
 * @date 2021-11-16
 * @descrption 获取雪花算法ID
 *
 */
public class SnowFlakeUtil {

    private final static Logger log = LoggerFactory.getLogger(SnowFlakeUtil.class);

    private static long workerId = 0;

    private volatile static Snowflake snowflake = null;
    private volatile static SnowFlakeUtil sfu = null;

    /**
     * 单例
     */
    private SnowFlakeUtil() {
    }

    public static SnowFlakeUtil getInstance() {
        if (null == sfu) {
            synchronized (SnowFlakeUtil.class) {
                if (null == sfu) {
                    sfu = new SnowFlakeUtil();
                }
            }
        }
        return sfu;
    }

    /**
     * 获取ID
     * @return
     */
    public long snowflakeId() {
        if (null == snowflake) {
            snowflake = getSnowflake();
        }
        return snowflake.nextId();
    }

    /**
     * Snowflake对象获取
     * @return
     */
    private synchronized Snowflake getSnowflake() {
        snowflake = new Snowflake();
        return snowflake;
    }

    @PostConstruct
    public void init() {
        try {
            workerId = NetUtil.ipv4ToLong(NetUtil.getLocalhostStr());
            log.info("当前机器的workId: {}", workerId);
        } catch (Exception e) {
            e.printStackTrace();
            log.error("当前机器的workId获取失败", e);
            workerId = NetUtil.getLocalhostStr().hashCode();
        }
    }

}

线程池30线程测试

ExecutorService executorService = Executors.newFixedThreadPool(5);
        Stream.iterate(0, x -> x + 1).limit(30).forEach(x -> {
            executorService.submit(() -> {
                long id = SnowFlakeUtil.getInstance().snowflakeId();
                System.out.println(id);
            });
        });
        executorService.shutdown();

结果

这篇关于分布式雪花算法生成唯一ID的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！