damn
This commit is contained in:
parent
c167025a3a
commit
c66d460235
@ -7,10 +7,10 @@ include $(GOROOT)/src/Make.inc
|
|||||||
TARG=gearman
|
TARG=gearman
|
||||||
GOFILES=\
|
GOFILES=\
|
||||||
gearman.go\
|
gearman.go\
|
||||||
workerjob.go\
|
worker/job.go\
|
||||||
workerjobclient.go\
|
worker/jobclient.go\
|
||||||
worker.go\
|
worker.go\
|
||||||
clientjob.go\
|
client/job.go\
|
||||||
client.go\
|
client.go\
|
||||||
|
|
||||||
CLEANFILES+=gearman_test
|
CLEANFILES+=gearman_test
|
||||||
|
99
src/pkg/gearman/client/job.go
Normal file
99
src/pkg/gearman/client/job.go
Normal file
@ -0,0 +1,99 @@
|
|||||||
|
// Copyright 2011 Xing Xing <mikespook@gmail.com> All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package gearman
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"bytes"
|
||||||
|
// "log"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Client side job
|
||||||
|
type ClientJob struct {
|
||||||
|
Data []byte
|
||||||
|
Handle, UniqueId string
|
||||||
|
magicCode, DataType uint32
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a new job
|
||||||
|
func NewClientJob(magiccode, datatype uint32, data []byte) (job *ClientJob) {
|
||||||
|
return &ClientJob{magicCode: magiccode,
|
||||||
|
DataType: datatype,
|
||||||
|
Data: data}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode a job from byte slice
|
||||||
|
func DecodeClientJob(data []byte) (job *ClientJob, err os.Error) {
|
||||||
|
if len(data) < 12 {
|
||||||
|
err = os.NewError("Data length is too small.")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
datatype := byteToUint32([4]byte{data[4], data[5], data[6], data[7]})
|
||||||
|
l := byteToUint32([4]byte{data[8], data[9], data[10], data[11]})
|
||||||
|
if len(data[12:]) != int(l) {
|
||||||
|
err = os.NewError("Invalid data length.")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
data = data[12:]
|
||||||
|
job = NewClientJob(RES, datatype, data)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Encode a job to byte slice
|
||||||
|
func (job *ClientJob) Encode() (data []byte) {
|
||||||
|
magiccode := uint32ToByte(job.magicCode)
|
||||||
|
datatype := uint32ToByte(job.DataType)
|
||||||
|
data = make([]byte, 0, 1024*64)
|
||||||
|
data = append(data, magiccode[:]...)
|
||||||
|
data = append(data, datatype[:]...)
|
||||||
|
l := len(job.Data)
|
||||||
|
datalength := uint32ToByte(uint32(l))
|
||||||
|
data = append(data, datalength[:]...)
|
||||||
|
data = append(data, job.Data...)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extract the job's result.
|
||||||
|
func (job *ClientJob) Result() (data []byte, err os.Error) {
|
||||||
|
switch job.DataType {
|
||||||
|
case WORK_FAIL:
|
||||||
|
job.Handle = string(job.Data)
|
||||||
|
err = os.NewError("Work fail.")
|
||||||
|
return
|
||||||
|
case WORK_EXCEPTION:
|
||||||
|
err = os.NewError("Work exception.")
|
||||||
|
fallthrough
|
||||||
|
case WORK_COMPLETE:
|
||||||
|
s := bytes.SplitN(job.Data, []byte{'\x00'}, 2)
|
||||||
|
if len(s) != 2 {
|
||||||
|
err = os.NewError("Invalid data.")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
job.Handle = string(s[0])
|
||||||
|
data = s[1]
|
||||||
|
default:
|
||||||
|
err = os.NewError("The job is not a result.")
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extract the job's update
|
||||||
|
func (job *ClientJob) Update() (data []byte, err os.Error) {
|
||||||
|
if job.DataType != WORK_DATA && job.DataType != WORK_WARNING {
|
||||||
|
err = os.NewError("The job is not a update.")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
s := bytes.SplitN(job.Data, []byte{'\x00'}, 2)
|
||||||
|
if len(s) != 2 {
|
||||||
|
err = os.NewError("Invalid data.")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if job.DataType == WORK_WARNING {
|
||||||
|
err = os.NewError("Work warning")
|
||||||
|
}
|
||||||
|
job.Handle = string(s[0])
|
||||||
|
data = s[1]
|
||||||
|
return
|
||||||
|
}
|
@ -1,99 +0,0 @@
|
|||||||
// Copyright 2011 Xing Xing <mikespook@gmail.com> All rights reserved.
|
|
||||||
// Use of this source code is governed by a MIT
|
|
||||||
// license that can be found in the LICENSE file.
|
|
||||||
|
|
||||||
package gearman
|
|
||||||
|
|
||||||
import (
|
|
||||||
"os"
|
|
||||||
"bytes"
|
|
||||||
// "log"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Client side job
|
|
||||||
type ClientJob struct {
|
|
||||||
Data []byte
|
|
||||||
Handle, UniqueId string
|
|
||||||
magicCode, DataType uint32
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create a new job
|
|
||||||
func NewClientJob(magiccode, datatype uint32, data []byte) (job *ClientJob) {
|
|
||||||
return &ClientJob{magicCode: magiccode,
|
|
||||||
DataType: datatype,
|
|
||||||
Data: data}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Decode a job from byte slice
|
|
||||||
func DecodeClientJob(data []byte) (job *ClientJob, err os.Error) {
|
|
||||||
if len(data) < 12 {
|
|
||||||
err = os.NewError("Data length is too small.")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
datatype := byteToUint32([4]byte{data[4], data[5], data[6], data[7]})
|
|
||||||
l := byteToUint32([4]byte{data[8], data[9], data[10], data[11]})
|
|
||||||
if len(data[12:]) != int(l) {
|
|
||||||
err = os.NewError("Invalid data length.")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
data = data[12:]
|
|
||||||
job = NewClientJob(RES, datatype, data)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Encode a job to byte slice
|
|
||||||
func (job *ClientJob) Encode() (data []byte) {
|
|
||||||
magiccode := uint32ToByte(job.magicCode)
|
|
||||||
datatype := uint32ToByte(job.DataType)
|
|
||||||
data = make([]byte, 0, 1024*64)
|
|
||||||
data = append(data, magiccode[:]...)
|
|
||||||
data = append(data, datatype[:]...)
|
|
||||||
l := len(job.Data)
|
|
||||||
datalength := uint32ToByte(uint32(l))
|
|
||||||
data = append(data, datalength[:]...)
|
|
||||||
data = append(data, job.Data...)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Extract the job's result.
|
|
||||||
func (job *ClientJob) Result() (data []byte, err os.Error) {
|
|
||||||
switch job.DataType {
|
|
||||||
case WORK_FAIL:
|
|
||||||
job.Handle = string(job.Data)
|
|
||||||
err = os.NewError("Work fail.")
|
|
||||||
return
|
|
||||||
case WORK_EXCEPTION:
|
|
||||||
err = os.NewError("Work exception.")
|
|
||||||
fallthrough
|
|
||||||
case WORK_COMPLETE:
|
|
||||||
s := bytes.SplitN(job.Data, []byte{'\x00'}, 2)
|
|
||||||
if len(s) != 2 {
|
|
||||||
err = os.NewError("Invalid data.")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
job.Handle = string(s[0])
|
|
||||||
data = s[1]
|
|
||||||
default:
|
|
||||||
err = os.NewError("The job is not a result.")
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Extract the job's update
|
|
||||||
func (job *ClientJob) Update() (data []byte, err os.Error) {
|
|
||||||
if job.DataType != WORK_DATA && job.DataType != WORK_WARNING {
|
|
||||||
err = os.NewError("The job is not a update.")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
s := bytes.SplitN(job.Data, []byte{'\x00'}, 2)
|
|
||||||
if len(s) != 2 {
|
|
||||||
err = os.NewError("Invalid data.")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if job.DataType == WORK_WARNING {
|
|
||||||
err = os.NewError("Work warning")
|
|
||||||
}
|
|
||||||
job.Handle = string(s[0])
|
|
||||||
data = s[1]
|
|
||||||
return
|
|
||||||
}
|
|
@ -27,7 +27,6 @@ func foobar(job *WorkerJob) ([]byte, os.Error) {
|
|||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
func TestWorkerAddFunction(t *testing.T) {
|
func TestWorkerAddFunction(t *testing.T) {
|
||||||
if err := worker.AddFunction("foobar", foobar, 0); err != nil {
|
if err := worker.AddFunction("foobar", foobar, 0); err != nil {
|
||||||
t.Error(err)
|
t.Error(err)
|
||||||
|
Loading…
Reference in New Issue
Block a user