go fmt
This commit is contained in:
parent
2a27eca7b7
commit
124e686699
@ -6,9 +6,9 @@
|
|||||||
package client
|
package client
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"bytes"
|
"bytes"
|
||||||
"errors"
|
"errors"
|
||||||
|
"fmt"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@ -40,3 +40,6 @@ func GetError(data []byte) (err error) {
|
|||||||
err = errors.New(fmt.Sprintf("%s: %s", rel[0], rel[1]))
|
err = errors.New(fmt.Sprintf("%s: %s", rel[0], rel[1]))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Error handler
|
||||||
|
type ErrorHandler func(error)
|
||||||
|
@ -1,11 +0,0 @@
|
|||||||
package client
|
|
||||||
|
|
||||||
// Response handler
|
|
||||||
type ResponseHandler func(*Response)
|
|
||||||
|
|
||||||
// Error handler
|
|
||||||
type ErrorHandler func(error)
|
|
||||||
|
|
||||||
// Status handler
|
|
||||||
// handle, known, running, numerator, denominator
|
|
||||||
type StatusHandler func(string, bool, bool, uint64, uint64)
|
|
@ -1,9 +1,9 @@
|
|||||||
package client
|
package client
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"time"
|
|
||||||
"strconv"
|
"strconv"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
|
@ -40,7 +40,7 @@ func TestPoolEcho(t *testing.T) {
|
|||||||
|
|
||||||
func TestPoolDoBg(t *testing.T) {
|
func TestPoolDoBg(t *testing.T) {
|
||||||
addr, handle, err := pool.DoBg("ToUpper",
|
addr, handle, err := pool.DoBg("ToUpper",
|
||||||
[]byte("abcdef"), JOB_LOW);
|
[]byte("abcdef"), JOB_LOW)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Error(err)
|
t.Error(err)
|
||||||
return
|
return
|
||||||
|
@ -6,12 +6,15 @@
|
|||||||
package client
|
package client
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"bytes"
|
"bytes"
|
||||||
"strconv"
|
|
||||||
"encoding/binary"
|
"encoding/binary"
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Response handler
|
||||||
|
type ResponseHandler func(*Response)
|
||||||
|
|
||||||
// response
|
// response
|
||||||
type Response struct {
|
type Response struct {
|
||||||
DataType uint32
|
DataType uint32
|
||||||
|
@ -1,5 +1,9 @@
|
|||||||
package client
|
package client
|
||||||
|
|
||||||
|
// Status handler
|
||||||
|
// handle, known, running, numerator, denominator
|
||||||
|
type StatusHandler func(string, bool, bool, uint64, uint64)
|
||||||
|
|
||||||
type Status struct {
|
type Status struct {
|
||||||
Handle string
|
Handle string
|
||||||
Known, Running bool
|
Known, Running bool
|
||||||
|
@ -1,54 +0,0 @@
|
|||||||
// Copyright 2011 - 2012 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 common
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
"syscall"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
ErrJobTimeOut = errors.New("Do a job time out")
|
|
||||||
ErrInvalidData = errors.New("Invalid data")
|
|
||||||
ErrWorkWarning = errors.New("Work warning")
|
|
||||||
ErrWorkFail = errors.New("Work fail")
|
|
||||||
ErrWorkException = errors.New("Work exeption")
|
|
||||||
ErrDataType = errors.New("Invalid data type")
|
|
||||||
ErrOutOfCap = errors.New("Out of the capability")
|
|
||||||
ErrNotConn = errors.New("Did not connect to job server")
|
|
||||||
ErrFuncNotFound = errors.New("The function was not found")
|
|
||||||
ErrConnection = errors.New("Connection error")
|
|
||||||
ErrNoActiveAgent = errors.New("No active agent")
|
|
||||||
ErrTimeOut = errors.New("Executing time out")
|
|
||||||
ErrUnknown = errors.New("Unknown error")
|
|
||||||
ErrConnClosed = errors.New("Connection closed")
|
|
||||||
)
|
|
||||||
func DisablePanic() {recover()}
|
|
||||||
|
|
||||||
// Extract the error message
|
|
||||||
func GetError(data []byte) (eno syscall.Errno, err error) {
|
|
||||||
rel := bytes.SplitN(data, []byte{'\x00'}, 2)
|
|
||||||
if len(rel) != 2 {
|
|
||||||
err = Errorf("Not a error data: %V", data)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
l := len(rel[0])
|
|
||||||
eno = syscall.Errno(BytesToUint32([4]byte{rel[0][l-4], rel[0][l-3], rel[0][l-2], rel[0][l-1]}))
|
|
||||||
err = errors.New(string(rel[1]))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get a formated error
|
|
||||||
func Errorf(format string, msg ... interface{}) error {
|
|
||||||
return errors.New(fmt.Sprintf(format, msg ... ))
|
|
||||||
}
|
|
||||||
|
|
||||||
// An error handler
|
|
||||||
type ErrorHandler func(error)
|
|
||||||
|
|
||||||
|
|
@ -1,86 +0,0 @@
|
|||||||
// Copyright 2011 - 2012 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 common
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"encoding/binary"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
NETWORK = "tcp"
|
|
||||||
// queue size
|
|
||||||
QUEUE_SIZE = 8
|
|
||||||
// read buffer size
|
|
||||||
BUFFER_SIZE = 1024
|
|
||||||
// min packet length
|
|
||||||
PACKET_LEN = 12
|
|
||||||
|
|
||||||
// \x00REQ
|
|
||||||
REQ = 5391697
|
|
||||||
REQ_STR = "\x00REQ"
|
|
||||||
// \x00RES
|
|
||||||
RES = 5391699
|
|
||||||
RES_STR = "\x00RES"
|
|
||||||
|
|
||||||
// package data type
|
|
||||||
CAN_DO = 1
|
|
||||||
CANT_DO = 2
|
|
||||||
RESET_ABILITIES = 3
|
|
||||||
PRE_SLEEP = 4
|
|
||||||
NOOP = 6
|
|
||||||
JOB_CREATED = 8
|
|
||||||
GRAB_JOB = 9
|
|
||||||
NO_JOB = 10
|
|
||||||
JOB_ASSIGN = 11
|
|
||||||
WORK_STATUS = 12
|
|
||||||
WORK_COMPLETE = 13
|
|
||||||
WORK_FAIL = 14
|
|
||||||
GET_STATUS = 15
|
|
||||||
ECHO_REQ = 16
|
|
||||||
ECHO_RES = 17
|
|
||||||
ERROR = 19
|
|
||||||
STATUS_RES = 20
|
|
||||||
SET_CLIENT_ID = 22
|
|
||||||
CAN_DO_TIMEOUT = 23
|
|
||||||
WORK_EXCEPTION = 25
|
|
||||||
WORK_DATA = 28
|
|
||||||
WORK_WARNING = 29
|
|
||||||
GRAB_JOB_UNIQ = 30
|
|
||||||
JOB_ASSIGN_UNIQ = 31
|
|
||||||
|
|
||||||
SUBMIT_JOB = 7
|
|
||||||
SUBMIT_JOB_BG = 18
|
|
||||||
SUBMIT_JOB_HIGH = 21
|
|
||||||
SUBMIT_JOB_HIGH_BG = 32
|
|
||||||
SUBMIT_JOB_LOW = 33
|
|
||||||
SUBMIT_JOB_LOW_BG = 34
|
|
||||||
)
|
|
||||||
|
|
||||||
// Decode [4]byte to uint32
|
|
||||||
func BytesToUint32(buf [4]byte) uint32 {
|
|
||||||
var r uint32
|
|
||||||
b := bytes.NewBuffer(buf[:])
|
|
||||||
err := binary.Read(b, binary.BigEndian, &r)
|
|
||||||
if err != nil {
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
return r
|
|
||||||
}
|
|
||||||
|
|
||||||
// Encode uint32 to [4]byte
|
|
||||||
func Uint32ToBytes(i uint32) [4]byte {
|
|
||||||
buf := new(bytes.Buffer)
|
|
||||||
err := binary.Write(buf, binary.BigEndian, i)
|
|
||||||
if err != nil {
|
|
||||||
return [4]byte{0, 0, 0, 0}
|
|
||||||
}
|
|
||||||
var r [4]byte
|
|
||||||
for k, v := range buf.Bytes() {
|
|
||||||
r[k] = v
|
|
||||||
}
|
|
||||||
return r
|
|
||||||
}
|
|
@ -1,48 +0,0 @@
|
|||||||
package common
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
testCase = map[uint32][4]byte {
|
|
||||||
0: [...]byte{0, 0, 0, 0},
|
|
||||||
1: [...]byte{0, 0, 0, 1},
|
|
||||||
256: [...]byte{0, 0, 1, 0},
|
|
||||||
256 * 256: [...]byte{0, 1, 0, 0},
|
|
||||||
256 * 256 * 256: [...]byte{1, 0, 0, 0},
|
|
||||||
256 * 256 * 256 + 256 * 256 + 256 + 1: [...]byte{1, 1, 1, 1},
|
|
||||||
4294967295 : [...]byte{0xFF, 0xFF, 0xFF, 0xFF},
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestUint32ToBytes(t *testing.T) {
|
|
||||||
for k, v := range testCase {
|
|
||||||
b := Uint32ToBytes(k)
|
|
||||||
if bytes.Compare(b[:], v[:]) != 0 {
|
|
||||||
t.Errorf("%v was expected, but %v was got", v, b)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBytesToUint32s(t *testing.T) {
|
|
||||||
for k, v := range testCase {
|
|
||||||
u := BytesToUint32([4]byte(v))
|
|
||||||
if u != k {
|
|
||||||
t.Errorf("%v was expected, but %v was got", k, u)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func BenchmarkByteToUnit32(b * testing.B) {
|
|
||||||
for i := 0; i < b.N; i++ {
|
|
||||||
BytesToUint32([4]byte{0xF, 0xF, 0xF, 0xF});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func BenchmarkUint32ToByte(b *testing.B) {
|
|
||||||
for i := 0; i < b.N; i++ {
|
|
||||||
Uint32ToBytes(123456);
|
|
||||||
}
|
|
||||||
}
|
|
@ -10,7 +10,7 @@ The protocol was implemented by native way.
|
|||||||
package gearman
|
package gearman
|
||||||
|
|
||||||
import (
|
import (
|
||||||
_ "github.com/mikespook/gearman-go/common"
|
|
||||||
_ "github.com/mikespook/gearman-go/client"
|
_ "github.com/mikespook/gearman-go/client"
|
||||||
|
_ "github.com/mikespook/gearman-go/common"
|
||||||
_ "github.com/mikespook/gearman-go/worker"
|
_ "github.com/mikespook/gearman-go/worker"
|
||||||
)
|
)
|
||||||
|
@ -10,12 +10,12 @@ The protocol was implemented by native way.
|
|||||||
package gearman
|
package gearman
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"time"
|
|
||||||
"sync"
|
|
||||||
"testing"
|
|
||||||
"strings"
|
|
||||||
"github.com/mikespook/gearman-go/client"
|
"github.com/mikespook/gearman-go/client"
|
||||||
"github.com/mikespook/gearman-go/worker"
|
"github.com/mikespook/gearman-go/worker"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@ -33,7 +33,6 @@ func Sleep(job *worker.Job) ([]byte, error) {
|
|||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
func TestJobs(t *testing.T) {
|
func TestJobs(t *testing.T) {
|
||||||
w := worker.New(worker.Unlimited)
|
w := worker.New(worker.Unlimited)
|
||||||
if err := w.AddServer(GEARMAND); err != nil {
|
if err := w.AddServer(GEARMAND); err != nil {
|
||||||
@ -72,7 +71,7 @@ func TestJobs(t *testing.T) {
|
|||||||
var w sync.WaitGroup
|
var w sync.WaitGroup
|
||||||
jobHandler := func(job *client.Response) {
|
jobHandler := func(job *client.Response) {
|
||||||
upper := strings.ToUpper(STR)
|
upper := strings.ToUpper(STR)
|
||||||
if (string(job.Data) != upper) {
|
if string(job.Data) != upper {
|
||||||
t.Errorf("%s expected, got %s", upper, job.Data)
|
t.Errorf("%s expected, got %s", upper, job.Data)
|
||||||
}
|
}
|
||||||
w.Done()
|
w.Done()
|
||||||
|
@ -7,7 +7,6 @@ package worker
|
|||||||
import (
|
import (
|
||||||
"io"
|
"io"
|
||||||
"net"
|
"net"
|
||||||
"github.com/mikespook/gearman-go/common"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// The agent of job server.
|
// The agent of job server.
|
||||||
|
56
worker/common.go
Normal file
56
worker/common.go
Normal file
@ -0,0 +1,56 @@
|
|||||||
|
// Copyright 2011 - 2012 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 worker
|
||||||
|
|
||||||
|
const (
|
||||||
|
NETWORK = "tcp"
|
||||||
|
// queue size
|
||||||
|
QUEUE_SIZE = 8
|
||||||
|
// read buffer size
|
||||||
|
BUFFER_SIZE = 1024
|
||||||
|
// min packet length
|
||||||
|
MIN_PACKET_LEN = 12
|
||||||
|
|
||||||
|
// \x00REQ
|
||||||
|
REQ = 5391697
|
||||||
|
REQ_STR = "\x00REQ"
|
||||||
|
// \x00RES
|
||||||
|
RES = 5391699
|
||||||
|
RES_STR = "\x00RES"
|
||||||
|
|
||||||
|
// package data type
|
||||||
|
CAN_DO = 1
|
||||||
|
CANT_DO = 2
|
||||||
|
RESET_ABILITIES = 3
|
||||||
|
PRE_SLEEP = 4
|
||||||
|
NOOP = 6
|
||||||
|
JOB_CREATED = 8
|
||||||
|
GRAB_JOB = 9
|
||||||
|
NO_JOB = 10
|
||||||
|
JOB_ASSIGN = 11
|
||||||
|
WORK_STATUS = 12
|
||||||
|
WORK_COMPLETE = 13
|
||||||
|
WORK_FAIL = 14
|
||||||
|
GET_STATUS = 15
|
||||||
|
ECHO_REQ = 16
|
||||||
|
ECHO_RES = 17
|
||||||
|
ERROR = 19
|
||||||
|
STATUS_RES = 20
|
||||||
|
SET_CLIENT_ID = 22
|
||||||
|
CAN_DO_TIMEOUT = 23
|
||||||
|
WORK_EXCEPTION = 25
|
||||||
|
WORK_DATA = 28
|
||||||
|
WORK_WARNING = 29
|
||||||
|
GRAB_JOB_UNIQ = 30
|
||||||
|
JOB_ASSIGN_UNIQ = 31
|
||||||
|
|
||||||
|
SUBMIT_JOB = 7
|
||||||
|
SUBMIT_JOB_BG = 18
|
||||||
|
SUBMIT_JOB_HIGH = 21
|
||||||
|
SUBMIT_JOB_HIGH_BG = 32
|
||||||
|
SUBMIT_JOB_LOW = 33
|
||||||
|
SUBMIT_JOB_LOW_BG = 34
|
||||||
|
)
|
45
worker/error.go
Normal file
45
worker/error.go
Normal file
@ -0,0 +1,45 @@
|
|||||||
|
// Copyright 2011 - 2012 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 worker
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
ErrJobTimeOut = errors.New("Do a job time out")
|
||||||
|
ErrInvalidData = errors.New("Invalid data")
|
||||||
|
ErrWorkWarning = errors.New("Work warning")
|
||||||
|
ErrWorkFail = errors.New("Work fail")
|
||||||
|
ErrWorkException = errors.New("Work exeption")
|
||||||
|
ErrDataType = errors.New("Invalid data type")
|
||||||
|
ErrOutOfCap = errors.New("Out of the capability")
|
||||||
|
ErrNotConn = errors.New("Did not connect to job server")
|
||||||
|
ErrFuncNotFound = errors.New("The function was not found")
|
||||||
|
ErrConnection = errors.New("Connection error")
|
||||||
|
ErrNoActiveAgent = errors.New("No active agent")
|
||||||
|
ErrTimeOut = errors.New("Executing time out")
|
||||||
|
ErrUnknown = errors.New("Unknown error")
|
||||||
|
ErrConnClosed = errors.New("Connection closed")
|
||||||
|
)
|
||||||
|
|
||||||
|
func DisablePanic() { recover() }
|
||||||
|
|
||||||
|
// Extract the error message
|
||||||
|
func GetError(data []byte) (err error) {
|
||||||
|
rel := bytes.SplitN(data, []byte{'\x00'}, 2)
|
||||||
|
if len(rel) != 2 {
|
||||||
|
err = fmt.Errorf("Not a error data: %V", data)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
err = errors.New(fmt.Sprintf("%s: %s", rel[0], rel[1]))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// An error handler
|
||||||
|
type ErrorHandler func(error)
|
@ -1,8 +1,8 @@
|
|||||||
package worker
|
package worker
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"runtime"
|
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"runtime"
|
||||||
)
|
)
|
||||||
|
|
||||||
type systemInfo struct {
|
type systemInfo struct {
|
||||||
|
134
worker/job.go
134
worker/job.go
@ -1,134 +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 worker
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"strconv"
|
|
||||||
"github.com/mikespook/gearman-go/common"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Worker side job
|
|
||||||
type Job struct {
|
|
||||||
Data []byte
|
|
||||||
Handle, UniqueId, Fn string
|
|
||||||
agent *agent
|
|
||||||
magicCode, DataType uint32
|
|
||||||
c chan bool
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create a new job
|
|
||||||
func newJob(magiccode, datatype uint32, data []byte) (job *Job) {
|
|
||||||
return &Job{magicCode: magiccode,
|
|
||||||
DataType: datatype,
|
|
||||||
Data: data,
|
|
||||||
c: make(chan bool),}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Decode 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:]
|
|
||||||
job = &Job{magicCode: common.RES, DataType: datatype, c: make(chan bool),}
|
|
||||||
switch datatype {
|
|
||||||
case common.JOB_ASSIGN:
|
|
||||||
s := bytes.SplitN(data, []byte{'\x00'}, 3)
|
|
||||||
if len(s) == 3 {
|
|
||||||
job.Handle = string(s[0])
|
|
||||||
job.Fn = string(s[1])
|
|
||||||
data = s[2]
|
|
||||||
}
|
|
||||||
case common.JOB_ASSIGN_UNIQ:
|
|
||||||
s := bytes.SplitN(data, []byte{'\x00'}, 4)
|
|
||||||
if len(s) == 4 {
|
|
||||||
job.Handle = string(s[0])
|
|
||||||
job.Fn = string(s[1])
|
|
||||||
job.UniqueId = string(s[2])
|
|
||||||
data = s[3]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
job.Data = data
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Encode a job to byte slice
|
|
||||||
func (job *Job) Encode() (data []byte) {
|
|
||||||
var l int
|
|
||||||
if job.DataType == common.WORK_FAIL {
|
|
||||||
l = len(job.Handle)
|
|
||||||
} else {
|
|
||||||
l = len(job.Data)
|
|
||||||
if job.Handle != "" {
|
|
||||||
l += len(job.Handle) + 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
data = make([]byte, 0, l + 12)
|
|
||||||
|
|
||||||
magiccode := common.Uint32ToBytes(job.magicCode)
|
|
||||||
datatype := common.Uint32ToBytes(job.DataType)
|
|
||||||
datalength := common.Uint32ToBytes(uint32(l))
|
|
||||||
|
|
||||||
data = append(data, magiccode[:]...)
|
|
||||||
data = append(data, datatype[:]...)
|
|
||||||
data = append(data, datalength[:]...)
|
|
||||||
if job.Handle != "" {
|
|
||||||
data = append(data, []byte(job.Handle)...)
|
|
||||||
if job.DataType != common.WORK_FAIL {
|
|
||||||
data = append(data, 0)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
data = append(data, job.Data...)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Send some datas to client.
|
|
||||||
// Using this in a job's executing.
|
|
||||||
func (job *Job) UpdateData(data []byte, iswarning bool) {
|
|
||||||
result := append([]byte(job.Handle), 0)
|
|
||||||
result = append(result, data...)
|
|
||||||
var datatype uint32
|
|
||||||
if iswarning {
|
|
||||||
datatype = common.WORK_WARNING
|
|
||||||
} else {
|
|
||||||
datatype = common.WORK_DATA
|
|
||||||
}
|
|
||||||
job.agent.WriteJob(newJob(common.REQ, datatype, result))
|
|
||||||
}
|
|
||||||
|
|
||||||
// Update status.
|
|
||||||
// Tall client how many percent job has been executed.
|
|
||||||
func (job *Job) UpdateStatus(numerator, denominator int) {
|
|
||||||
n := []byte(strconv.Itoa(numerator))
|
|
||||||
d := []byte(strconv.Itoa(denominator))
|
|
||||||
result := append([]byte(job.Handle), '\x00')
|
|
||||||
result = append(result, n...)
|
|
||||||
result = append(result, '\x00')
|
|
||||||
result = append(result, d...)
|
|
||||||
job.agent.WriteJob(newJob(common.REQ, common.WORK_STATUS, result))
|
|
||||||
}
|
|
||||||
|
|
||||||
// close the job
|
|
||||||
func (job *Job) Close() {
|
|
||||||
close(job.c)
|
|
||||||
}
|
|
||||||
|
|
||||||
// cancel the job executing
|
|
||||||
func (job *Job) cancel() {
|
|
||||||
defer func() {recover()}()
|
|
||||||
job.c <- true
|
|
||||||
}
|
|
||||||
|
|
||||||
// When a job was canceled, return a true form a channel
|
|
||||||
func (job *Job) Canceled() <-chan bool {
|
|
||||||
return job.c
|
|
||||||
}
|
|
133
worker/request.go
Normal file
133
worker/request.go
Normal file
@ -0,0 +1,133 @@
|
|||||||
|
// 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 worker
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Worker side job
|
||||||
|
type Job struct {
|
||||||
|
Data []byte
|
||||||
|
Handle, UniqueId, Fn string
|
||||||
|
agent *agent
|
||||||
|
magicCode, DataType uint32
|
||||||
|
c chan bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a new job
|
||||||
|
func newJob(magiccode, datatype uint32, data []byte) (job *Job) {
|
||||||
|
return &Job{magicCode: magiccode,
|
||||||
|
DataType: datatype,
|
||||||
|
Data: data,
|
||||||
|
c: make(chan bool)}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode 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:]
|
||||||
|
job = &Job{magicCode: common.RES, DataType: datatype, c: make(chan bool)}
|
||||||
|
switch datatype {
|
||||||
|
case common.JOB_ASSIGN:
|
||||||
|
s := bytes.SplitN(data, []byte{'\x00'}, 3)
|
||||||
|
if len(s) == 3 {
|
||||||
|
job.Handle = string(s[0])
|
||||||
|
job.Fn = string(s[1])
|
||||||
|
data = s[2]
|
||||||
|
}
|
||||||
|
case common.JOB_ASSIGN_UNIQ:
|
||||||
|
s := bytes.SplitN(data, []byte{'\x00'}, 4)
|
||||||
|
if len(s) == 4 {
|
||||||
|
job.Handle = string(s[0])
|
||||||
|
job.Fn = string(s[1])
|
||||||
|
job.UniqueId = string(s[2])
|
||||||
|
data = s[3]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
job.Data = data
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Encode a job to byte slice
|
||||||
|
func (job *Job) Encode() (data []byte) {
|
||||||
|
var l int
|
||||||
|
if job.DataType == common.WORK_FAIL {
|
||||||
|
l = len(job.Handle)
|
||||||
|
} else {
|
||||||
|
l = len(job.Data)
|
||||||
|
if job.Handle != "" {
|
||||||
|
l += len(job.Handle) + 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
data = make([]byte, 0, l+12)
|
||||||
|
|
||||||
|
magiccode := common.Uint32ToBytes(job.magicCode)
|
||||||
|
datatype := common.Uint32ToBytes(job.DataType)
|
||||||
|
datalength := common.Uint32ToBytes(uint32(l))
|
||||||
|
|
||||||
|
data = append(data, magiccode[:]...)
|
||||||
|
data = append(data, datatype[:]...)
|
||||||
|
data = append(data, datalength[:]...)
|
||||||
|
if job.Handle != "" {
|
||||||
|
data = append(data, []byte(job.Handle)...)
|
||||||
|
if job.DataType != common.WORK_FAIL {
|
||||||
|
data = append(data, 0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
data = append(data, job.Data...)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Send some datas to client.
|
||||||
|
// Using this in a job's executing.
|
||||||
|
func (job *Job) UpdateData(data []byte, iswarning bool) {
|
||||||
|
result := append([]byte(job.Handle), 0)
|
||||||
|
result = append(result, data...)
|
||||||
|
var datatype uint32
|
||||||
|
if iswarning {
|
||||||
|
datatype = common.WORK_WARNING
|
||||||
|
} else {
|
||||||
|
datatype = common.WORK_DATA
|
||||||
|
}
|
||||||
|
job.agent.WriteJob(newJob(common.REQ, datatype, result))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update status.
|
||||||
|
// Tall client how many percent job has been executed.
|
||||||
|
func (job *Job) UpdateStatus(numerator, denominator int) {
|
||||||
|
n := []byte(strconv.Itoa(numerator))
|
||||||
|
d := []byte(strconv.Itoa(denominator))
|
||||||
|
result := append([]byte(job.Handle), '\x00')
|
||||||
|
result = append(result, n...)
|
||||||
|
result = append(result, '\x00')
|
||||||
|
result = append(result, d...)
|
||||||
|
job.agent.WriteJob(newJob(common.REQ, common.WORK_STATUS, result))
|
||||||
|
}
|
||||||
|
|
||||||
|
// close the job
|
||||||
|
func (job *Job) Close() {
|
||||||
|
close(job.c)
|
||||||
|
}
|
||||||
|
|
||||||
|
// cancel the job executing
|
||||||
|
func (job *Job) cancel() {
|
||||||
|
defer func() { recover() }()
|
||||||
|
job.c <- true
|
||||||
|
}
|
||||||
|
|
||||||
|
// When a job was canceled, return a true form a channel
|
||||||
|
func (job *Job) Canceled() <-chan bool {
|
||||||
|
return job.c
|
||||||
|
}
|
133
worker/response.go
Normal file
133
worker/response.go
Normal file
@ -0,0 +1,133 @@
|
|||||||
|
// 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 worker
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Worker side job
|
||||||
|
type Job struct {
|
||||||
|
Data []byte
|
||||||
|
Handle, UniqueId, Fn string
|
||||||
|
agent *agent
|
||||||
|
magicCode, DataType uint32
|
||||||
|
c chan bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a new job
|
||||||
|
func newJob(magiccode, datatype uint32, data []byte) (job *Job) {
|
||||||
|
return &Job{magicCode: magiccode,
|
||||||
|
DataType: datatype,
|
||||||
|
Data: data,
|
||||||
|
c: make(chan bool)}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode 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:]
|
||||||
|
job = &Job{magicCode: common.RES, DataType: datatype, c: make(chan bool)}
|
||||||
|
switch datatype {
|
||||||
|
case common.JOB_ASSIGN:
|
||||||
|
s := bytes.SplitN(data, []byte{'\x00'}, 3)
|
||||||
|
if len(s) == 3 {
|
||||||
|
job.Handle = string(s[0])
|
||||||
|
job.Fn = string(s[1])
|
||||||
|
data = s[2]
|
||||||
|
}
|
||||||
|
case common.JOB_ASSIGN_UNIQ:
|
||||||
|
s := bytes.SplitN(data, []byte{'\x00'}, 4)
|
||||||
|
if len(s) == 4 {
|
||||||
|
job.Handle = string(s[0])
|
||||||
|
job.Fn = string(s[1])
|
||||||
|
job.UniqueId = string(s[2])
|
||||||
|
data = s[3]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
job.Data = data
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Encode a job to byte slice
|
||||||
|
func (job *Job) Encode() (data []byte) {
|
||||||
|
var l int
|
||||||
|
if job.DataType == common.WORK_FAIL {
|
||||||
|
l = len(job.Handle)
|
||||||
|
} else {
|
||||||
|
l = len(job.Data)
|
||||||
|
if job.Handle != "" {
|
||||||
|
l += len(job.Handle) + 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
data = make([]byte, 0, l+12)
|
||||||
|
|
||||||
|
magiccode := common.Uint32ToBytes(job.magicCode)
|
||||||
|
datatype := common.Uint32ToBytes(job.DataType)
|
||||||
|
datalength := common.Uint32ToBytes(uint32(l))
|
||||||
|
|
||||||
|
data = append(data, magiccode[:]...)
|
||||||
|
data = append(data, datatype[:]...)
|
||||||
|
data = append(data, datalength[:]...)
|
||||||
|
if job.Handle != "" {
|
||||||
|
data = append(data, []byte(job.Handle)...)
|
||||||
|
if job.DataType != common.WORK_FAIL {
|
||||||
|
data = append(data, 0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
data = append(data, job.Data...)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Send some datas to client.
|
||||||
|
// Using this in a job's executing.
|
||||||
|
func (job *Job) UpdateData(data []byte, iswarning bool) {
|
||||||
|
result := append([]byte(job.Handle), 0)
|
||||||
|
result = append(result, data...)
|
||||||
|
var datatype uint32
|
||||||
|
if iswarning {
|
||||||
|
datatype = common.WORK_WARNING
|
||||||
|
} else {
|
||||||
|
datatype = common.WORK_DATA
|
||||||
|
}
|
||||||
|
job.agent.WriteJob(newJob(common.REQ, datatype, result))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update status.
|
||||||
|
// Tall client how many percent job has been executed.
|
||||||
|
func (job *Job) UpdateStatus(numerator, denominator int) {
|
||||||
|
n := []byte(strconv.Itoa(numerator))
|
||||||
|
d := []byte(strconv.Itoa(denominator))
|
||||||
|
result := append([]byte(job.Handle), '\x00')
|
||||||
|
result = append(result, n...)
|
||||||
|
result = append(result, '\x00')
|
||||||
|
result = append(result, d...)
|
||||||
|
job.agent.WriteJob(newJob(common.REQ, common.WORK_STATUS, result))
|
||||||
|
}
|
||||||
|
|
||||||
|
// close the job
|
||||||
|
func (job *Job) Close() {
|
||||||
|
close(job.c)
|
||||||
|
}
|
||||||
|
|
||||||
|
// cancel the job executing
|
||||||
|
func (job *Job) cancel() {
|
||||||
|
defer func() { recover() }()
|
||||||
|
job.c <- true
|
||||||
|
}
|
||||||
|
|
||||||
|
// When a job was canceled, return a true form a channel
|
||||||
|
func (job *Job) Canceled() <-chan bool {
|
||||||
|
return job.c
|
||||||
|
}
|
@ -6,7 +6,6 @@ package worker
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"time"
|
"time"
|
||||||
"github.com/mikespook/gearman-go/common"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@ -19,6 +18,7 @@ const (
|
|||||||
var (
|
var (
|
||||||
ErrConnection = common.ErrConnection
|
ErrConnection = common.ErrConnection
|
||||||
)
|
)
|
||||||
|
|
||||||
// Job handler
|
// Job handler
|
||||||
type JobHandler func(*Job) error
|
type JobHandler func(*Job) error
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user