gearman-go/src/pkg/gearman/client/job.go

package gearman

import (
    "os"
//    "log"
)

type ClientJob struct {
    Data []byte
    Handle string
    UniqueId string
    magicCode, dataType uint32
}

func NewClientJob(magiccode, datatype uint32, data []byte) (job *ClientJob) {
    return &ClientJob{magicCode:magiccode,
        dataType:datatype,
        Data:data}
}

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
}

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[:] ...)
    data = append(data, []byte{0, 0, 0, 0} ...)
    l := len(job.Data)
    data = append(data, job.Data ...)
    datalength := uint32ToByte(uint32(l))
    copy(data[8:12], datalength[:])
    return
}

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 := splitByteArray(job.Data, '\x00')
            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
}

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 := splitByteArray(job.Data, '\x00')
    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
}