Bläddra i källkod

benchmark: change complete notifier first

Fix https://github.com/coreos/etcd/issues/4708.
Gyu-Ho Lee 9 år sedan
förälder
incheckning
b8912c9fb1
1 ändrade filer med 5 tillägg och 3 borttagningar
  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{}{}
-		}
 	}
 }