From 9529f761cfb010a91ba7bd4df1394e7d8a4fe77c Mon Sep 17 00:00:00 2001
From: keanji-x <jianxie0@gmail.com>
Date: Wed, 28 Jan 2026 16:16:51 +0800
Subject: [PATCH] add faucet mode

---
 src/main.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index c3ee9e0..92edc24 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -41,6 +41,9 @@ struct Args {
 
     #[arg(long, default_value = "bench_config.toml")]
     config: String,
+
+    #[arg(long, default_value_t = false)]
+    faucet_only: bool,
 }
 
 // mod uniswap;
@@ -479,6 +482,11 @@ async fn start_bench() -> Result<()> {
         .await?;
     }
 
+    if args.faucet_only {
+        info!("Faucet distribution complete. Faucet-only mode enabled, exiting.");
+        return Ok(());
+    }
+
     let tps = benchmark_config.target_tps as usize;
     let duration_secs = benchmark_config.performance.duration_secs;
     if benchmark_config.enable_swap_token {