From 2e89ec910586f91700a4aebe00c29982b21da227 Mon Sep 17 00:00:00 2001
From: sakridge <stephen@solana.com>
Date: Thu, 9 Jan 2020 17:48:18 -0800
Subject: [PATCH] Don't keep generating transactions in non-sustained bench-tps
 mode (#7577)

---
 bench-tps/src/bench.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/bench-tps/src/bench.rs b/bench-tps/src/bench.rs
index 4e44397ab..268d266a1 100644
--- a/bench-tps/src/bench.rs
+++ b/bench-tps/src/bench.rs
@@ -197,7 +197,9 @@ where
                 sleep(Duration::from_millis(1));
             }
         } else {
-            while shared_tx_active_thread_count.load(Ordering::Relaxed) > 0 {
+            while !shared_txs.read().unwrap().is_empty()
+                || shared_tx_active_thread_count.load(Ordering::Relaxed) > 0
+            {
                 sleep(Duration::from_millis(1));
             }
         }