79 lines
2.3 KiB
Rust
79 lines
2.3 KiB
Rust
|
use std::fs;
|
||
|
|
|
||
|
|
use swactor_gossip::report::generate_html_report;
|
||
|
|
use swactor_gossip::sim::{run_simulation, SimConfig, Topology};
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
let scenarios = vec![
|
||
|
|
SimConfig {
|
||
|
|
name: "Ring (5 nodes)".into(),
|
||
|
|
topology: Topology::Ring,
|
||
|
|
num_nodes: 5,
|
||
|
|
initial_data: test_data(3),
|
||
|
|
num_rounds: 15,
|
||
|
|
ticks_per_round: 4,
|
||
|
|
heal_after_round: None,
|
||
|
|
num_threads: 1,
|
||
|
|
},
|
||
|
|
SimConfig {
|
||
|
|
name: "Star (7 nodes)".into(),
|
||
|
|
topology: Topology::Star,
|
||
|
|
num_nodes: 7,
|
||
|
|
initial_data: test_data(3),
|
||
|
|
num_rounds: 10,
|
||
|
|
ticks_per_round: 4,
|
||
|
|
heal_after_round: None,
|
||
|
|
num_threads: 1,
|
||
|
|
},
|
||
|
|
SimConfig {
|
||
|
|
name: "Full Mesh (5 nodes)".into(),
|
||
|
|
topology: Topology::FullMesh,
|
||
|
|
num_nodes: 5,
|
||
|
|
initial_data: test_data(3),
|
||
|
|
num_rounds: 8,
|
||
|
|
ticks_per_round: 4,
|
||
|
|
heal_after_round: None,
|
||
|
|
num_threads: 1,
|
||
|
|
},
|
||
|
|
SimConfig {
|
||
|
|
name: "Chain (8 nodes)".into(),
|
||
|
|
topology: Topology::Chain,
|
||
|
|
num_nodes: 8,
|
||
|
|
initial_data: test_data(3),
|
||
|
|
num_rounds: 20,
|
||
|
|
ticks_per_round: 4,
|
||
|
|
heal_after_round: None,
|
||
|
|
num_threads: 1,
|
||
|
|
},
|
||
|
|
SimConfig {
|
||
|
|
name: "Partition & Heal (6 nodes)".into(),
|
||
|
|
topology: Topology::Partitioned,
|
||
|
|
num_nodes: 6,
|
||
|
|
initial_data: test_data(3),
|
||
|
|
num_rounds: 20,
|
||
|
|
ticks_per_round: 4,
|
||
|
|
heal_after_round: Some(10),
|
||
|
|
num_threads: 1,
|
||
|
|
},
|
||
|
|
];
|
||
|
|
|
||
|
|
for config in scenarios {
|
||
|
|
let filename = format!(
|
||
|
|
"gossip_report_{}.html",
|
||
|
|
config.name.to_lowercase().replace(' ', "_").replace(['(', ')'], "")
|
||
|
|
);
|
||
|
|
println!("Running scenario: {} ...", config.name);
|
||
|
|
let trace = run_simulation(config);
|
||
|
|
let html = generate_html_report(&trace);
|
||
|
|
fs::write(&filename, &html).expect("failed to write report");
|
||
|
|
println!(" -> wrote {filename} ({} bytes)", html.len());
|
||
|
|
}
|
||
|
|
println!("Done.");
|
||
|
|
}
|
||
|
|
|
||
|
|
fn test_data(n: usize) -> Vec<(String, Vec<u8>)> {
|
||
|
|
(0..n)
|
||
|
|
.map(|i| (format!("key-{i}"), format!("value-{i}").into_bytes()))
|
||
|
|
.collect()
|
||
|
|
}
|