-
Notifications
You must be signed in to change notification settings - Fork 0
/
snow.go
60 lines (50 loc) · 1.02 KB
/
snow.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package id
import (
"errors"
"net"
"strconv"
"time"
"github.com/sony/sonyflake"
"github.com/crochee/lirity/variable"
)
var sf = sonyflake.NewSonyflake(sonyflake.Settings{
StartTime: time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC),
MachineID: machineID})
// NextID generate id
func NextID() (uint64, error) {
return sf.NextID()
}
// NextIDString generate id
func NextIDString() (string, error) {
id, err := sf.NextID()
if err != nil {
return "", err
}
return strconv.FormatUint(id, variable.DecimalSystem), nil
}
func machineID() (uint16, error) {
ip, err := lower16BitIPV4()
if err != nil {
return 0, err
}
return uint16(ip[2])<<8 + uint16(ip[3]), nil
}
func lower16BitIPV4() (net.IP, error) {
as, err := net.InterfaceAddrs()
if err != nil {
return nil, err
}
for _, a := range as {
inet, ok := a.(*net.IPNet)
if !ok || inet.IP.IsLoopback() {
continue
}
ip := inet.IP.To4()
// Pass ipv6 address
if ip == nil {
continue
}
return ip, nil
}
return nil, errors.New("no private ip address")
}