7614c2678a
--HG-- branch : 0.1 rename : gearman/client/client.go => client/client.go rename : gearman/client/client_test.go => client/client_test.go rename : gearman/client/clientjob.go => client/job.go rename : gearman/gearman.go => common/gearman.go rename : gearman/worker/jobagent.go => worker/jobagent.go rename : gearman/worker/worker.go => worker/worker.go rename : gearman/worker/worker_test.go => worker/worker_test.go rename : gearman/worker/workerjob.go => worker/workerjob.go
106 lines
2.8 KiB
Go
106 lines
2.8 KiB
Go
// 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 client
|
|
|
|
import (
|
|
"bytes"
|
|
"bitbucket.org/mikespook/gearman-go/common"
|
|
)
|
|
// An error handler
|
|
type ErrorHandler func(error)
|
|
|
|
// Client side job
|
|
type Job struct {
|
|
Data []byte
|
|
Handle, UniqueId string
|
|
magicCode, DataType uint32
|
|
}
|
|
|
|
// Create a new job
|
|
func newJob(magiccode, datatype uint32, data []byte) (job *Job) {
|
|
return &Job{magicCode: magiccode,
|
|
DataType: datatype,
|
|
Data: data}
|
|
}
|
|
|
|
// Decode a job from byte slice
|
|
func decodeJob(data []byte) (job *Job, err error) {
|
|
if len(data) < 12 {
|
|
return nil, common.Errorf("Invalid data: %V", data)
|
|
}
|
|
datatype := common.BytesToUint32([4]byte{data[4], data[5], data[6], data[7]})
|
|
l := common.BytesToUint32([4]byte{data[8], data[9], data[10], data[11]})
|
|
if len(data[12:]) != int(l) {
|
|
return nil, common.Errorf("Invalid data: %V", data)
|
|
}
|
|
data = data[12:]
|
|
return newJob(common.RES, datatype, data), nil
|
|
}
|
|
|
|
// Encode a job to byte slice
|
|
func (job *Job) Encode() (data []byte) {
|
|
l := len(job.Data) + 12
|
|
data = make([]byte, l)
|
|
|
|
magiccode := common.Uint32ToBytes(job.magicCode)
|
|
datatype := common.Uint32ToBytes(job.DataType)
|
|
datalength := common.Uint32ToBytes(uint32(l))
|
|
for i := 0; i < l; i ++ {
|
|
switch {
|
|
case i < 4:
|
|
data[i] = magiccode[i]
|
|
case i < 8:
|
|
data[i] = datatype[i - 4]
|
|
case i < 12:
|
|
data[i] = datalength[i - 8]
|
|
default:
|
|
data[i] = job.Data[i - 12]
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// Extract the job's result.
|
|
func (job *Job) Result() (data []byte, err error) {
|
|
switch job.DataType {
|
|
case common.WORK_FAIL:
|
|
job.Handle = string(job.Data)
|
|
return nil, common.ErrWorkFail
|
|
case common.WORK_EXCEPTION:
|
|
err = common.ErrWorkException
|
|
fallthrough
|
|
case common.WORK_COMPLETE:
|
|
s := bytes.SplitN(job.Data, []byte{'\x00'}, 2)
|
|
if len(s) != 2 {
|
|
return nil, common.Errorf("Invalid data: %V", job.Data)
|
|
}
|
|
job.Handle = string(s[0])
|
|
data = s[1]
|
|
default:
|
|
err = common.ErrDataType
|
|
}
|
|
return
|
|
}
|
|
|
|
// Extract the job's update
|
|
func (job *Job) Update() (data []byte, err error) {
|
|
if job.DataType != common.WORK_DATA && job.DataType != common.WORK_WARNING {
|
|
err = common.ErrDataType
|
|
return
|
|
}
|
|
s := bytes.SplitN(job.Data, []byte{'\x00'}, 2)
|
|
if len(s) != 2 {
|
|
err = common.ErrInvalidData
|
|
return
|
|
}
|
|
if job.DataType == common.WORK_WARNING {
|
|
err = common.ErrWorkWarning
|
|
}
|
|
job.Handle = string(s[0])
|
|
data = s[1]
|
|
return
|
|
}
|