57_Wolve/sonyflake
1
0
Fork 0
mirror of https://github.com/sony/sonyflake.git synced 2025-12-23 05:05:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support generating id with custom now.

Browse Source
This commit is contained in:
PengWeikang
2023-05-05 11:44:36 +08:00
parent eafab81cd5
commit d5234c2d4e
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
sonyflake.go
View File

@@ -89,12 +89,16 @@ func NewSonyflake(st Settings) *Sonyflake {
// NextID generates a next unique ID.
// After the Sonyflake time overflows, NextID returns an error.
func (sf *Sonyflake) NextID() (uint64, error) {
return sf.NextIDWithCustomNow(time.Now())
}
func (sf *Sonyflake) NextIDWithCustomNow(now time.Time) (uint64, error) {
const maskSequence = uint16(1<<BitLenSequence - 1)
sf.mutex.Lock()
defer sf.mutex.Unlock()
current := currentElapsedTime(sf.startTime)
current := currentElapsedTime(now, sf.startTime)
if sf.elapsedTime < current {
sf.elapsedTime = current
sf.sequence = 0
@@ -116,8 +120,8 @@ func toSonyflakeTime(t time.Time) int64 {
return t.UTC().UnixNano() / sonyflakeTimeUnit
}
func currentElapsedTime(startTime int64) int64 {
return toSonyflakeTime(time.Now()) - startTime
func currentElapsedTime(now time.Time, startTime int64) int64 {
return toSonyflakeTime(now) - startTime
}
func sleepTime(overtime int64) time.Duration {