yuxh/gearman-go
1
0
Fork 1
Code Issues Pull Requests Releases Wiki Activity

Merge 0deaa25a0b into d32eb195e1

Browse Source
This commit is contained in:
wickning1 2014-12-10 01:15:04 +00:00
commit c9a5aea1cf
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
client/client.go
View File

@ -191,7 +191,7 @@ func (client *Client) do(funcname string, data []byte,
} }
handle = resp.Handle handle = resp.Handle
} }
id := IdGen.Id() id := IdGen.Id(funcname, string(data))
req := getJob(id, []byte(funcname), data) req := getJob(id, []byte(funcname), data)
req.DataType = flag req.DataType = flag
if err = client.write(req); err != nil { if err = client.write(req); err != nil {

6
client/id.go
View File

@ -16,10 +16,10 @@ func init() {
IdGen = NewAutoIncId() IdGen = NewAutoIncId()
} }
// ID generator interface. Users can implament this for // ID generator interface. Users can implement this for
// their own generator. // their own generator.
type IdGenerator interface { type IdGenerator interface {
Id() string Id(funcname, payload string) string
} }
// AutoIncId // AutoIncId
@ -27,7 +27,7 @@ type autoincId struct {
value int64 value int64
} }
func (ai *autoincId) Id() string { func (ai *autoincId) Id(funcname, payload string) string {
next := atomic.AddInt64(&ai.value, 1) next := atomic.AddInt64(&ai.value, 1)
return strconv.FormatInt(next, 10) return strconv.FormatInt(next, 10)
} }

4
client/id_test.go
View File

@ -7,9 +7,9 @@ import (
func TestAutoInc(t *testing.T) { func TestAutoInc(t *testing.T) {
ai := NewAutoIncId() ai := NewAutoIncId()
previous := ai.Id() previous := ai.Id("testfuncname", "fakepayload")
for i := 0; i < 10; i++ { for i := 0; i < 10; i++ {
id := ai.Id() id := ai.Id("testfuncname", "fakepayload2")
if id == previous { if id == previous {
t.Errorf("Id not unique, previous and current %s", id) t.Errorf("Id not unique, previous and current %s", id)
} }