Format codes by gofmt.

This commit is contained in:
mikespook 2011-05-24 13:20:22 +08:00
parent 82f0ab2cd5
commit 7a4bbcdc55
13 changed files with 219 additions and 213 deletions

3
README
View File

@ -16,6 +16,9 @@ It was implemented a native protocol for both worker and client API.
$ make client $ make client
$ ./client $ ./client
- Code format
gofmt -spaces=true -tabwidth=4 -w=true -tabindent=false $(DIR)
---- ----
xingxing<mikespook@gmail.com> xingxing<mikespook@gmail.com>
http://mikespook.com http://mikespook.com

View File

@ -16,3 +16,6 @@ include $(GOROOT)/src/Make.pkg
%: %.go %: %.go
$(GC) $*.go $(GC) $*.go
$(LD) -o $@ $*.$O $(LD) -o $@ $*.$O
fmt:
gofmt -spaces=true -tabwidth=4 -w=true -tabindent=false ./

View File

@ -36,7 +36,7 @@ func main() {
log.Println("Unknown") log.Println("Unknown")
} }
if running { if running {
log.Printf("%g%%\n", float32(numerator) * 100 / float32(denominator)) log.Printf("%g%%\n", float32(numerator)*100/float32(denominator))
} else { } else {
log.Println("Not running") log.Println("Not running")
} }

View File

@ -20,3 +20,6 @@ include $(GOROOT)/src/Make.pkg
%: install %.go %: install %.go
$(GC) $*.go $(GC) $*.go
$(LD) -o $@ $*.$O $(LD) -o $@ $*.$O
fmt:
gofmt -spaces=true -tabwidth=4 -w=true -tabindent=false ./

View File

@ -8,7 +8,7 @@ import (
"os" "os"
"net" "net"
"sync" "sync"
// "log" // "log"
"strconv" "strconv"
) )
@ -30,10 +30,10 @@ type Client struct {
} }
// Create a new client. // 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),
UId:1} UId: 1}
return return
} }
@ -60,12 +60,12 @@ func (client *Client) read() (data []byte, err os.Error) {
buf := make([]byte, BUFFER_SIZE) buf := make([]byte, BUFFER_SIZE)
var n int var n int
if n, err = client.conn.Read(buf); err != nil { if n, err = client.conn.Read(buf); err != nil {
if (err == os.EOF && n == 0) { if err == os.EOF && n == 0 {
break break
} }
return return
} }
data = append(data, buf[0: n] ...) data = append(data, buf[0:n]...)
if n < BUFFER_SIZE { if n < BUFFER_SIZE {
break break
} }
@ -74,7 +74,7 @@ func (client *Client) read() (data []byte, err os.Error) {
// split package // split package
start, end := 0, 4 start, end := 0, 4
tl := len(data) tl := len(data)
for i := 0; i < tl; i ++{ for i := 0; i < tl; i++ {
if string(data[start:end]) == RES_STR { if string(data[start:end]) == RES_STR {
l := int(byteToUint32([4]byte{data[start+8], data[start+9], data[start+10], data[start+11]})) l := int(byteToUint32([4]byte{data[start+8], data[start+9], data[start+10], data[start+11]}))
total := l + 12 total := l + 12
@ -104,7 +104,7 @@ func (client *Client) ReadJob() (job *ClientJob, err os.Error) {
if job, err = DecodeClientJob(rel); err != nil { if job, err = DecodeClientJob(rel); err != nil {
return return
} else { } else {
switch(job.DataType) { switch job.DataType {
case ERROR: case ERROR:
_, err = getError(job.Data) _, err = getError(job.Data)
return return
@ -123,34 +123,34 @@ func (client *Client) ReadJob() (job *ClientJob, err os.Error) {
// JOB_LOW | JOB_BG means the job is running with low level in background. // 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 {
if flag & JOB_BG == JOB_BG { if flag&JOB_BG == JOB_BG {
datatype = SUBMIT_JOB_LOW_BG datatype = SUBMIT_JOB_LOW_BG
} else { } else {
datatype = SUBMIT_JOB_LOW datatype = SUBMIT_JOB_LOW
} }
} else if flag & JOB_HIGH == JOB_HIGH { } else if flag&JOB_HIGH == JOB_HIGH {
if flag & JOB_BG == JOB_BG { if flag&JOB_BG == JOB_BG {
datatype = SUBMIT_JOB_HIGH_BG datatype = SUBMIT_JOB_HIGH_BG
} else { } else {
datatype = SUBMIT_JOB_HIGH datatype = SUBMIT_JOB_HIGH
} }
} else if flag & JOB_BG == JOB_BG { } else if flag&JOB_BG == JOB_BG {
datatype = SUBMIT_JOB_BG datatype = SUBMIT_JOB_BG
} else { } else {
datatype = SUBMIT_JOB datatype = SUBMIT_JOB
} }
rel := make([]byte, 0, 1024 * 64) rel := make([]byte, 0, 1024*64)
rel = append(rel, []byte(funcname) ...) rel = append(rel, []byte(funcname)...)
rel = append(rel, '\x00') rel = append(rel, '\x00')
client.mutex.Lock() client.mutex.Lock()
uid := strconv.Itoa(int(client.UId)) uid := strconv.Itoa(int(client.UId))
client.UId ++ client.UId++
rel = append(rel, []byte(uid) ...) rel = append(rel, []byte(uid)...)
client.mutex.Unlock() client.mutex.Unlock()
rel = append(rel, '\x00') rel = append(rel, '\x00')
rel = append(rel, data ...) rel = append(rel, data...)
if err = client.WriteJob(NewClientJob(REQ, datatype, rel)); err != nil { if err = client.WriteJob(NewClientJob(REQ, datatype, rel)); err != nil {
return return
} }
@ -160,7 +160,7 @@ func (client *Client) Do(funcname string, data []byte, flag byte) (handle string
} }
handle = string(job.Data) handle = string(job.Data)
go func() { go func() {
if flag & JOB_BG != JOB_BG { if flag&JOB_BG != JOB_BG {
for { for {
if job, err = client.ReadJob(); err != nil { if job, err = client.ReadJob(); err != nil {
return return
@ -178,7 +178,7 @@ func (client *Client) Do(funcname string, data []byte, flag byte) (handle string
} }
// Internal read last job // 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 {
return return
@ -200,7 +200,7 @@ func (client *Client) Status(handle string) (known, running bool, numerator, den
if err = client.WriteJob(NewClientJob(REQ, GET_STATUS, []byte(handle))); err != nil { if err = client.WriteJob(NewClientJob(REQ, GET_STATUS, []byte(handle))); err != nil {
return return
} }
var job * ClientJob var job *ClientJob
if job, err = client.readLastJob(STATUS_RES); err != nil { if job, err = client.readLastJob(STATUS_RES); err != nil {
return return
} }
@ -245,7 +245,7 @@ func (client *Client) LastJob() (job *ClientJob) {
if l == 0 { if l == 0 {
return return
} }
for i := 0; i < l - 1; i ++ { for i := 0; i < l-1; i++ {
<-client.JobQueue <-client.JobQueue
} }
} }

View File

@ -6,7 +6,7 @@ package gearman
import ( import (
"os" "os"
// "log" // "log"
) )
// Client side job // Client side job
@ -18,13 +18,13 @@ type ClientJob struct {
// Create a new job // 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 // 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.")
return return
@ -44,18 +44,18 @@ func DecodeClientJob(data []byte) (job * ClientJob, err os.Error) {
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)
data = make([]byte, 0, 1024 * 64) data = make([]byte, 0, 1024*64)
data = append(data, magiccode[:] ...) data = append(data, magiccode[:]...)
data = append(data, datatype[:] ...) data = append(data, datatype[:]...)
l := len(job.Data) l := len(job.Data)
datalength := uint32ToByte(uint32(l)) datalength := uint32ToByte(uint32(l))
data = append(data, datalength[:] ...) data = append(data, datalength[:]...)
data = append(data, job.Data ...) data = append(data, job.Data...)
return return
} }
// Extract the job's result. // 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:
job.Handle = string(job.Data) job.Handle = string(job.Data)

View File

@ -2,7 +2,7 @@ package gearman
import ( import (
"testing" "testing"
// "os" // "os"
) )
var client *Client var client *Client
@ -11,14 +11,14 @@ func init() {
client = NewClient() client = NewClient()
} }
func TestClientAddServer(t * testing.T) { func TestClientAddServer(t *testing.T) {
t.Log("Add local server 127.0.0.1:4730") t.Log("Add local server 127.0.0.1:4730")
if err := client.AddServer("127.0.0.1:4730"); err != nil { if err := client.AddServer("127.0.0.1:4730"); err != nil {
t.Error(err) t.Error(err)
} }
} }
func TestClientEcho(t * testing.T) { func TestClientEcho(t *testing.T) {
if echo, err := client.Echo([]byte("Hello world")); err != nil { if echo, err := client.Echo([]byte("Hello world")); err != nil {
t.Error(err) t.Error(err)
} else { } else {
@ -26,17 +26,16 @@ func TestClientEcho(t * testing.T) {
} }
} }
func TestClientDo(t * testing.T) { func TestClientDo(t *testing.T) {
if handle, err := client.Do("ToUpper", []byte("abcdef"), JOB_LOW | JOB_BG); err != nil { if handle, err := client.Do("ToUpper", []byte("abcdef"), JOB_LOW|JOB_BG); err != nil {
t.Error(err) t.Error(err)
} else { } else {
t.Log(handle) t.Log(handle)
} }
} }
func TestClientClose(t * testing.T) { func TestClientClose(t *testing.T) {
if err := client.Close(); err != nil { if err := client.Close(); err != nil {
t.Error(err) t.Error(err)
} }
} }

View File

@ -85,7 +85,7 @@ type Job interface {
} }
// Splite the byte array by a 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
for i, v := range slice { for i, v := range slice {
@ -93,9 +93,9 @@ func splitByteArray(slice []byte, spot byte) (data [][]byte){
if start != end { if start != end {
data = append(data, slice[start:end]) data = append(data, slice[start:end])
} }
start, end = i + 1, i + 1 start, end = i+1, i+1
} else { } else {
end ++ end++
} }
} }
data = append(data, slice[start:]) data = append(data, slice[start:])
@ -110,16 +110,16 @@ func getError(data []byte) (eno os.Errno, err os.Error) {
return return
} }
l := len(rel[0]) l := len(rel[0])
eno = os.Errno(byteToUint32([4]byte{rel[0][l-4], rel[0][l-3], rel[0][l-2], rel[0][l - 1]})) eno = os.Errno(byteToUint32([4]byte{rel[0][l-4], rel[0][l-3], rel[0][l-2], rel[0][l-1]}))
err = os.NewError(string(rel[1])) err = os.NewError(string(rel[1]))
return return
} }
// Decode [4]byte to uint32 // 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 +
uint32(buf[2]) << 8 + uint32(buf[2])<<8 +
uint32(buf[3]) uint32(buf[3])
} }

View File

@ -4,10 +4,10 @@
package gearman package gearman
import( import (
"os" "os"
"sync" "sync"
// "log" // "log"
) )
// The definition of the callback function. // The definition of the callback function.
@ -48,7 +48,7 @@ type Worker struct {
func NewWorker() (worker *Worker) { func NewWorker() (worker *Worker) {
worker = &Worker{ worker = &Worker{
// job server list // job server list
clients:make([]*jobClient, 0, WORKER_SERVER_CAP), clients: make([]*jobClient, 0, WORKER_SERVER_CAP),
// function list // function list
functions: make(JobFunctionMap), functions: make(JobFunctionMap),
incoming: make(chan *WorkerJob, QUEUE_CAP), incoming: make(chan *WorkerJob, QUEUE_CAP),
@ -61,7 +61,7 @@ func NewWorker() (worker *Worker) {
// Add a server. The addr should be 'host:port' format. // Add a server. The addr should be 'host:port' format.
// The connection is established at this time. // 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()
@ -76,7 +76,7 @@ func (worker * Worker) AddServer(addr string) (err os.Error) {
} }
n := len(worker.clients) n := len(worker.clients)
worker.clients = worker.clients[0: n + 1] worker.clients = worker.clients[0 : n+1]
worker.clients[n] = server worker.clients[n] = server
return return
} }
@ -85,8 +85,8 @@ func (worker * Worker) AddServer(addr string) (err os.Error) {
// Add a function. // Add a function.
// Plz added job servers first, then functions. // Plz added job servers first, then functions.
// The API will tell every connected job server that 'I can do this' // 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 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.")
} }
@ -103,7 +103,7 @@ func (worker * Worker) AddFunction(funcname string,
datatype = CAN_DO_TIMEOUT datatype = CAN_DO_TIMEOUT
data = []byte(funcname + "\x00") data = []byte(funcname + "\x00")
t := uint32ToByte(timeout) t := uint32ToByte(timeout)
data = append(data, t[:] ...) data = append(data, t[:]...)
} }
job := NewWorkerJob(REQ, datatype, data) job := NewWorkerJob(REQ, datatype, data)
worker.WriteJob(job) worker.WriteJob(job)
@ -112,7 +112,7 @@ func (worker * Worker) AddFunction(funcname string,
// Remove a function. // Remove a function.
// Tell job servers 'I can not do this now' at the same time. // 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()
@ -126,7 +126,7 @@ func (worker * Worker) RemoveFunction(funcname string) (err os.Error) {
} }
// Main loop // 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()
} }
@ -159,12 +159,12 @@ func (worker * Worker) Work() {
// If there are more than one job in the queue, // If there are more than one job in the queue,
// the last one will be returned, // the last one will be returned,
// the others will be lost. // the others will be lost.
func (worker * Worker) LastJob() (job *WorkerJob) { 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
} }
for i := 0; i < l - 1; i ++ { for i := 0; i < l-1; i++ {
<-worker.JobQueue <-worker.JobQueue
} }
} }
@ -172,7 +172,7 @@ func (worker * Worker) LastJob() (job *WorkerJob) {
} }
// Close. // Close.
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 {
err = v.Close() err = v.Close()
@ -184,25 +184,25 @@ func (worker * Worker) Close() (err os.Error){
// Write a job to job server. // Write a job to job server.
// Here, the job's mean is not the oraginal mean. // 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. // 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 {
go func() { go func() {
e <- v.WriteJob(job) e <- v.WriteJob(job)
}() }()
} }
return <- e return <-e
} }
// Send a something out, get the samething back. // 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)
} }
// Remove all of functions. // Remove all of functions.
// Both from the worker or job servers. // 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)
worker.functions = make(JobFunctionMap) worker.functions = make(JobFunctionMap)
@ -210,13 +210,13 @@ func (worker * Worker) Reset() (err os.Error){
} }
// Set the worker's unique id. // 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)
} }
// Execute the job. And send back the result. // 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])
funcname := string(jobdata[1]) funcname := string(jobdata[1])
@ -234,7 +234,7 @@ func (worker * Worker) exec(job *WorkerJob) (err os.Error) {
var datatype uint32 var datatype uint32
if err == nil { if err == nil {
datatype = WORK_COMPLETE datatype = WORK_COMPLETE
} else{ } else {
if result == nil { if result == nil {
datatype = WORK_FAIL datatype = WORK_FAIL
} else { } else {

View File

@ -7,7 +7,7 @@ package gearman
import ( import (
"os" "os"
"strconv" "strconv"
// "log" // "log"
) )
// Worker side job // Worker side job
@ -21,9 +21,9 @@ type WorkerJob struct {
// Create a new 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 // Decode job from byte slice
@ -47,17 +47,17 @@ func DecodeWorkerJob(data []byte) (job *WorkerJob, err os.Error) {
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)
data = make([]byte, 0, 1024 * 64) data = make([]byte, 0, 1024*64)
data = append(data, magiccode[:] ...) data = append(data, magiccode[:]...)
data = append(data, datatype[:] ...) data = append(data, datatype[:]...)
data = append(data, []byte{0, 0, 0, 0} ...) data = append(data, []byte{0, 0, 0, 0}...)
l := len(job.Data) l := len(job.Data)
if job.Handle != "" { if job.Handle != "" {
data = append(data, []byte(job.Handle) ...) data = append(data, []byte(job.Handle)...)
data = append(data, 0) data = append(data, 0)
l += len(job.Handle) + 1 l += len(job.Handle) + 1
} }
data = append(data, job.Data ...) data = append(data, job.Data...)
datalength := uint32ToByte(uint32(l)) datalength := uint32ToByte(uint32(l))
copy(data[8:12], datalength[:]) copy(data[8:12], datalength[:])
return return
@ -65,9 +65,9 @@ func (job *WorkerJob) Encode() (data []byte) {
// Send some datas to client. // Send some datas to client.
// Using this in a job's executing. // 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...)
var datatype uint32 var datatype uint32
if iswaring { if iswaring {
datatype = WORK_WARNING datatype = WORK_WARNING
@ -79,13 +79,11 @@ func (job * WorkerJob) UpdateData(data []byte, iswaring bool) (err os.Error) {
// Update status. // Update status.
// Tall client how many percent job has been executed. // Tall client how many percent job has been executed.
func (job * WorkerJob) UpdateStatus(numerator, denominator int) (err os.Error) { func (job *WorkerJob) UpdateStatus(numerator, denominator int) (err os.Error) {
n := []byte(strconv.Itoa(numerator)) n := []byte(strconv.Itoa(numerator))
d := []byte(strconv.Itoa(denominator)) 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))
} }

View File

@ -7,7 +7,7 @@ package gearman
import ( import (
"net" "net"
"os" "os"
// "log" // "log"
) )
// The client of job server. // The client of job server.
@ -24,12 +24,12 @@ func newJobClient(addr string, worker *Worker) (jobclient *jobClient, err os.Err
if err != nil { if err != nil {
return nil, err return nil, err
} }
jobclient = &jobClient{conn:conn, worker:worker, running:true, incoming: make(chan []byte, QUEUE_CAP)} jobclient = &jobClient{conn: conn, worker: worker, running: true, incoming: make(chan []byte, QUEUE_CAP)}
return jobclient, err return jobclient, err
} }
// Internal read // Internal read
func (client *jobClient) read() (data []byte, err os.Error){ func (client *jobClient) read() (data []byte, err os.Error) {
if len(client.incoming) > 0 { if len(client.incoming) > 0 {
// incoming queue is not empty // incoming queue is not empty
data = <-client.incoming data = <-client.incoming
@ -44,7 +44,7 @@ func (client *jobClient) read() (data []byte, err os.Error){
} }
return return
} }
data = append(data, buf[0: n] ...) data = append(data, buf[0:n]...)
if n < BUFFER_SIZE { if n < BUFFER_SIZE {
break break
} }
@ -53,7 +53,7 @@ func (client *jobClient) read() (data []byte, err os.Error){
// split package // split package
start := 0 start := 0
tl := len(data) tl := len(data)
for i := 0; i < tl; i ++{ for i := 0; i < tl; i++ {
if string(data[start:start+4]) == RES_STR { if string(data[start:start+4]) == RES_STR {
l := int(byteToUint32([4]byte{data[start+8], data[start+9], data[start+10], data[start+11]})) l := int(byteToUint32([4]byte{data[start+8], data[start+9], data[start+10], data[start+11]}))
total := l + 12 total := l + 12
@ -90,7 +90,7 @@ func (client *jobClient) Work() {
client.worker.ErrQueue <- err client.worker.ErrQueue <- err
continue continue
} else { } else {
switch(job.DataType) { switch job.DataType {
case NOOP: case NOOP:
noop = true noop = true
case NO_JOB: case NO_JOB:

View File

@ -41,7 +41,7 @@ func TestWorkerAddFunction(t *testing.T) {
} }
} }
func TestWorkerEcho(t * testing.T) { func TestWorkerEcho(t *testing.T) {
if err := worker.Echo([]byte("Hello World")); err != nil { if err := worker.Echo([]byte("Hello World")); err != nil {
t.Error(err) t.Error(err)
} }
@ -56,13 +56,13 @@ func TestWorkerResult(t *testing.T) {
} }
*/ */
func TestWorkerRemoveFunction(t * testing.T) { func TestWorkerRemoveFunction(t *testing.T) {
if err := worker.RemoveFunction("foobar"); err != nil { if err := worker.RemoveFunction("foobar"); err != nil {
t.Error(err) t.Error(err)
} }
} }
func TestWorkerReset(t * testing.T) { func TestWorkerReset(t *testing.T) {
if err := worker.Reset(); err != nil { if err := worker.Reset(); err != nil {
t.Error(err) t.Error(err)
} }