feat:完善项目
This commit is contained in:
55
utils/snowflake.go
Normal file
55
utils/snowflake.go
Normal file
@ -0,0 +1,55 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
const (
|
||||
workerBits uint8 = 10
|
||||
numberBits uint8 = 12
|
||||
workerMax int64 = -1 ^ (-1 << workerBits)
|
||||
numberMax int64 = -1 ^ (-1 << numberBits)
|
||||
timeShift uint8 = workerBits + numberBits
|
||||
workerShift uint8 = numberBits
|
||||
startTime int64 = 1136185445000
|
||||
)
|
||||
|
||||
type Worker struct {
|
||||
mu sync.Mutex
|
||||
timestamp int64
|
||||
workerID int64
|
||||
number int64
|
||||
}
|
||||
|
||||
func (w *Worker) GetID() int64 {
|
||||
w.mu.Lock()
|
||||
defer w.mu.Unlock()
|
||||
now := time.Now().UnixNano() / 1e6
|
||||
|
||||
if w.timestamp == now {
|
||||
w.number++
|
||||
if w.number > numberMax {
|
||||
for now <= w.timestamp {
|
||||
now = time.Now().UnixNano() / 1e6
|
||||
}
|
||||
}
|
||||
} else {
|
||||
w.number = 0
|
||||
w.timestamp = now
|
||||
}
|
||||
return (now-startTime)<<timeShift | (w.workerID << workerShift) | (w.number)
|
||||
}
|
||||
|
||||
func NewSnowflake(workerID int64) (*Worker, error) {
|
||||
if workerID < 0 || workerID > workerMax {
|
||||
return nil, errors.New("Worker ID Excess Of Quantity")
|
||||
}
|
||||
// 生成一个新节点
|
||||
return &Worker{
|
||||
timestamp: 0,
|
||||
workerID: workerID,
|
||||
number: 0,
|
||||
}, nil
|
||||
}
|
Reference in New Issue
Block a user