damn
This commit is contained in:
parent
c167025a3a
commit
c66d460235
@ -7,10 +7,10 @@ include $(GOROOT)/src/Make.inc
|
||||
TARG=gearman
|
||||
GOFILES=\
|
||||
gearman.go\
|
||||
workerjob.go\
|
||||
workerjobclient.go\
|
||||
worker/job.go\
|
||||
worker/jobclient.go\
|
||||
worker.go\
|
||||
clientjob.go\
|
||||
client/job.go\
|
||||
client.go\
|
||||
|
||||
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
|
||||
}
|
||||
|
||||
|
||||
func TestWorkerAddFunction(t *testing.T) {
|
||||
if err := worker.AddFunction("foobar", foobar, 0); err != nil {
|
||||
t.Error(err)
|
||||
|
Loading…
Reference in New Issue
Block a user