Merge pull request #29 from kuzxnia/run_with_progress

fix progress exit
kuzxnia · Feb 21, 2024 · ac23513 · ac23513
2 parents 38a0df7 + e1b32f1
commit ac23513
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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:
-				}
 
 			}
 		}