forked from yuxh/gearman-go
43 lines
747 B
Go
43 lines
747 B
Go
package client
|
|
|
|
import (
|
|
"strconv"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
// Global ID generator
|
|
// Default is an autoincrement ID generator
|
|
IdGen IdGenerator
|
|
)
|
|
|
|
func init() {
|
|
IdGen = NewAutoIncId()
|
|
}
|
|
|
|
// ID generator interface. Users can implament this for
|
|
// their own generator.
|
|
type IdGenerator interface {
|
|
Id() string
|
|
}
|
|
|
|
// AutoIncId
|
|
type autoincId struct {
|
|
value int64
|
|
}
|
|
|
|
func (ai *autoincId) Id() string {
|
|
next := atomic.AddInt64(&ai.value, 1)
|
|
return strconv.FormatInt(next, 10)
|
|
}
|
|
|
|
// NewAutoIncId returns an autoincrement ID generator
|
|
func NewAutoIncId() IdGenerator {
|
|
// we'll consider the nano fraction of a second at startup unique
|
|
// and count up from there.
|
|
return &autoincId{
|
|
value: int64(time.Now().Nanosecond()) << 32,
|
|
}
|
|
}
|