Pass timescale from CLI

Author: SupernaviX

sim-rs/src/config.rs

@@ -146,8 +146,11 @@ fn compute_latency(loc1: Location, loc2: Location, extra_ms: Option<u64>) -> Dur
     geo_latency + extra_latency
 }
 
-pub fn read_config(filename: &Path) -> Result<SimConfiguration> {
+pub fn read_config(filename: &Path, timescale: Option<u32>) -> Result<SimConfiguration> {
     let file = fs::read_to_string(filename)?;
-    let raw_config: RawConfig = toml::from_str(&file)?;
+    let mut raw_config: RawConfig = toml::from_str(&file)?;
+    if let Some(ts) = timescale {
+        raw_config.timescale = Some(ts);
+    }
     Ok(raw_config.into())
 }

sim-rs/src/main.rs

@@ -23,6 +23,8 @@ mod sim;
 struct Args {
     filename: PathBuf,
     output: Option<PathBuf>,
+    #[clap(short, long)]
+    timescale: Option<u32>,
 }
 
 #[tokio::main]
@@ -42,7 +44,7 @@ async fn main() -> Result<()> {
     })?;
 
     let args = Args::parse();
-    let config = read_config(&args.filename)?;
+    let config = read_config(&args.filename, args.timescale)?;
 
     let (events_sink, events_source) = mpsc::unbounded_channel();
     let monitor = EventMonitor::new(&config, events_source, args.output).run();

sim-rs/test_data/simple.toml

@@ -1,5 +1,4 @@
 seed = 0x3c3373756e646165
-timescale = 16
 nodes = [
     # producers
     { location = [-80,  40], stake = 1000000 },