lab/snowflake
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Optimize code

Browse Source
This commit is contained in:
Nishaad Ajani 2019-04-11 13:03:43 +08:00
parent 1c6d654675
commit 4c4a27940a
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
snowflake.go
View File

@ -79,7 +79,7 @@ var ErrInvalidBase32 = errors.New("invalid base32")
type Node struct { type Node struct {
mu sync.Mutex mu sync.Mutex
epoch time.Time epoch time.Time
time time.Duration time int64
node int64 node int64
step int64 step int64
@ -132,14 +132,14 @@ func (n *Node) Generate() ID {
n.mu.Lock() n.mu.Lock()
now := time.Since(n.epoch) now := time.Since(n.epoch).Nanoseconds() / 1000000
if now-n.time < time.Millisecond { if now == n.time {
n.step = (n.step + 1) & n.stepMask n.step = (n.step + 1) & n.stepMask
if n.step == 0 { if n.step == 0 {
for now-n.time < time.Millisecond { for now <= n.time {
now = time.Since(n.epoch) now = time.Since(n.epoch).Nanoseconds() / 1000000
} }
} }
} else { } else {
@ -148,7 +148,7 @@ func (n *Node) Generate() ID {
n.time = now n.time = now
r := ID((now.Nanoseconds()/1000000)<<n.timeShift | r := ID((now)<<n.timeShift |
(n.node << n.nodeShift) | (n.node << n.nodeShift) |
(n.step), (n.step),
) )