Merge pull request etcd-io#6610 from heyitsanthony/bench-lease

benchmark: submit keepalive requests concurrently with report.Run()
adamgomulka · Oct 10, 2016 · 3a6fe61 · 3a6fe61
2 parents fd60205 + c1d115b
commit 3a6fe61
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tools/benchmark/cmd/lease.go b/tools/benchmark/cmd/lease.go
@@ -68,10 +68,14 @@ func leaseKeepaliveFunc(cmd *cobra.Command, args []string) {
 		}(clients[i])
 	}
 
-	for i := 0; i < leaseKeepaliveTotal; i++ {
-		requests <- struct{}{}
-	}
-	close(requests)
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		for i := 0; i < leaseKeepaliveTotal; i++ {
+			requests <- struct{}{}
+		}
+		close(requests)
+	}()
 
 	rc := r.Run()
 	wg.Wait()