DeforaNetworks/corteza
3
0
Fork 0
Code Activity

Allow use of custom machine ID for sonyflake ID

Browse Source 
Use SONYFLAKE_MACHINE_ID with uint16 to specifiy custom
machine ID to be used by Sonyflake. This is usefuly
when your infrastructure network configuration does not follow
IANA rules for private IPs.

Ref: #15
This commit is contained in:
Denis Arh
2022-11-14 08:39:25 +01:00
parent 88bafc2f15
commit 204738f7c9
1 changed files with 60 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/id/sonyflake.go
+60 -6
View File
@@ -1,26 +1,80 @@
package id
import (
"fmt"
"github.com/sony/sonyflake"
"os"
"strconv"
"time"
)
// Midnight, January 1st 2017
const jan1st2017 = 1483228800
const (
// Midnight, January 1st 2017
jan1st2017 = 1483228800
var _sonyflake *sonyflake.Sonyflake
// Env key for custom machine ID
//
// Sonyflake assumes that your private IP falls under the following ranges:
// 10.0.0.0 - 10.255.255.255
// 172.16.0.0 - 172.31.255.255
// 192.168.0.0 - 192.168.255.255
//
// When running in a container, this could not always the case.
// To override this, set SONYFLAKE_MACHINE_ID to a valid uint16
// (0..65535) and sonyflake will use that as machine ID.
envKeyMachineID = "SONYFLAKE_MACHINE_ID"
// When set to uint32 value, we no longer use sonyflake
// but a simple atomic counter
//
// This is useful for testing and debugging
// and should not be used in production
// @todo implement
// envKeySimpleIncrement = "SONYFLAKE_SIMPLE_INCREMENT"
)
var sf *sonyflake.Sonyflake
func init() {
_sonyflake = sonyflake.NewSonyflake(sonyflake.Settings{
if err := initSonyflake(); err != nil {
panic(fmt.Errorf("sonyflake init failed: %w", err))
}
}
func initSonyflake() error {
settings := sonyflake.Settings{
StartTime: time.Unix(jan1st2017, 0),
})
}
// check if SONYFLAKE_MACHINE_ID is set and create
// MachineID function that returns it
if machineID := os.Getenv(envKeyMachineID); machineID != "" {
// check if machineID is a valid uint16
if id, err := strconv.ParseUint(machineID, 10, 16); err != nil {
// should crash right away
return fmt.Errorf(
"could not use %s (%s), expecting uint16 (0..65535): %w",
envKeyMachineID,
machineID,
err,
)
} else {
settings.MachineID = func() (uint16, error) {
return uint16(id), nil
}
}
}
sf = sonyflake.NewSonyflake(settings)
_, err := sf.NextID()
return err
}
// NextID returns uint64 ID, or panics
//
// See https://github.com/sony/sonyflake for details
func nextSonyflake() uint64 {
if id, err := _sonyflake.NextID(); err != nil {
if id, err := sf.NextID(); err != nil {
panic(err)
} else {
return id