dora/examples/benchmark/run.rs

use eyre::{bail, Context};
use std::path::Path;
use tracing::metadata::LevelFilter;
use tracing_subscriber::Layer;

#[tokio::main]
async fn main() -> eyre::Result<()> {
    set_up_tracing().wrap_err("failed to set up tracing subscriber")?;

    let root = Path::new(env!("CARGO_MANIFEST_DIR"));
    std::env::set_current_dir(root.join(file!()).parent().unwrap())
        .wrap_err("failed to set working dir")?;

    let dataflow_zenoh = Path::new("dataflow-zenoh.yml");
    build_dataflow(dataflow_zenoh).await?;

    let dataflow_iceoryx = Path::new("dataflow-iceoryx.yml");
    build_dataflow(dataflow_iceoryx).await?;

    println!("ZENOH:");
    dora_coordinator::run(dora_coordinator::Args {
        run_dataflow: dataflow_zenoh.to_owned().into(),
        runtime: None,
    })
    .await?;
    println!("\n\nICEORYX:");
    dora_coordinator::run(dora_coordinator::Args {
        run_dataflow: dataflow_iceoryx.to_owned().into(),
        runtime: None,
    })
    .await?;

    Ok(())
}

async fn build_dataflow(dataflow: &Path) -> eyre::Result<()> {
    let cargo = std::env::var("CARGO").unwrap();
    let mut cmd = tokio::process::Command::new(&cargo);
    cmd.arg("run");
    cmd.arg("--package").arg("dora-cli");
    cmd.arg("--").arg("build").arg(dataflow);
    if !cmd.status().await?.success() {
        bail!("failed to build dataflow");
    };
    Ok(())
}

fn set_up_tracing() -> eyre::Result<()> {
    use tracing_subscriber::prelude::__tracing_subscriber_SubscriberExt;

    let stdout_log = tracing_subscriber::fmt::layer()
        .pretty()
        .with_filter(LevelFilter::DEBUG);
    let subscriber = tracing_subscriber::Registry::default().with(stdout_log);
    tracing::subscriber::set_global_default(subscriber)
        .context("failed to set tracing global subscriber")
}