benchmark: change complete notifier first

Fix https://github.com/coreos/etcd/issues/4708.
This commit is contained in:
Gyu-Ho Lee 2016-03-07 14:54:11 -08:00
parent e29ed898d4
commit b8912c9fb1

View File

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