Problem: output artifact gets expired easily (#1671)

* Problem: output artifact gets expired easily record in log * cleanup * fix --------- Co-authored-by: HuangYi <[email protected]>
crypto-org-chain · Nov 4, 2024 · 45aaff0 · 45aaff0
1 parent 9a9018f
commit 45aaff0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/testground/benchmark/benchmark/stateless.py b/testground/benchmark/benchmark/stateless.py
@@ -28,7 +28,7 @@
 from .stats import dump_block_stats
 from .topology import connect_all
 from .types import PeerPacket
-from .utils import block_height, block_txs, wait_for_block, wait_for_port
+from .utils import Tee, block_height, block_txs, wait_for_block, wait_for_port
 
 # use cronosd on host machine
 LOCAL_CRONOSD_PATH = "cronosd"
@@ -303,7 +303,7 @@ def do_run(
     detect_idle_halted(cfg["num_idle"], 5)
 
     with (home / "block_stats.log").open("w") as logfile:
-        dump_block_stats(logfile)
+        dump_block_stats(Tee(logfile, sys.stdout))
 
     proc.kill()
     proc.wait(20)

diff --git a/testground/benchmark/benchmark/utils.py b/testground/benchmark/benchmark/utils.py
@@ -207,3 +207,15 @@ def split_batch(a: int, size: int):
     if m:
         parts.append((k * size, a))
     return parts
+
+
+class Tee:
+    def __init__(self, f1, f2):
+        self.f1 = f1
+        self.f2 = f2
+
+    def write(self, s) -> int:
+        s1 = self.f1.write(s)
+        s2 = self.f2.write(s)
+        assert s1 == s2
+        return s1