forked from yuxh/gearman-go
go fmt
This commit is contained in:
parent
a003eac543
commit
ad9b3cb988
@ -3,8 +3,8 @@ package main
|
||||
import (
|
||||
"github.com/mikespook/gearman-go/client"
|
||||
"log"
|
||||
"sync"
|
||||
"os"
|
||||
"sync"
|
||||
)
|
||||
|
||||
func main() {
|
||||
|
@ -4,10 +4,10 @@ import (
|
||||
"github.com/mikespook/gearman-go/worker"
|
||||
"github.com/mikespook/golib/signal"
|
||||
"log"
|
||||
"net"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
"net"
|
||||
)
|
||||
|
||||
func ToUpper(job worker.Job) ([]byte, error) {
|
||||
@ -41,7 +41,7 @@ func main() {
|
||||
w.ErrorHandler = func(e error) {
|
||||
log.Println(e)
|
||||
if opErr, ok := e.(*net.OpError); ok {
|
||||
if ! opErr.Temporary() {
|
||||
if !opErr.Temporary() {
|
||||
proc, err := os.FindProcess(os.Getpid())
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
|
@ -4,9 +4,9 @@ import (
|
||||
"bufio"
|
||||
"bytes"
|
||||
"encoding/binary"
|
||||
"io"
|
||||
"net"
|
||||
"sync"
|
||||
"io"
|
||||
)
|
||||
|
||||
// The agent of job server.
|
||||
@ -59,14 +59,14 @@ func (a *agent) work() {
|
||||
if opErr, ok := err.(*net.OpError); ok {
|
||||
if opErr.Temporary() {
|
||||
continue
|
||||
}else{
|
||||
} else {
|
||||
a.disconnect_error(err)
|
||||
// else - we're probably dc'ing due to a Close()
|
||||
|
||||
break
|
||||
}
|
||||
|
||||
} else if( err == io.EOF ){
|
||||
|
||||
} else if err == io.EOF {
|
||||
a.disconnect_error(err)
|
||||
break
|
||||
}
|
||||
@ -104,11 +104,11 @@ func (a *agent) work() {
|
||||
}
|
||||
}
|
||||
|
||||
func (a * agent) disconnect_error( err error ){
|
||||
if( a.conn != nil ){
|
||||
func (a *agent) disconnect_error(err error) {
|
||||
if a.conn != nil {
|
||||
err = &WorkerDisconnectError{
|
||||
err : err,
|
||||
agent : a,
|
||||
err: err,
|
||||
agent: a,
|
||||
}
|
||||
a.worker.err(err)
|
||||
}
|
||||
@ -129,7 +129,7 @@ func (a *agent) Grab() {
|
||||
a.grab()
|
||||
}
|
||||
|
||||
func (a *agent) grab(){
|
||||
func (a *agent) grab() {
|
||||
outpack := getOutPack()
|
||||
outpack.dataType = dtGrabJobUniq
|
||||
a.write(outpack)
|
||||
@ -143,16 +143,16 @@ func (a *agent) PreSleep() {
|
||||
a.write(outpack)
|
||||
}
|
||||
|
||||
func (a *agent) reconnect() (error){
|
||||
a.Lock()
|
||||
func (a *agent) reconnect() error {
|
||||
a.Lock()
|
||||
defer a.Unlock()
|
||||
conn, err := net.Dial(a.net, a.addr)
|
||||
if err != nil {
|
||||
return err;
|
||||
return err
|
||||
}
|
||||
a.conn = conn
|
||||
a.rw = bufio.NewReadWriter(bufio.NewReader(a.conn),
|
||||
bufio.NewWriter(a.conn))
|
||||
bufio.NewWriter(a.conn))
|
||||
a.grab()
|
||||
a.worker.reRegisterFuncsForAgent(a)
|
||||
|
||||
|
@ -50,8 +50,8 @@ func run_gearman() {
|
||||
panic(`Unable to start gearman aborting test`)
|
||||
}
|
||||
gearman_ready <- true
|
||||
|
||||
<- kill_gearman
|
||||
|
||||
<-kill_gearman
|
||||
}
|
||||
|
||||
func check_gearman_present() bool {
|
||||
@ -134,14 +134,14 @@ func TestBasicDisconnect(t *testing.T) {
|
||||
send_client_request()
|
||||
|
||||
select {
|
||||
case <- done:
|
||||
case <-done:
|
||||
t.Error("Client request handled (somehow), did we magically reconnect?")
|
||||
case <-timeout:
|
||||
t.Error("Test timed out waiting for the error handler")
|
||||
case <-c_error:
|
||||
// error was handled!
|
||||
if ! was_dc_err {
|
||||
t.Error( "Disconnect didn't manifest as a net.OpError?")
|
||||
if !was_dc_err {
|
||||
t.Error("Disconnect didn't manifest as a net.OpError?")
|
||||
}
|
||||
}
|
||||
worker.Close()
|
||||
@ -175,24 +175,23 @@ func TestDcRc(t *testing.T) {
|
||||
|
||||
worker.ErrorHandler = func(e error) {
|
||||
wdc, wdcok := e.(*WorkerDisconnectError)
|
||||
|
||||
if( wdcok){
|
||||
|
||||
if wdcok {
|
||||
log.Println("Reconnecting!")
|
||||
reconnected := false
|
||||
for tries := 20 ; ! reconnected && tries > 0; tries -- {
|
||||
for tries := 20; !reconnected && tries > 0; tries-- {
|
||||
rcerr := wdc.Reconnect()
|
||||
if rcerr != nil{
|
||||
if rcerr != nil {
|
||||
time.Sleep(250 * time.Millisecond)
|
||||
} else{
|
||||
reconnected = true;
|
||||
} else {
|
||||
reconnected = true
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
} else{
|
||||
|
||||
} else {
|
||||
panic("Some other kind of error " + e.Error())
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
go func() {
|
||||
@ -219,9 +218,9 @@ func TestDcRc(t *testing.T) {
|
||||
}
|
||||
|
||||
send_client_request()
|
||||
|
||||
|
||||
select {
|
||||
case <- done:
|
||||
case <-done:
|
||||
case <-timeout:
|
||||
t.Error("Test timed out")
|
||||
}
|
||||
|
@ -139,7 +139,7 @@ func TestWorkerClose(t *testing.T) {
|
||||
worker.Close()
|
||||
}
|
||||
|
||||
func TestWorkWithoutReady(t * testing.T){
|
||||
func TestWorkWithoutReady(t *testing.T) {
|
||||
other_worker := New(Unlimited)
|
||||
|
||||
if err := other_worker.AddServer(Network, "127.0.0.1:4730"); err != nil {
|
||||
@ -148,15 +148,15 @@ func TestWorkWithoutReady(t * testing.T){
|
||||
if err := other_worker.AddFunc("gearman-go-workertest", foobar, 0); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
timeout := make(chan bool, 1)
|
||||
done := make( chan bool, 1)
|
||||
|
||||
other_worker.JobHandler = func( j Job ) error {
|
||||
if( ! other_worker.ready ){
|
||||
t.Error("Worker not ready as expected");
|
||||
timeout := make(chan bool, 1)
|
||||
done := make(chan bool, 1)
|
||||
|
||||
other_worker.JobHandler = func(j Job) error {
|
||||
if !other_worker.ready {
|
||||
t.Error("Worker not ready as expected")
|
||||
}
|
||||
done <-true
|
||||
done <- true
|
||||
return nil
|
||||
}
|
||||
go func() {
|
||||
@ -164,15 +164,15 @@ func TestWorkWithoutReady(t * testing.T){
|
||||
timeout <- true
|
||||
}()
|
||||
|
||||
go func(){
|
||||
other_worker.Work();
|
||||
go func() {
|
||||
other_worker.Work()
|
||||
}()
|
||||
|
||||
// With the all-in-one Work() we don't know if the
|
||||
// With the all-in-one Work() we don't know if the
|
||||
// worker is ready at this stage so we may have to wait a sec:
|
||||
go func(){
|
||||
go func() {
|
||||
tries := 5
|
||||
for( tries > 0 ){
|
||||
for tries > 0 {
|
||||
if other_worker.ready {
|
||||
other_worker.Echo([]byte("Hello"))
|
||||
break
|
||||
@ -183,24 +183,24 @@ func TestWorkWithoutReady(t * testing.T){
|
||||
tries--
|
||||
}
|
||||
}()
|
||||
|
||||
|
||||
// determine if we've finished or timed out:
|
||||
select{
|
||||
case <- timeout:
|
||||
select {
|
||||
case <-timeout:
|
||||
t.Error("Test timed out waiting for the worker")
|
||||
case <- done:
|
||||
case <-done:
|
||||
}
|
||||
}
|
||||
|
||||
func TestWorkWithoutReadyWithPanic(t * testing.T){
|
||||
func TestWorkWithoutReadyWithPanic(t *testing.T) {
|
||||
other_worker := New(Unlimited)
|
||||
|
||||
|
||||
timeout := make(chan bool, 1)
|
||||
done := make( chan bool, 1)
|
||||
done := make(chan bool, 1)
|
||||
|
||||
// Going to work with no worker setup.
|
||||
// when Work (hopefully) calls Ready it will get an error which should cause it to panic()
|
||||
go func(){
|
||||
go func() {
|
||||
defer func() {
|
||||
if err := recover(); err != nil {
|
||||
done <- true
|
||||
@ -209,17 +209,17 @@ func TestWorkWithoutReadyWithPanic(t * testing.T){
|
||||
t.Error("Work should raise a panic.")
|
||||
done <- true
|
||||
}()
|
||||
other_worker.Work();
|
||||
other_worker.Work()
|
||||
}()
|
||||
go func() {
|
||||
time.Sleep(2 * time.Second)
|
||||
timeout <- true
|
||||
}()
|
||||
|
||||
select{
|
||||
case <- timeout:
|
||||
select {
|
||||
case <-timeout:
|
||||
t.Error("Test timed out waiting for the worker")
|
||||
case <- done:
|
||||
case <-done:
|
||||
}
|
||||
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user