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

Bug fix and optimization

Browse Source 
* Fix for default Epoch to use monotonic clock.
* Using common case optimization in the time pkg.
This commit is contained in:
Nishaad Ajani 2019-04-10 17:02:30 +08:00
parent 2983bd9d27
commit 55825bae4b
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
snowflake.go
View File

@ -12,9 +12,11 @@ import (
)
var (
curTime = time.Now()
// Epoch is set to the twitter snowflake epoch of Nov 04 2010 01:42:54 UTC
// You may customize this to set a different epoch for your application.
Epoch = time.Date(2010, time.November, 4, 1, 42, 54, 0, time.UTC)
// Note: add time.Duration to time.Now() to make sure we use the monotonic clock if available.
Epoch = curTime.Add(time.Date(2010, time.November, 4, 1, 42, 54, 0, time.UTC).Sub(curTime))
// Number of bits to use for Node
// Remember, you have a total 22 bits to share between Node/Step
@ -108,14 +110,14 @@ func (n *Node) Generate() ID {
n.mu.Lock()
now := time.Now().Sub(Epoch)
now := time.Since(Epoch)
if now-n.time < time.Millisecond {
n.step = (n.step + 1) & stepMask
if n.step == 0 {
for now-n.time < time.Millisecond {
now = time.Now().Sub(Epoch)
now = time.Since(Epoch)
}
}
} else {