Go語言實(shí)現(xiàn)Snowflake雪花算法

type Snowflake struct {
 sync.Mutex     // 鎖
 timestamp    int64 // 時(shí)間戳 ，毫秒
 workerid     int64  // 工作節(jié)點(diǎn)
 datacenterid int64 // 數(shù)據(jù)中心機(jī)房id
 sequence     int64 // 序列號(hào)
}

const (
  epoch             = int64(1577808000000)                           // 設(shè)置起始時(shí)間(時(shí)間戳/毫秒)：2020-01-01 00:00:00，有效期69年
 timestampBits     = uint(41)                                       // 時(shí)間戳占用位數(shù)
 datacenteridBits  = uint(2)                                        // 數(shù)據(jù)中心id所占位數(shù)
 workeridBits      = uint(7)                                        // 機(jī)器id所占位數(shù)
 sequenceBits      = uint(12)                                       // 序列所占的位數(shù)
 timestampMax      = int64(-1 ^ (-1  timestampBits))              // 時(shí)間戳最大值
 datacenteridMax   = int64(-1 ^ (-1  datacenteridBits))           // 支持的最大數(shù)據(jù)中心id數(shù)量
 workeridMax       = int64(-1 ^ (-1  workeridBits))               // 支持的最大機(jī)器id數(shù)量
 sequenceMask      = int64(-1 ^ (-1  sequenceBits))               // 支持的最大序列id數(shù)量
 workeridShift     = sequenceBits                                   // 機(jī)器id左移位數(shù)
 datacenteridShift = sequenceBits + workeridBits                    // 數(shù)據(jù)中心id左移位數(shù)
 timestampShift    = sequenceBits + workeridBits + datacenteridBits // 時(shí)間戳左移位數(shù)
)

func (s *Snowflake) NextVal() int64 {
 s.Lock()
 now := time.Now().UnixNano() / 1000000 // 轉(zhuǎn)毫秒
 if s.timestamp == now {
  // 當(dāng)同一時(shí)間戳（精度：毫秒）下多次生成id會(huì)增加序列號(hào)
  s.sequence = (s.sequence + 1)  sequenceMask
  if s.sequence == 0 {
   // 如果當(dāng)前序列超出12bit長度，則需要等待下一毫秒
   // 下一毫秒將使用sequence:0
   for now = s.timestamp {
    now = time.Now().UnixNano() / 1000000
   }
  }
 } else {
  // 不同時(shí)間戳（精度：毫秒）下直接使用序列號(hào)：0
  s.sequence = 0
 }
 t := now - epoch
 if t > timestampMax {
  s.Unlock()
  glog.Errorf("epoch must be between 0 and %d", timestampMax-1)
  return 0
 }
 s.timestamp = now
 r := int64((t)timestampShift | (s.datacenterid  datacenteridShift) | (s.workerid  workeridShift) | (s.sequence))
 s.Unlock()
 return r
}