diff --git a/lbot/progress.go b/lbot/progress.go index 2d6c81a..b13d21e 100644 --- a/lbot/progress.go +++ b/lbot/progress.go @@ -39,6 +39,12 @@ func (p *ProgressProcess) Run(request *proto.ProgressRequest, srv proto.Progress return !worker.IsDone() }) for range ticker.C { + select { + case <-p.lbot.done: + fmt.Println("workload done") + done <- true + default: + } for _, w := range notDoneWorkers { isWorkerFinished := w.IsDone() resp := proto.ProgressResponse{ @@ -62,12 +68,6 @@ func (p *ProgressProcess) Run(request *proto.ProgressRequest, srv proto.Progress return !worker.IsDone() }) } - select { - case <-p.lbot.done: - fmt.Println("workload done") - done <- true - default: - } } }