added code
This commit is contained in:
parent
ffcf226edf
commit
4b5793ee6d
43
id.go
Normal file
43
id.go
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
package idg
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/md5"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// This thing contains a channel that when initialized (see NewIdGenerator)
|
||||||
|
// will return a bunch of (as best as I can tell) unique md5 hashes.
|
||||||
|
//
|
||||||
|
// we use this for naming players, games, etc.
|
||||||
|
//
|
||||||
|
// It will consume a single goroutine
|
||||||
|
type IdGenerator struct {
|
||||||
|
id chan int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewIdGenerator() *IdGenerator {
|
||||||
|
return &IdGenerator{
|
||||||
|
id: make(chan int64),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Run is called (typically in a gorotine) to allow for queries to be made
|
||||||
|
// against IdGenerator.id throgh the Hash method.
|
||||||
|
func (idg *IdGenerator) Run() {
|
||||||
|
var i int64
|
||||||
|
for i = 0; ; i++ {
|
||||||
|
idg.id <- i
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Hash is the method used by a properly instantiated IdGenerator that gives
|
||||||
|
// fairly unique strings. They are currently truncated md5 hashes of the time
|
||||||
|
// plus a unique counter
|
||||||
|
func (id *IdGenerator) Hash() string {
|
||||||
|
h := md5.New()
|
||||||
|
ns := time.Now().UnixNano() + <-id.id
|
||||||
|
io.WriteString(h, fmt.Sprintf("%d", ns))
|
||||||
|
return fmt.Sprintf("%x", h.Sum(nil))[:8]
|
||||||
|
}
|
19
id_test.go
Normal file
19
id_test.go
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
package idg
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestIDGenerator(t *testing.T) {
|
||||||
|
cache := make(map[string]bool)
|
||||||
|
var cur string
|
||||||
|
gg := NewIdGenerator()
|
||||||
|
go gg.Run()
|
||||||
|
for i := 0; i < 10000; i++ {
|
||||||
|
cur = gg.Hash()
|
||||||
|
if _, ok := cache[cur]; ok {
|
||||||
|
t.Errorf("unexpected duplicate key")
|
||||||
|
}
|
||||||
|
cache[cur] = true
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user