feat: config

Author: transmissions11

Cargo.lock

@@ -0,0 +1,49 @@
+[network]
+target_url = "http://127.0.0.1:8545"
+total_connections = 10_000 # Limited by # of ephemeral ports.
+
+batch_factor = 1
+
+error_sleep_ms = 100
+tx_queue_empty_sleep_ms = 25
+
+[transaction]
+chain_id = 1337
+num_accounts = 25_000 # Limited by the number in the genesis (see generate_genesis_alloc.rs)
+
+token_contract_address = "0x2000000000000000000000000000000000000001"
+
+gas_price = 100000000000 # 100 gwei
+gas_limit = 100_000
+
+mnemonic = "test test test test test test test test test test test junk"
+
+recipient_distribution_factor = 20 # 1/20 of accounts receive transfers.
+max_transfer_amount = 10
+
+[rate_limiting]
+initial_ratelimit = 100 # txs/s
+
+# Rate limit thresholds: [(threshold, rate_limit)]
+# threshold = number of transactions sent
+# rate_limit = new transactions per second limit
+ratelimit_thresholds = [
+    [1_562, 250],      # NUM_ACCOUNTS / 16
+    [3_125, 500],      # NUM_ACCOUNTS / 8
+    [25_000, 1_000],    # NUM_ACCOUNTS
+    [50_000, 2_500],    # NUM_ACCOUNTS * 2
+    [100_000, 5_000],   # NUM_ACCOUNTS * 4
+    [200_000, 7_500],   # NUM_ACCOUNTS * 8
+    [250_000, 12_500],  # NUM_ACCOUNTS * 10
+    [300_000, 15_000],  # NUM_ACCOUNTS * 12
+]
+
+[worker]
+thread_pinning = true
+
+tx_gen_worker_percentage = 0.1
+network_worker_percentage = 0.9
+
+[reporting]
+tx_queue_report_interval_secs = 3
+network_stats_report_interval_secs = 3

configs/default.toml

@@ -26,4 +26,6 @@ serde_json = "1.0"
 hex = "0.4"
 simple-tqdm = { version = "0.2.0", features = ["rayon"] }
 rand = "0.9.1"
-ratelimit = "0.10"
+ratelimit = "0.10"
+clap = { version = "4.5", features = ["derive", "env"] }
+toml = "0.8"

crescendo/Cargo.toml

@@ -0,0 +1,81 @@
+use std::path::PathBuf;
+use std::sync::OnceLock;
+
+use serde::{Deserialize, Serialize};
+
+/// Global configuration instance for the application.
+static CONFIG_INSTANCE: OnceLock<Config> = OnceLock::new();
+
+/// Initialize the global configuration instance.
+pub fn init(config: Config) {
+    CONFIG_INSTANCE.set(config).unwrap();
+}
+
+/// Gets the global configuration instance's value,
+/// blocking until initialized if necessary.
+pub fn get() -> &'static Config {
+    CONFIG_INSTANCE.wait()
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct Config {
+    pub tx_gen_worker: TxGenWorkerConfig,
+    pub network_worker: NetworkWorkerConfig,
+    pub rate_limiting: RateLimitingConfig,
+
+    pub workers: WorkersConfig,
+    pub reporters: ReportersConfig,
+}
+
+impl Config {
+    pub fn from_file(path: &PathBuf) -> Result<Self, Box<dyn std::error::Error>> {
+        let config_str = std::fs::read_to_string(path)?;
+        let config: Config = toml::from_str(&config_str)?;
+        Ok(config)
+    }
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct NetworkWorkerConfig {
+    pub target_url: String,
+    pub total_connections: u64,
+
+    pub batch_factor: usize,
+
+    pub error_sleep_ms: u64,
+    pub tx_queue_empty_sleep_ms: u64,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct TxGenWorkerConfig {
+    pub chain_id: u64,
+
+    pub mnemonic: String,
+    pub num_accounts: u32,
+
+    pub gas_price: u128,
+    pub gas_limit: u64,
+
+    pub token_contract_address: String,
+    pub recipient_distribution_factor: u32,
+    pub max_transfer_amount: u64,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct RateLimitingConfig {
+    pub initial_ratelimit: u64,
+    pub ratelimit_thresholds: Vec<(u32, u64)>,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct WorkersConfig {
+    pub thread_pinning: bool,
+    pub tx_gen_worker_percentage: f64,
+    pub network_worker_percentage: f64,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct ReportersConfig {
+    pub tx_queue_report_interval_secs: u64,
+    pub network_stats_report_interval_secs: u64,
+}