Bladeren bron

Merge pull request #4704 from gyuho/print_rate

benchmark: change complete notifier first
Gyu-Ho Lee 9 jaren geleden
bovenliggende
commit
969e42c3fa
1 gewijzigde bestanden met toevoegingen van 5 en 3 verwijderingen
  1. 5 3
      tools/benchmark/cmd/watch.go

+ 5 - 3
tools/benchmark/cmd/watch.go

@@ -198,14 +198,16 @@ func doWatch(stream v3.Watcher, requests <-chan string) {
 
 func recvWatchChan(wch v3.WatchChan) {
 	for range wch {
+		if atomic.LoadInt32(&nrRecvCompleted) == int32(eventsTotal) {
+			recvCompletedNotifier <- struct{}{}
+			break
+		}
+
 		st := time.Now()
 		results <- result{duration: time.Since(st)}
 		bar.Increment()
 
 		atomic.AddInt32(&nrRecvCompleted, 1)
-		if atomic.LoadInt32(&nrRecvCompleted) == int32(eventsTotal) {
-			recvCompletedNotifier <- struct{}{}
-		}
 	}
 }