|
|
@@ -143,10 +143,14 @@ func (a *agent) Grab() { |
|
|
|
a.grab() |
|
|
|
} |
|
|
|
|
|
|
|
func (a *agent) grab() { |
|
|
|
func (a *agent) grab() bool { |
|
|
|
if a.worker.closed != nil { |
|
|
|
return false |
|
|
|
} |
|
|
|
outpack := getOutPack() |
|
|
|
outpack.dataType = dtGrabJobUniq |
|
|
|
a.write(outpack) |
|
|
|
return true |
|
|
|
} |
|
|
|
|
|
|
|
func (a *agent) PreSleep() { |
|
|
@@ -169,9 +173,10 @@ func (a *agent) reconnect() error { |
|
|
|
bufio.NewWriter(a.conn)) |
|
|
|
|
|
|
|
a.worker.reRegisterFuncsForAgent(a) |
|
|
|
a.grab() |
|
|
|
if a.grab() { |
|
|
|
go a.work() |
|
|
|
} |
|
|
|
|
|
|
|
go a.work() |
|
|
|
return nil |
|
|
|
} |
|
|
|
|
|
|
|