print tps

Author: mmsqe

testground/benchmark/benchmark/main.py

@@ -2,12 +2,10 @@
 import subprocess
 from pathlib import Path
 
-import web3
-
 from .cli import ChainCommand
 from .context import Context
 from .peer import CONTAINER_CRONOSD_PATH, bootstrap
-from .sendtx import generate_load
+from .sendtx import collect_output, generate_load
 from .utils import wait_for_block, wait_for_port
 
 
@@ -43,11 +41,7 @@ def entrypoint(ctx: Context):
         )
 
     if ctx.is_fullnode_leader:
-        # collect output
-        w3 = web3.Web3(web3.providers.HTTPProvider("http://localhost:8545"))
-        for i in range(w3.eth.block_number):
-            blk = w3.eth.get_block(i)
-            print(i, len(blk.transactions), blk.timestamp)
+        collect_output()
 
     # halt after all tasks are done
     ctx.sync.signal_and_wait("halt", ctx.params.test_instance_count)

testground/benchmark/benchmark/sendtx.py

@@ -69,3 +69,11 @@ def generate_load(cli, num_accounts, num_txs, **kwargs):
                 fut.result()
             except Exception as e:
                 print("test task failed", e)
+
+
+def collect_output():
+    # collect output
+    w3 = web3.Web3(web3.providers.HTTPProvider("http://localhost:8545"))
+    for i in range(w3.eth.block_number):
+        blk = w3.eth.get_block(i)
+        print(i, len(blk.transactions), blk.timestamp)

testground/benchmark/benchmark/stateless.py

@@ -16,7 +16,7 @@
     init_node,
     patch_configs,
 )
-from .sendtx import generate_load
+from .sendtx import collect_output, generate_load
 from .topology import connect_all
 from .types import PeerPacket
 from .utils import wait_for_block, wait_for_port
@@ -95,7 +95,7 @@ def run(
             proc.wait()
         else:
             generate_load(cli, num_accounts, num_txs, home=home)
-
+            collect_output()
             proc.kill()
             try:
                 proc.wait()