spring cloud中全局唯一ID生成服务Snowflake的使用

 

直接上代码

/**
 * @author tbicf
 */
@Component
public class IdService {
    @Resource
    private Properties properties;

    private static SnowflakeIdWorker snowflakeIdWorker;

    @PostConstruct
    public void idServiceInit() {
        /*
         * 获取当前服务器ip地址
         */
        PrimaryKeyProperties.IpAddress ipAddress = properties.getIpAddress(
                StringUtil.getLocalIp(properties.getStrIp())
        );
        /*
         * 工作机器ID(0~31)
         */
        long workerId = ipAddress.getWorkerId();
        /*
         * 数据中心ID(0~31)
         */
        long datacenterId = ipAddress.getDatacenterId();

        snowflakeIdWorker = new SnowflakeIdWorker(workerId, datacenterId);
    }

    public static String id() {
        return String.valueOf(snowflakeIdWorker.nextId());
    }
}

    public static String getLocalIp(String ipStart) {
        Enumeration<NetworkInterface> netInterfaces;
        try {
            netInterfaces = NetworkInterface.getNetworkInterfaces();
        } catch (SocketException e) {
            return "127.0.0.1";
        }
        while (netInterfaces.hasMoreElements()) {
            NetworkInterface nif = netInterfaces.nextElement();
            Enumeration<InetAddress> inetAddress = nif.getInetAddresses();
            while (inetAddress.hasMoreElements()) {
                try {
                    String ip = inetAddress.nextElement().getHostAddress();
                    if (ip.startsWith(ipStart)) {
                        return ip;
                    }
                } catch (Exception ignore) {
                }
            }
        }
        return "127.0.0.1";
    }

snowflakeIdWorker就不啰嗦了

配置文件

@Data
@Component
@ConfigurationProperties(prefix = "spring.primary")
public class PrimaryKeyProperties {

    private List<IpAddress> ip;

    private String ipStart;

    public IpAddress getIpAddress(String ipAddress) {
        return this.getIp().get(find(ipAddress));
    }

    private int find(String ipAddress) {
        for (int i = 0; i < this.ip.size(); i++) {
            if (ip.get(i).address.equals(ipAddress)) {
                return i;
            }
        }
        return 0;
    }

    @Data
    public static class IpAddress {

        private String address;

        private long workerId;

        private long datacenterId;
    }

}

spring:
    primary:
            ip:
              - worker-id: 2
                datacenter-id: 1
                address: 10.1.xxx.xxx
              - worker-id: 3
                datacenter-id: 1
                address: 10.1.xxx.xxx
              - worker-id: 4
                datacenter-id: 1
                address: 10.1.xxx.xxx
            ip-start: 10.1