gearman-go/client/id.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,
	}
}