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

Use constants for bits, for easier changing

Browse Source
This commit is contained in:
Shannon Wynter 2017-10-01 22:07:22 +10:00
parent 1c0147d077
commit 3a07db85c1
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 (
) )
const ( const (
spareBits = 22
nodeBits = 10 nodeBits = 10
stepBits = 12 stepBits = spareBits - nodeBits
nodeMax = -1 ^ (-1 << nodeBits) nodeMax = -1 ^ (-1 << nodeBits)
nodeMask = nodeMax << stepBits
stepMask int64 = -1 ^ (-1 << stepBits) stepMask int64 = -1 ^ (-1 << stepBits)
timeShift uint8 = nodeBits + stepBits timeShift uint8 = nodeBits + stepBits
nodeShift uint8 = stepBits nodeShift uint8 = stepBits
@ -189,12 +191,12 @@ func (f ID) Time() int64 {
// Node returns an int64 of the snowflake ID node number // Node returns an int64 of the snowflake ID node number
func (f ID) Node() int64 { func (f ID) Node() int64 {
return int64(f) & 0x00000000003FF000 >> nodeShift return int64(f) & nodeMask >> nodeShift
} }
// Step returns an int64 of the snowflake step (or sequence) number // Step returns an int64 of the snowflake step (or sequence) number
func (f ID) Step() int64 { func (f ID) Step() int64 {
return int64(f) & 0x0000000000000FFF return int64(f) & stepMask
} }
// MarshalJSON returns a json byte array string of the snowflake ID. // MarshalJSON returns a json byte array string of the snowflake ID.