Quellcode durchsuchen

调用Close后不再向JobServer获取任务

tags/0.2.1
余 欣怀 vor 1 Jahr
Ursprung
Commit
abd6ce9be3
2 geänderte Dateien mit 9 neuen und 4 gelöschten Zeilen
  1. +8
    -3
      worker/agent.go
  2. +1
    -1
      worker/worker.go

+ 8
- 3
worker/agent.go Datei anzeigen

@@ -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
}



+ 1
- 1
worker/worker.go Datei anzeigen

@@ -344,7 +344,7 @@ func (worker *Worker) exec(inpack *inPack) (err error) {
}
outpack.handle = inpack.handle
outpack.data = r.data
err = inpack.a.Write(outpack)
_ = inpack.a.Write(outpack)
if worker.leftJobs != nil {
worker.leftJobs <- struct{}{}
}


Laden…
Abbrechen
Speichern