feat: add benchmark startup analysis tools and improve demo.sh

- Add generate-schedule.sh script to create container schedules from addresses.txt
- Add benchmark-startup Python script for analyzing container startup times
- Update demo.sh to print timestamps and wait for start signal at /oar-p2p/start
- Add comprehensive statistics including startup, start signal, and waiting times
- Support for synchronized container coordination via start signal file

1 files changed, 28 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
index 06cf322..e2d356e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -4,6 +4,7 @@ use std::{
     net::Ipv4Addr,
     path::{Path, PathBuf},
     process::Output,
+    time::Duration,
 };
 
 use clap::{Args, Parser, Subcommand};
@@ -108,6 +109,9 @@ struct RunArgs {
     #[clap(long)]
     output_dir: PathBuf,
 
+    #[clap(long, default_value = "10")]
+    signal_delay: u64,
+
     schedule: Option<PathBuf>,
 }
 
@@ -307,6 +311,17 @@ async fn cmd_run(args: RunArgs) -> Result<()> {
     )
     .await?;
 
+    tracing::info!("waiting {} seconds before signaling", args.signal_delay);
+    tokio::time::sleep(Duration::from_secs(args.signal_delay)).await;
+
+    machine::for_each(
+        machines
+            .iter()
+            .filter(|&machine| containers.iter().any(|c| c.machine == *machine)),
+        |machine| machine_signal_containers(&ctx, machine),
+    )
+    .await?;
+
     tracing::info!("waiting for all containers to exit");
     machine::for_each(&machines, |machine| {
         let ctx = ctx.clone();
@@ -350,10 +365,15 @@ async fn cmd_run(args: RunArgs) -> Result<()> {
 fn machine_containers_create_script(containers: &[ScheduledContainer]) -> String {
     let mut script = String::default();
     for (idx, container) in containers.iter().enumerate() {
+        // remove the start signal file if it exists
+        script.push_str("mkdir -p /tmp/oar-p2p-signal\n");
+        script.push_str("rm /tmp/oar-p2p-signal/start 2>/dev/null || true\n");
+
         script.push_str("docker create \\\n");
         script.push_str("\t--pull=always \\\n");
         script.push_str("\t--network=host \\\n");
         script.push_str("\t--restart=no \\\n");
+        script.push_str("\t--volume /tmp/oar-p2p-signal:/oar-p2p\\\n");
         script.push_str(&format!("\t--name {} \\\n", container.name));
         for (key, val) in container.variables.iter() {
             script.push_str("\t-e ");
@@ -406,6 +426,14 @@ async fn machine_start_containers(ctx: &Context, machine: Machine) -> Result<()>
     Ok(())
 }
 
+#[tracing::instrument(ret, err, skip(ctx))]
+async fn machine_signal_containers(ctx: &Context, machine: Machine) -> Result<()> {
+    tracing::info!("signaling containers");
+    machine_run_script(ctx, machine, "touch /tmp/oar-p2p-signal/start").await?;
+    tracing::info!("containers signaled");
+    Ok(())
+}
+
 fn machine_containers_wait_script(containers: &[ScheduledContainer]) -> String {
     let mut script = String::default();
     for container in containers {