spring boot雪花算法

原文地址

首先生成雪花id，这样虽然可以直接调用也就是通过new这个类的对象来生成，但是因为使用了spring框架，交给容器管理更好，所以下面就通过配置来实现将类注入到容器中：

import java.security.SecureRandom;

public class SnowflakeManager {
    private static final long EPOCH_STAMP = 1262275200000L;//开始时间戳，可自定义
    private static final long SEQUENCE_BIT = 12L;//序号位 12
    private static final long MACHINE_BIT = 5L;//机器码位 5
    private static final long DATA_CENTER_BIT = 5L;//数据中心 5
    private static final long MAX_SEQUENCE_NUM = -1L ^ (-1L << SEQUENCE_BIT);
    private static final long MAX_MACHINE_NUM = -1L ^ (-1L << MACHINE_BIT);
    private static final long MAX_DATA_CENTER_NUM = -1L ^ (-1L << DATA_CENTER_BIT);
    private static final long MACHINE_LEFT = SEQUENCE_BIT;
    private static final long DATA_CENTER_LEFT = SEQUENCE_BIT + MACHINE_BIT;
    private static final long TIMESTAMP_LEFT = SEQUENCE_BIT + MACHINE_BIT + DATA_CENTER_BIT;
    private final long machineId;//机器ID
    private final long dataCenterId;//数据中心ID
    private long sequence = 0L;
    private long lastTimestamp = -1L;//上次生成ID的时间戳

    public SnowflakeManager(long machineId, long dataCenterId) {
        if (machineId > MAX_MACHINE_NUM || machineId < 0) {
            throw new IllegalArgumentException(String.format("machine id can't be greater than %d or less than 0", MAX_MACHINE_NUM));
        }
        if (dataCenterId > MAX_DATA_CENTER_NUM || dataCenterId < 0) {
            throw new IllegalArgumentException(String.format("data center id can't be greater than %d or less than 0", MAX_DATA_CENTER_NUM));
        }
        this.machineId = machineId;
        this.dataCenterId = dataCenterId;
    }

    public synchronized long nextValue() throws Exception {
        String os = System.getProperty("os.name");
        SecureRandom secureRandom;
        if (os.toLowerCase().startsWith("win")) {
            // windows机器用
            secureRandom = SecureRandom.getInstanceStrong();
        } else {
            // linux机器用
            secureRandom = SecureRandom.getInstance("NativePRNGNonBlocking");
        }
        //SecureRandom secureRandom = SecureRandom.getInstanceStrong();
        long currentTimeMillis = this.currentTimeMillis();
        if(currentTimeMillis < this.lastTimestamp) {
            throw new Exception(String.format("Clock moved backwards. Refusing to generate id for %d milliseconds", (this.lastTimestamp-currentTimeMillis)));
        }

        if(this.lastTimestamp == currentTimeMillis) {
            this.sequence = (this.sequence+1) & MAX_SEQUENCE_NUM;
            if (this.sequence == 0) {
                this.sequence = secureRandom.nextInt(Long.valueOf(SEQUENCE_BIT).intValue());
                currentTimeMillis = this.tilNextMillis(this.lastTimestamp);
            }
        } else {
            this.sequence = secureRandom.nextInt(Long.valueOf(SEQUENCE_BIT).intValue());
        }
        this.lastTimestamp = currentTimeMillis;

        // 64 Bit ID (42(Millis)+5(Data Center ID)+5(Machine ID)+12(Repeat Sequence Summation))
        long nextId = ((currentTimeMillis-EPOCH_STAMP) << TIMESTAMP_LEFT)
                | (this.dataCenterId << DATA_CENTER_LEFT)
                | (this.machineId << MACHINE_LEFT)
                | this.sequence;

        return nextId;
    }

    private long tilNextMillis(long lastTimestamp) {
        long currentTimeMillis = this.currentTimeMillis();
        while (currentTimeMillis <= lastTimestamp) {
            currentTimeMillis = this.currentTimeMillis();
        }
        return currentTimeMillis;
    }

    private long currentTimeMillis() {
        return System.currentTimeMillis();
    }

    public static void main(String[] args) throws Exception {
        SnowflakeManager snowflakeManager = new SnowflakeManager(0L,0L);
        long l = snowflakeManager.nextValue();
        System.out.println(l);
    }
}

• 1、因为生成雪花id时要传入两个long型的参数，所以将这两个参数提到配置文件中来
  application.yml

• com:
    linshan:
      demo1:
        snowflake:
          machine-id: 1
          data-center-id: 1
  

  我的包结构如下：
  

• 2、创建一个类 TestProperties ，来获取配置文件中的信息，一个类 SnowflakeProperties 来存需要传入的两个参数。
  TestProperties

• import com.linshan.demo1.entity.SnowflakeProperties;
  import lombok.Data;
  import org.springframework.boot.context.properties.ConfigurationProperties;
  
  @Data
  @ConfigurationProperties(prefix = "com.linshan.demo1")
  public class TestProperties {
  
      private SnowflakeProperties snowflake;
  
  }
  

  SnowflakeProperties

• import lombok.Getter;
  import lombok.Setter;
  
  @Getter
  @Setter
  public class SnowflakeProperties {
      private long machineId;
      private long dataCenterId;
  }
  

• 3、创建 BossAutoConfiguration 类
• 这里注意的是@EnableConfigurationProperties注解的作用是：使使用 @ConfigurationProperties 注解的类生效。

• import com.linshan.demo1.entity.SnowflakeManager;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
  import org.springframework.boot.context.properties.EnableConfigurationProperties;
  import org.springframework.context.annotation.Bean;
  import org.springframework.context.annotation.Configuration;
  
  @Configuration
  @EnableConfigurationProperties(TestProperties.class)
  public class BossAutoConfiguration {
      @Autowired
      private TestProperties properties;
  
      @Bean
      @ConditionalOnMissingBean
      public SnowflakeManager snowflakeManager() {
          return new SnowflakeManager(this.properties.getSnowflake().getMachineId(), this.properties.getSnowflake().getDataCenterId());
      }
  }
  

• 4、最后就是直接在需要使用的类中注入即可

• //注入
  @Autowired
  private SnowflakeManager snowflakeManager;
  //调用
  long snowflakeID = snowflakeManager.nextValue();