123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481 |
- package uuid
- import (
- "bytes"
- "crypto/md5"
- "crypto/rand"
- "crypto/sha1"
- "database/sql/driver"
- "encoding/binary"
- "encoding/hex"
- "fmt"
- "hash"
- "net"
- "os"
- "sync"
- "time"
- )
- const (
- VariantNCS = iota
- VariantRFC4122
- VariantMicrosoft
- VariantFuture
- )
- const (
- DomainPerson = iota
- DomainGroup
- DomainOrg
- )
- const epochStart = 122192928000000000
- const dash byte = '-'
- var (
- storageMutex sync.Mutex
- storageOnce sync.Once
- epochFunc = unixTimeFunc
- clockSequence uint16
- lastTime uint64
- hardwareAddr [6]byte
- posixUID = uint32(os.Getuid())
- posixGID = uint32(os.Getgid())
- )
- var (
- urnPrefix = []byte("urn:uuid:")
- byteGroups = []int{8, 4, 4, 4, 12}
- )
- func initClockSequence() {
- buf := make([]byte, 2)
- safeRandom(buf)
- clockSequence = binary.BigEndian.Uint16(buf)
- }
- func initHardwareAddr() {
- interfaces, err := net.Interfaces()
- if err == nil {
- for _, iface := range interfaces {
- if len(iface.HardwareAddr) >= 6 {
- copy(hardwareAddr[:], iface.HardwareAddr)
- return
- }
- }
- }
-
-
- safeRandom(hardwareAddr[:])
-
- hardwareAddr[0] |= 0x01
- }
- func initStorage() {
- initClockSequence()
- initHardwareAddr()
- }
- func safeRandom(dest []byte) {
- if _, err := rand.Read(dest); err != nil {
- panic(err)
- }
- }
- func unixTimeFunc() uint64 {
- return epochStart + uint64(time.Now().UnixNano()/100)
- }
- type UUID [16]byte
- type NullUUID struct {
- UUID UUID
- Valid bool
- }
- var Nil = UUID{}
- var (
- NamespaceDNS, _ = FromString("6ba7b810-9dad-11d1-80b4-00c04fd430c8")
- NamespaceURL, _ = FromString("6ba7b811-9dad-11d1-80b4-00c04fd430c8")
- NamespaceOID, _ = FromString("6ba7b812-9dad-11d1-80b4-00c04fd430c8")
- NamespaceX500, _ = FromString("6ba7b814-9dad-11d1-80b4-00c04fd430c8")
- )
- func And(u1 UUID, u2 UUID) UUID {
- u := UUID{}
- for i := 0; i < 16; i++ {
- u[i] = u1[i] & u2[i]
- }
- return u
- }
- func Or(u1 UUID, u2 UUID) UUID {
- u := UUID{}
- for i := 0; i < 16; i++ {
- u[i] = u1[i] | u2[i]
- }
- return u
- }
- func Equal(u1 UUID, u2 UUID) bool {
- return bytes.Equal(u1[:], u2[:])
- }
- func (u UUID) Version() uint {
- return uint(u[6] >> 4)
- }
- func (u UUID) Variant() uint {
- switch {
- case (u[8] & 0x80) == 0x00:
- return VariantNCS
- case (u[8]&0xc0)|0x80 == 0x80:
- return VariantRFC4122
- case (u[8]&0xe0)|0xc0 == 0xc0:
- return VariantMicrosoft
- }
- return VariantFuture
- }
- func (u UUID) Bytes() []byte {
- return u[:]
- }
- func (u UUID) String() string {
- buf := make([]byte, 36)
- hex.Encode(buf[0:8], u[0:4])
- buf[8] = dash
- hex.Encode(buf[9:13], u[4:6])
- buf[13] = dash
- hex.Encode(buf[14:18], u[6:8])
- buf[18] = dash
- hex.Encode(buf[19:23], u[8:10])
- buf[23] = dash
- hex.Encode(buf[24:], u[10:])
- return string(buf)
- }
- func (u *UUID) SetVersion(v byte) {
- u[6] = (u[6] & 0x0f) | (v << 4)
- }
- func (u *UUID) SetVariant() {
- u[8] = (u[8] & 0xbf) | 0x80
- }
- func (u UUID) MarshalText() (text []byte, err error) {
- text = []byte(u.String())
- return
- }
- func (u *UUID) UnmarshalText(text []byte) (err error) {
- if len(text) < 32 {
- err = fmt.Errorf("uuid: UUID string too short: %s", text)
- return
- }
- t := text[:]
- braced := false
- if bytes.Equal(t[:9], urnPrefix) {
- t = t[9:]
- } else if t[0] == '{' {
- braced = true
- t = t[1:]
- }
- b := u[:]
- for i, byteGroup := range byteGroups {
- if i > 0 {
- if t[0] != '-' {
- err = fmt.Errorf("uuid: invalid string format")
- return
- }
- t = t[1:]
- }
- if len(t) < byteGroup {
- err = fmt.Errorf("uuid: UUID string too short: %s", text)
- return
- }
- if i == 4 && len(t) > byteGroup &&
- ((braced && t[byteGroup] != '}') || len(t[byteGroup:]) > 1 || !braced) {
- err = fmt.Errorf("uuid: UUID string too long: %s", text)
- return
- }
- _, err = hex.Decode(b[:byteGroup/2], t[:byteGroup])
- if err != nil {
- return
- }
- t = t[byteGroup:]
- b = b[byteGroup/2:]
- }
- return
- }
- func (u UUID) MarshalBinary() (data []byte, err error) {
- data = u.Bytes()
- return
- }
- func (u *UUID) UnmarshalBinary(data []byte) (err error) {
- if len(data) != 16 {
- err = fmt.Errorf("uuid: UUID must be exactly 16 bytes long, got %d bytes", len(data))
- return
- }
- copy(u[:], data)
- return
- }
- func (u UUID) Value() (driver.Value, error) {
- return u.String(), nil
- }
- func (u *UUID) Scan(src interface{}) error {
- switch src := src.(type) {
- case []byte:
- if len(src) == 16 {
- return u.UnmarshalBinary(src)
- }
- return u.UnmarshalText(src)
- case string:
- return u.UnmarshalText([]byte(src))
- }
- return fmt.Errorf("uuid: cannot convert %T to UUID", src)
- }
- func (u NullUUID) Value() (driver.Value, error) {
- if !u.Valid {
- return nil, nil
- }
-
- return u.UUID.Value()
- }
- func (u *NullUUID) Scan(src interface{}) error {
- if src == nil {
- u.UUID, u.Valid = Nil, false
- return nil
- }
-
- u.Valid = true
- return u.UUID.Scan(src)
- }
- func FromBytes(input []byte) (u UUID, err error) {
- err = u.UnmarshalBinary(input)
- return
- }
- func FromBytesOrNil(input []byte) UUID {
- uuid, err := FromBytes(input)
- if err != nil {
- return Nil
- }
- return uuid
- }
- func FromString(input string) (u UUID, err error) {
- err = u.UnmarshalText([]byte(input))
- return
- }
- func FromStringOrNil(input string) UUID {
- uuid, err := FromString(input)
- if err != nil {
- return Nil
- }
- return uuid
- }
- func getStorage() (uint64, uint16, []byte) {
- storageOnce.Do(initStorage)
- storageMutex.Lock()
- defer storageMutex.Unlock()
- timeNow := epochFunc()
-
-
- if timeNow <= lastTime {
- clockSequence++
- }
- lastTime = timeNow
- return timeNow, clockSequence, hardwareAddr[:]
- }
- func NewV1() UUID {
- u := UUID{}
- timeNow, clockSeq, hardwareAddr := getStorage()
- binary.BigEndian.PutUint32(u[0:], uint32(timeNow))
- binary.BigEndian.PutUint16(u[4:], uint16(timeNow>>32))
- binary.BigEndian.PutUint16(u[6:], uint16(timeNow>>48))
- binary.BigEndian.PutUint16(u[8:], clockSeq)
- copy(u[10:], hardwareAddr)
- u.SetVersion(1)
- u.SetVariant()
- return u
- }
- func NewV2(domain byte) UUID {
- u := UUID{}
- timeNow, clockSeq, hardwareAddr := getStorage()
- switch domain {
- case DomainPerson:
- binary.BigEndian.PutUint32(u[0:], posixUID)
- case DomainGroup:
- binary.BigEndian.PutUint32(u[0:], posixGID)
- }
- binary.BigEndian.PutUint16(u[4:], uint16(timeNow>>32))
- binary.BigEndian.PutUint16(u[6:], uint16(timeNow>>48))
- binary.BigEndian.PutUint16(u[8:], clockSeq)
- u[9] = domain
- copy(u[10:], hardwareAddr)
- u.SetVersion(2)
- u.SetVariant()
- return u
- }
- func NewV3(ns UUID, name string) UUID {
- u := newFromHash(md5.New(), ns, name)
- u.SetVersion(3)
- u.SetVariant()
- return u
- }
- func NewV4() UUID {
- u := UUID{}
- safeRandom(u[:])
- u.SetVersion(4)
- u.SetVariant()
- return u
- }
- func NewV5(ns UUID, name string) UUID {
- u := newFromHash(sha1.New(), ns, name)
- u.SetVersion(5)
- u.SetVariant()
- return u
- }
- func newFromHash(h hash.Hash, ns UUID, name string) UUID {
- u := UUID{}
- h.Write(ns[:])
- h.Write([]byte(name))
- copy(u[:], h.Sum(nil))
- return u
- }
|