Added some comments.
This commit is contained in:
parent
2d62a1d5d1
commit
7eb966bc36
3
README
3
README
@ -19,5 +19,4 @@ It was implemented a native protocol for both worker and client API.
|
|||||||
----
|
----
|
||||||
xingxing<mikespook@gmail.com>
|
xingxing<mikespook@gmail.com>
|
||||||
http://mikespook.com
|
http://mikespook.com
|
||||||
|
http://twitter.com/mikespook
|
||||||
|
|
||||||
|
@ -1,3 +1,7 @@
|
|||||||
|
// 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
|
package gearman
|
||||||
|
|
||||||
import (
|
import (
|
||||||
@ -8,6 +12,15 @@ import (
|
|||||||
"strconv"
|
"strconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
/*
|
||||||
|
The client side api for gearman.
|
||||||
|
|
||||||
|
usage:
|
||||||
|
client = NewClient()
|
||||||
|
client.AddServer("127.0.0.1:4730")
|
||||||
|
handle := client.Do("foobar", []byte("data here"), JOB_LOW | JOB_BG)
|
||||||
|
|
||||||
|
*/
|
||||||
type Client struct {
|
type Client struct {
|
||||||
mutex sync.Mutex
|
mutex sync.Mutex
|
||||||
conn net.Conn
|
conn net.Conn
|
||||||
@ -16,6 +29,7 @@ type Client struct {
|
|||||||
UId uint32
|
UId uint32
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Create a new client.
|
||||||
func NewClient() (client * Client){
|
func NewClient() (client * Client){
|
||||||
client = &Client{JobQueue:make(chan *ClientJob, QUEUE_CAP),
|
client = &Client{JobQueue:make(chan *ClientJob, QUEUE_CAP),
|
||||||
incoming:make(chan []byte, QUEUE_CAP),
|
incoming:make(chan []byte, QUEUE_CAP),
|
||||||
@ -23,6 +37,9 @@ func NewClient() (client * Client){
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Add a server.
|
||||||
|
// In this version, one client connect to one job server.
|
||||||
|
// Sample is better. Plz do the load balancing by your self.
|
||||||
func (client *Client) AddServer(addr string) (err os.Error) {
|
func (client *Client) AddServer(addr string) (err os.Error) {
|
||||||
conn, err := net.Dial(TCP, addr)
|
conn, err := net.Dial(TCP, addr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -32,10 +49,13 @@ func (client *Client) AddServer(addr string) (err os.Error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Internal read
|
||||||
func (client *Client) read() (data []byte, err os.Error) {
|
func (client *Client) read() (data []byte, err os.Error) {
|
||||||
if len(client.incoming) > 0 {
|
if len(client.incoming) > 0 {
|
||||||
|
// incoming queue is not empty
|
||||||
data = <-client.incoming
|
data = <-client.incoming
|
||||||
} else {
|
} else {
|
||||||
|
// empty queue, read data from socket
|
||||||
for {
|
for {
|
||||||
buf := make([]byte, BUFFER_SIZE)
|
buf := make([]byte, BUFFER_SIZE)
|
||||||
var n int
|
var n int
|
||||||
@ -51,6 +71,7 @@ func (client *Client) read() (data []byte, err os.Error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
// split package
|
||||||
start, end := 0, 4
|
start, end := 0, 4
|
||||||
for i := 0; i < len(data); i ++{
|
for i := 0; i < len(data); i ++{
|
||||||
if string(data[start:end]) == RES_STR {
|
if string(data[start:end]) == RES_STR {
|
||||||
@ -72,6 +93,8 @@ func (client *Client) read() (data []byte, err os.Error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Read a job from job server.
|
||||||
|
// This function will return the job, and add it to the job queue.
|
||||||
func (client *Client) ReadJob() (job *ClientJob, err os.Error) {
|
func (client *Client) ReadJob() (job *ClientJob, err os.Error) {
|
||||||
var rel []byte
|
var rel []byte
|
||||||
if rel, err = client.read(); err != nil {
|
if rel, err = client.read(); err != nil {
|
||||||
@ -91,6 +114,12 @@ func (client *Client) ReadJob() (job *ClientJob, err os.Error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Do the function.
|
||||||
|
// funcname is a string with function name.
|
||||||
|
// data is encoding to byte array.
|
||||||
|
// flag set the job type, include running level: JOB_LOW, JOB_NORMAL, JOB_HIGH,
|
||||||
|
// and if it is background job: JOB_BG.
|
||||||
|
// JOB_LOW | JOB_BG means the job is running with low level in background.
|
||||||
func (client *Client) Do(funcname string, data []byte, flag byte) (handle string, err os.Error) {
|
func (client *Client) Do(funcname string, data []byte, flag byte) (handle string, err os.Error) {
|
||||||
var datatype uint32
|
var datatype uint32
|
||||||
if flag & JOB_LOW == JOB_LOW {
|
if flag & JOB_LOW == JOB_LOW {
|
||||||
@ -147,6 +176,7 @@ func (client *Client) Do(funcname string, data []byte, flag byte) (handle string
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Internal read last job
|
||||||
func (client *Client) readLastJob(datatype uint32) (job *ClientJob, err os.Error){
|
func (client *Client) readLastJob(datatype uint32) (job *ClientJob, err os.Error){
|
||||||
for {
|
for {
|
||||||
if job, err = client.ReadJob(); err != nil {
|
if job, err = client.ReadJob(); err != nil {
|
||||||
@ -162,6 +192,8 @@ func (client *Client) readLastJob(datatype uint32) (job *ClientJob, err os.Error
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Get job status from job server.
|
||||||
|
// !!!Not fully tested.!!!
|
||||||
func (client *Client) Status(handle string) (known, running bool, numerator, denominator uint, err os.Error) {
|
func (client *Client) Status(handle string) (known, running bool, numerator, denominator uint, err os.Error) {
|
||||||
|
|
||||||
if err = client.WriteJob(NewClientJob(REQ, GET_STATUS, []byte(handle))); err != nil {
|
if err = client.WriteJob(NewClientJob(REQ, GET_STATUS, []byte(handle))); err != nil {
|
||||||
@ -191,6 +223,7 @@ func (client *Client) Status(handle string) (known, running bool, numerator, den
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Send a something out, get the samething back.
|
||||||
func (client *Client) Echo(data []byte) (echo []byte, err os.Error) {
|
func (client *Client) Echo(data []byte) (echo []byte, err os.Error) {
|
||||||
if err = client.WriteJob(NewClientJob(REQ, ECHO_REQ, data)); err != nil {
|
if err = client.WriteJob(NewClientJob(REQ, ECHO_REQ, data)); err != nil {
|
||||||
return
|
return
|
||||||
@ -203,7 +236,10 @@ func (client *Client) Echo(data []byte) (echo []byte, err os.Error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (client *Client) LastResult() (job *ClientJob) {
|
// Get the last job.
|
||||||
|
// the job means a network package.
|
||||||
|
// Normally, it is the job executed result.
|
||||||
|
func (client *Client) LastJob() (job *ClientJob) {
|
||||||
if l := len(client.JobQueue); l != 1 {
|
if l := len(client.JobQueue); l != 1 {
|
||||||
if l == 0 {
|
if l == 0 {
|
||||||
return
|
return
|
||||||
@ -215,10 +251,12 @@ func (client *Client) LastResult() (job *ClientJob) {
|
|||||||
return <-client.JobQueue
|
return <-client.JobQueue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Send the job to job server.
|
||||||
func (client *Client) WriteJob(job *ClientJob) (err os.Error) {
|
func (client *Client) WriteJob(job *ClientJob) (err os.Error) {
|
||||||
return client.write(job.Encode())
|
return client.write(job.Encode())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Internal write
|
||||||
func (client *Client) write(buf []byte) (err os.Error) {
|
func (client *Client) write(buf []byte) (err os.Error) {
|
||||||
var n int
|
var n int
|
||||||
for i := 0; i < len(buf); i += n {
|
for i := 0; i < len(buf); i += n {
|
||||||
@ -230,6 +268,7 @@ func (client *Client) write(buf []byte) (err os.Error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Close.
|
||||||
func (client *Client) Close() (err os.Error) {
|
func (client *Client) Close() (err os.Error) {
|
||||||
err = client.conn.Close()
|
err = client.conn.Close()
|
||||||
close(client.JobQueue)
|
close(client.JobQueue)
|
||||||
|
@ -1,3 +1,7 @@
|
|||||||
|
// 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
|
package gearman
|
||||||
|
|
||||||
import (
|
import (
|
||||||
@ -5,18 +9,21 @@ import (
|
|||||||
// "log"
|
// "log"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Client side job
|
||||||
type ClientJob struct {
|
type ClientJob struct {
|
||||||
Data []byte
|
Data []byte
|
||||||
Handle, UniqueId string
|
Handle, UniqueId string
|
||||||
magicCode, DataType uint32
|
magicCode, DataType uint32
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Create a new job
|
||||||
func NewClientJob(magiccode, datatype uint32, data []byte) (job *ClientJob) {
|
func NewClientJob(magiccode, datatype uint32, data []byte) (job *ClientJob) {
|
||||||
return &ClientJob{magicCode:magiccode,
|
return &ClientJob{magicCode:magiccode,
|
||||||
DataType:datatype,
|
DataType:datatype,
|
||||||
Data:data}
|
Data:data}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Decode a job from byte slice
|
||||||
func DecodeClientJob(data []byte) (job * ClientJob, err os.Error) {
|
func DecodeClientJob(data []byte) (job * ClientJob, err os.Error) {
|
||||||
if len(data) < 12 {
|
if len(data) < 12 {
|
||||||
err = os.NewError("Data length is too small.")
|
err = os.NewError("Data length is too small.")
|
||||||
@ -33,6 +40,7 @@ func DecodeClientJob(data []byte) (job * ClientJob, err os.Error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Encode a job to byte slice
|
||||||
func (job *ClientJob) Encode() (data []byte) {
|
func (job *ClientJob) Encode() (data []byte) {
|
||||||
magiccode := uint32ToByte(job.magicCode)
|
magiccode := uint32ToByte(job.magicCode)
|
||||||
datatype := uint32ToByte(job.DataType)
|
datatype := uint32ToByte(job.DataType)
|
||||||
@ -46,6 +54,7 @@ func (job *ClientJob) Encode() (data []byte) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Extract the job's result.
|
||||||
func (job * ClientJob) Result() (data []byte, err os.Error){
|
func (job * ClientJob) Result() (data []byte, err os.Error){
|
||||||
switch job.DataType {
|
switch job.DataType {
|
||||||
case WORK_FAIL:
|
case WORK_FAIL:
|
||||||
@ -69,6 +78,7 @@ func (job * ClientJob) Result() (data []byte, err os.Error){
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Extract the job's update
|
||||||
func (job *ClientJob) Update() (data []byte, err os.Error) {
|
func (job *ClientJob) Update() (data []byte, err os.Error) {
|
||||||
if job.DataType != WORK_DATA && job.DataType != WORK_WARNING {
|
if job.DataType != WORK_DATA && job.DataType != WORK_WARNING {
|
||||||
err = os.NewError("The job is not a update.")
|
err = os.NewError("The job is not a update.")
|
||||||
|
@ -1,3 +1,12 @@
|
|||||||
|
// 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.
|
||||||
|
|
||||||
|
/*
|
||||||
|
This module is Gearman API for golang.
|
||||||
|
The protocol was implemented by native way.
|
||||||
|
*/
|
||||||
|
|
||||||
package gearman
|
package gearman
|
||||||
|
|
||||||
import (
|
import (
|
||||||
@ -5,20 +14,27 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
|
// tcp4 is tested. You can modify this to 'tcp' for both ipv4 and ipv6,
|
||||||
|
// or 'tcp6' only for ipv6.
|
||||||
TCP = "tcp4"
|
TCP = "tcp4"
|
||||||
|
// the number limited for job servers.
|
||||||
WORKER_SERVER_CAP = 32
|
WORKER_SERVER_CAP = 32
|
||||||
|
// the number limited for functions.
|
||||||
WORKER_FUNCTION_CAP = 512
|
WORKER_FUNCTION_CAP = 512
|
||||||
|
// queue size
|
||||||
QUEUE_CAP = 512
|
QUEUE_CAP = 512
|
||||||
|
// read buffer size
|
||||||
BUFFER_SIZE = 1024
|
BUFFER_SIZE = 1024
|
||||||
|
|
||||||
|
|
||||||
// \x00REQ
|
// \x00REQ
|
||||||
REQ = 5391697
|
REQ = 5391697
|
||||||
REQ_STR = "\x00REQ"
|
REQ_STR = "\x00REQ"
|
||||||
// \x00RES
|
// \x00RES
|
||||||
RES = 5391699
|
RES = 5391699
|
||||||
RES_STR = "\x00RES"
|
RES_STR = "\x00RES"
|
||||||
|
|
||||||
|
// package data type
|
||||||
CAN_DO = 1
|
CAN_DO = 1
|
||||||
CANT_DO = 2
|
CANT_DO = 2
|
||||||
RESET_ABILITIES = 3
|
RESET_ABILITIES = 3
|
||||||
@ -51,16 +67,24 @@ const (
|
|||||||
SUBMIT_JOB_LOW = 33
|
SUBMIT_JOB_LOW = 33
|
||||||
SUBMIT_JOB_LOW_BG = 34
|
SUBMIT_JOB_LOW_BG = 34
|
||||||
|
|
||||||
|
// Job type
|
||||||
|
// JOB_NORMAL | JOB_BG means a normal level job run in background
|
||||||
|
// normal level
|
||||||
JOB_NORMAL = 0
|
JOB_NORMAL = 0
|
||||||
|
// background job
|
||||||
JOB_BG = 1
|
JOB_BG = 1
|
||||||
|
// low level
|
||||||
JOB_LOW = 2
|
JOB_LOW = 2
|
||||||
|
// high level
|
||||||
JOB_HIGH = 4
|
JOB_HIGH = 4
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// No use
|
||||||
type Job interface {
|
type Job interface {
|
||||||
Encode() []byte
|
Encode() []byte
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Splite the byte array by a byte
|
||||||
func splitByteArray(slice []byte, spot byte) (data [][]byte){
|
func splitByteArray(slice []byte, spot byte) (data [][]byte){
|
||||||
data = make([][]byte, 0, 10)
|
data = make([][]byte, 0, 10)
|
||||||
start, end := 0, 0
|
start, end := 0, 0
|
||||||
@ -78,6 +102,7 @@ func splitByteArray(slice []byte, spot byte) (data [][]byte){
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Extract the error message
|
||||||
func getError(data []byte) (eno os.Errno, err os.Error) {
|
func getError(data []byte) (eno os.Errno, err os.Error) {
|
||||||
rel := splitByteArray(data, '\x00')
|
rel := splitByteArray(data, '\x00')
|
||||||
if len(rel) != 2 {
|
if len(rel) != 2 {
|
||||||
@ -90,6 +115,7 @@ func getError(data []byte) (eno os.Errno, err os.Error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Decode [4]byte to uint32
|
||||||
func byteToUint32(buf [4]byte) uint32 {
|
func byteToUint32(buf [4]byte) uint32 {
|
||||||
return uint32(buf[0]) << 24 +
|
return uint32(buf[0]) << 24 +
|
||||||
uint32(buf[1]) << 16 +
|
uint32(buf[1]) << 16 +
|
||||||
@ -97,6 +123,7 @@ func byteToUint32(buf [4]byte) uint32 {
|
|||||||
uint32(buf[3])
|
uint32(buf[3])
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Encode uint32 to [4]byte
|
||||||
func uint32ToByte(i uint32) (data [4]byte) {
|
func uint32ToByte(i uint32) (data [4]byte) {
|
||||||
data[0] = byte((i >> 24) & 0xff)
|
data[0] = byte((i >> 24) & 0xff)
|
||||||
data[1] = byte((i >> 16) & 0xff)
|
data[1] = byte((i >> 16) & 0xff)
|
||||||
|
@ -1,3 +1,7 @@
|
|||||||
|
// 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
|
package gearman
|
||||||
|
|
||||||
import(
|
import(
|
||||||
@ -6,9 +10,29 @@ import(
|
|||||||
// "log"
|
// "log"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// The definition of the callback function.
|
||||||
type JobFunction func(job *WorkerJob) ([]byte, os.Error)
|
type JobFunction func(job *WorkerJob) ([]byte, os.Error)
|
||||||
|
// Map for added function.
|
||||||
type JobFunctionMap map[string]JobFunction
|
type JobFunctionMap map[string]JobFunction
|
||||||
|
|
||||||
|
/*
|
||||||
|
Worker side api for gearman.
|
||||||
|
|
||||||
|
usage:
|
||||||
|
worker = NewWorker()
|
||||||
|
worker.AddFunction("foobar", foobar)
|
||||||
|
worker.AddServer("127.0.0.1:4730")
|
||||||
|
worker.Work() // Enter the worker's main loop
|
||||||
|
|
||||||
|
The definition of the callback function 'foobar' should suit for the type 'JobFunction'.
|
||||||
|
It looks like this:
|
||||||
|
|
||||||
|
func foobar(job *WorkerJob) (data []byte, err os.Error) {
|
||||||
|
//sth. here
|
||||||
|
//plaplapla...
|
||||||
|
return
|
||||||
|
}
|
||||||
|
*/
|
||||||
type Worker struct {
|
type Worker struct {
|
||||||
clients []*jobClient
|
clients []*jobClient
|
||||||
functions JobFunctionMap
|
functions JobFunctionMap
|
||||||
@ -20,6 +44,7 @@ type Worker struct {
|
|||||||
ErrQueue chan os.Error
|
ErrQueue chan os.Error
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Get a new worker
|
||||||
func NewWorker() (worker *Worker) {
|
func NewWorker() (worker *Worker) {
|
||||||
worker = &Worker{
|
worker = &Worker{
|
||||||
// job server list
|
// job server list
|
||||||
@ -34,8 +59,8 @@ func NewWorker() (worker *Worker) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// add server
|
// Add a server. The addr should be 'host:port' format.
|
||||||
// worker.AddServer("127.0.0.1:4730")
|
// The connection is established at this time.
|
||||||
func (worker * Worker) AddServer(addr string) (err os.Error) {
|
func (worker * Worker) AddServer(addr string) (err os.Error) {
|
||||||
worker.mutex.Lock()
|
worker.mutex.Lock()
|
||||||
defer worker.mutex.Unlock()
|
defer worker.mutex.Unlock()
|
||||||
@ -57,12 +82,11 @@ func (worker * Worker) AddServer(addr string) (err os.Error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// add function
|
// Add a function.
|
||||||
|
// Plz added job servers first, then functions.
|
||||||
|
// The API will tell every connected job server that 'I can do this'
|
||||||
func (worker * Worker) AddFunction(funcname string,
|
func (worker * Worker) AddFunction(funcname string,
|
||||||
f JobFunction, timeout uint32) (err os.Error) {
|
f JobFunction, timeout uint32) (err os.Error) {
|
||||||
if f == nil {
|
|
||||||
return os.NewError("Job function should not be nil.")
|
|
||||||
}
|
|
||||||
if len(worker.clients) < 1 {
|
if len(worker.clients) < 1 {
|
||||||
return os.NewError("Did not connect to Job Server.")
|
return os.NewError("Did not connect to Job Server.")
|
||||||
}
|
}
|
||||||
@ -86,7 +110,8 @@ func (worker * Worker) AddFunction(funcname string,
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// remove function
|
// Remove a function.
|
||||||
|
// Tell job servers 'I can not do this now' at the same time.
|
||||||
func (worker * Worker) RemoveFunction(funcname string) (err os.Error) {
|
func (worker * Worker) RemoveFunction(funcname string) (err os.Error) {
|
||||||
worker.mutex.Lock()
|
worker.mutex.Lock()
|
||||||
defer worker.mutex.Unlock()
|
defer worker.mutex.Unlock()
|
||||||
@ -100,7 +125,7 @@ func (worker * Worker) RemoveFunction(funcname string) (err os.Error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// work
|
// Main loop
|
||||||
func (worker * Worker) Work() {
|
func (worker * Worker) Work() {
|
||||||
for _, v := range worker.clients {
|
for _, v := range worker.clients {
|
||||||
go v.Work()
|
go v.Work()
|
||||||
@ -118,10 +143,11 @@ func (worker * Worker) Work() {
|
|||||||
_, err := getError(job.Data)
|
_, err := getError(job.Data)
|
||||||
worker.ErrQueue <- err
|
worker.ErrQueue <- err
|
||||||
case JOB_ASSIGN, JOB_ASSIGN_UNIQ:
|
case JOB_ASSIGN, JOB_ASSIGN_UNIQ:
|
||||||
if err := worker.exec(job); err != nil {
|
go func() {
|
||||||
worker.ErrQueue <- err
|
if err := worker.exec(job); err != nil {
|
||||||
continue
|
worker.ErrQueue <- err
|
||||||
}
|
}
|
||||||
|
}()
|
||||||
default:
|
default:
|
||||||
worker.JobQueue <- job
|
worker.JobQueue <- job
|
||||||
}
|
}
|
||||||
@ -129,7 +155,11 @@ func (worker * Worker) Work() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (worker * Worker) LastResult() (job *WorkerJob) {
|
// Get the last job in queue.
|
||||||
|
// If there are more than one job in the queue,
|
||||||
|
// the last one will be returned,
|
||||||
|
// the others will be lost.
|
||||||
|
func (worker * Worker) LastJob() (job *WorkerJob) {
|
||||||
if l := len(worker.JobQueue); l != 1 {
|
if l := len(worker.JobQueue); l != 1 {
|
||||||
if l == 0 {
|
if l == 0 {
|
||||||
return
|
return
|
||||||
@ -141,8 +171,7 @@ func (worker * Worker) LastResult() (job *WorkerJob) {
|
|||||||
return <-worker.JobQueue
|
return <-worker.JobQueue
|
||||||
}
|
}
|
||||||
|
|
||||||
// Close
|
// Close.
|
||||||
// should used as defer
|
|
||||||
func (worker * Worker) Close() (err os.Error){
|
func (worker * Worker) Close() (err os.Error){
|
||||||
worker.running = false
|
worker.running = false
|
||||||
for _, v := range worker.clients {
|
for _, v := range worker.clients {
|
||||||
@ -152,6 +181,9 @@ func (worker * Worker) Close() (err os.Error){
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Write a job to job server.
|
||||||
|
// Here, the job's mean is not the oraginal mean.
|
||||||
|
// Just looks like a network package for job's result or tell job server, there was a fail.
|
||||||
func (worker * Worker) WriteJob(job *WorkerJob) (err os.Error) {
|
func (worker * Worker) WriteJob(job *WorkerJob) (err os.Error) {
|
||||||
e := make(chan os.Error)
|
e := make(chan os.Error)
|
||||||
for _, v := range worker.clients {
|
for _, v := range worker.clients {
|
||||||
@ -162,13 +194,14 @@ func (worker * Worker) WriteJob(job *WorkerJob) (err os.Error) {
|
|||||||
return <- e
|
return <- e
|
||||||
}
|
}
|
||||||
|
|
||||||
// Echo
|
// Send a something out, get the samething back.
|
||||||
func (worker * Worker) Echo(data []byte) (err os.Error) {
|
func (worker * Worker) Echo(data []byte) (err os.Error) {
|
||||||
job := NewWorkerJob(REQ, ECHO_REQ, data)
|
job := NewWorkerJob(REQ, ECHO_REQ, data)
|
||||||
return worker.WriteJob(job)
|
return worker.WriteJob(job)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Reset
|
// Remove all of functions.
|
||||||
|
// Both from the worker or job servers.
|
||||||
func (worker * Worker) Reset() (err os.Error){
|
func (worker * Worker) Reset() (err os.Error){
|
||||||
job := NewWorkerJob(REQ, RESET_ABILITIES, nil)
|
job := NewWorkerJob(REQ, RESET_ABILITIES, nil)
|
||||||
err = worker.WriteJob(job)
|
err = worker.WriteJob(job)
|
||||||
@ -176,13 +209,13 @@ func (worker * Worker) Reset() (err os.Error){
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetId
|
// Set the worker's unique id.
|
||||||
func (worker * Worker) SetId(id string) (err os.Error) {
|
func (worker * Worker) SetId(id string) (err os.Error) {
|
||||||
job := NewWorkerJob(REQ, SET_CLIENT_ID, []byte(id))
|
job := NewWorkerJob(REQ, SET_CLIENT_ID, []byte(id))
|
||||||
return worker.WriteJob(job)
|
return worker.WriteJob(job)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Exec
|
// Execute the job. And send back the result.
|
||||||
func (worker * Worker) exec(job *WorkerJob) (err os.Error) {
|
func (worker * Worker) exec(job *WorkerJob) (err os.Error) {
|
||||||
jobdata := splitByteArray(job.Data, '\x00')
|
jobdata := splitByteArray(job.Data, '\x00')
|
||||||
job.Handle = string(jobdata[0])
|
job.Handle = string(jobdata[0])
|
||||||
|
@ -1,10 +1,16 @@
|
|||||||
|
// 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
|
package gearman
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"os"
|
"os"
|
||||||
|
"strconv"
|
||||||
// "log"
|
// "log"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Worker side job
|
||||||
type WorkerJob struct {
|
type WorkerJob struct {
|
||||||
Data []byte
|
Data []byte
|
||||||
Handle, UniqueId string
|
Handle, UniqueId string
|
||||||
@ -13,12 +19,14 @@ type WorkerJob struct {
|
|||||||
Job
|
Job
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Create a new job
|
||||||
func NewWorkerJob(magiccode, datatype uint32, data []byte) (job *WorkerJob) {
|
func NewWorkerJob(magiccode, datatype uint32, data []byte) (job *WorkerJob) {
|
||||||
return &WorkerJob{magicCode:magiccode,
|
return &WorkerJob{magicCode:magiccode,
|
||||||
DataType: datatype,
|
DataType: datatype,
|
||||||
Data:data}
|
Data:data}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Decode job from byte slice
|
||||||
func DecodeWorkerJob(data []byte) (job *WorkerJob, err os.Error) {
|
func DecodeWorkerJob(data []byte) (job *WorkerJob, err os.Error) {
|
||||||
if len(data) < 12 {
|
if len(data) < 12 {
|
||||||
err = os.NewError("Data length is too small.")
|
err = os.NewError("Data length is too small.")
|
||||||
@ -35,6 +43,7 @@ func DecodeWorkerJob(data []byte) (job *WorkerJob, err os.Error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Encode a job to byte slice
|
||||||
func (job *WorkerJob) Encode() (data []byte) {
|
func (job *WorkerJob) Encode() (data []byte) {
|
||||||
magiccode := uint32ToByte(job.magicCode)
|
magiccode := uint32ToByte(job.magicCode)
|
||||||
datatype := uint32ToByte(job.DataType)
|
datatype := uint32ToByte(job.DataType)
|
||||||
@ -54,7 +63,8 @@ func (job *WorkerJob) Encode() (data []byte) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// update data
|
// Send some datas to client.
|
||||||
|
// Using this in a job's executing.
|
||||||
func (job * WorkerJob) UpdateData(data []byte, iswaring bool) (err os.Error) {
|
func (job * WorkerJob) UpdateData(data []byte, iswaring bool) (err os.Error) {
|
||||||
result := append([]byte(job.Handle), 0)
|
result := append([]byte(job.Handle), 0)
|
||||||
result = append(result, data ...)
|
result = append(result, data ...)
|
||||||
@ -67,13 +77,14 @@ func (job * WorkerJob) UpdateData(data []byte, iswaring bool) (err os.Error) {
|
|||||||
return job.client.WriteJob(NewWorkerJob(REQ, datatype, result))
|
return job.client.WriteJob(NewWorkerJob(REQ, datatype, result))
|
||||||
}
|
}
|
||||||
|
|
||||||
// update status
|
// Update status.
|
||||||
func (job * WorkerJob) UpdateStatus(numerator, denominator uint32) (err os.Error) {
|
// Tall client how many percent job has been executed.
|
||||||
n := uint32ToByte(numerator)
|
func (job * WorkerJob) UpdateStatus(numerator, denominator int) (err os.Error) {
|
||||||
d := uint32ToByte(denominator)
|
n := []byte(strconv.Itoa(numerator))
|
||||||
|
d := []byte(strconv.Itoa(denominator))
|
||||||
result := append([]byte(job.Handle), 0)
|
result := append([]byte(job.Handle), 0)
|
||||||
result = append(result, n[:] ...)
|
result = append(result, n ...)
|
||||||
result = append(result, d[:] ...)
|
result = append(result, d ...)
|
||||||
return job.client.WriteJob(NewWorkerJob(REQ, WORK_STATUS, result))
|
return job.client.WriteJob(NewWorkerJob(REQ, WORK_STATUS, result))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,3 +1,7 @@
|
|||||||
|
// 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
|
package gearman
|
||||||
|
|
||||||
import (
|
import (
|
||||||
@ -6,12 +10,14 @@ import (
|
|||||||
// "log"
|
// "log"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// The client of job server.
|
||||||
type jobClient struct {
|
type jobClient struct {
|
||||||
conn net.Conn
|
conn net.Conn
|
||||||
worker *Worker
|
worker *Worker
|
||||||
running bool
|
running bool
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Create the client of job server.
|
||||||
func newJobClient(addr string, worker *Worker) (jobclient *jobClient, err os.Error) {
|
func newJobClient(addr string, worker *Worker) (jobclient *jobClient, err os.Error) {
|
||||||
conn, err := net.Dial(TCP, addr)
|
conn, err := net.Dial(TCP, addr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -21,6 +27,7 @@ func newJobClient(addr string, worker *Worker) (jobclient *jobClient, err os.Err
|
|||||||
return jobclient, err
|
return jobclient, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Main loop.
|
||||||
func (client *jobClient) Work() {
|
func (client *jobClient) Work() {
|
||||||
noop := true
|
noop := true
|
||||||
OUT: for client.running {
|
OUT: for client.running {
|
||||||
@ -64,10 +71,12 @@ func (client *jobClient) Work() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Send a job to the job server.
|
||||||
func (client *jobClient) WriteJob(job *WorkerJob) (err os.Error) {
|
func (client *jobClient) WriteJob(job *WorkerJob) (err os.Error) {
|
||||||
return client.Write(job.Encode())
|
return client.Write(job.Encode())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Write the encoded job.
|
||||||
func (client *jobClient) Write(buf []byte) (err os.Error) {
|
func (client *jobClient) Write(buf []byte) (err os.Error) {
|
||||||
var n int
|
var n int
|
||||||
for i := 0; i < len(buf); i += n {
|
for i := 0; i < len(buf); i += n {
|
||||||
@ -79,6 +88,7 @@ func (client *jobClient) Write(buf []byte) (err os.Error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Close.
|
||||||
func (client *jobClient) Close() (err os.Error) {
|
func (client *jobClient) Close() (err os.Error) {
|
||||||
client.running = false
|
client.running = false
|
||||||
err = client.conn.Close()
|
err = client.conn.Close()
|
||||||
|
Loading…
Reference in New Issue
Block a user