logos-blockchain-testing/logos/examples/src/bin/compose_runner.rs

use std::{process, time::Duration};

use anyhow::{Context as _, Result};
use lb_ext::{
    CoreBuilderExt as _, LbcComposeDeployer as ComposeDeployer, LbcExtEnv, ScenarioBuilder,
    ScenarioBuilderExt as _, configs::network::Libp2pNetworkLayout,
};
use runner_examples::{DeployerKind, demo, read_env_any, read_topology_seed_or_default};
use testing_framework_core::scenario::{Deployer as _, Runner};
use testing_framework_runner_compose::ComposeRunnerError;
use tracing::{info, warn};

#[tokio::main]
async fn main() {
    runner_examples::defaults::init_node_log_dir_defaults(DeployerKind::Compose);

    tracing_subscriber::fmt::init();

    let nodes = read_env_any(&["LOGOS_BLOCKCHAIN_DEMO_NODES"], demo::DEFAULT_NODES);

    let run_secs = read_env_any(&["LOGOS_BLOCKCHAIN_DEMO_RUN_SECS"], demo::DEFAULT_RUN_SECS);

    info!(nodes, run_secs, "starting compose runner demo");

    if let Err(err) = run_compose_case(nodes, Duration::from_secs(run_secs)).await {
        warn!("compose runner demo failed: {err:#}");
        process::exit(1);
    }
}

async fn run_compose_case(nodes: usize, run_duration: Duration) -> Result<()> {
    info!(
        nodes,
        duration_secs = run_duration.as_secs(),
        "building scenario plan"
    );

    let seed = read_topology_seed_or_default();

    let scenario = ScenarioBuilder::deployment_with(|t| {
        t.with_network_layout(Libp2pNetworkLayout::Star)
            .with_node_count(nodes)
    })
    .enable_node_control()
    .with_run_duration(run_duration)
    .with_deployment_seed(seed)
    .initialize_wallet(
        demo::DEFAULT_TOTAL_WALLETS as u64 * 100,
        demo::DEFAULT_TOTAL_WALLETS,
    )
    .transactions_with(|txs| {
        txs.rate(demo::DEFAULT_MIXED_TXS_PER_BLOCK)
            .users(demo::DEFAULT_TRANSACTION_WALLETS)
    })
    .expect_consensus_liveness();

    let mut plan = scenario.build()?;

    let deployer = ComposeDeployer::new();
    info!("deploying compose stack");

    let runner: Runner<LbcExtEnv> = match deployer.deploy(&plan).await {
        Ok(runner) => runner,
        Err(ComposeRunnerError::DockerUnavailable) => {
            warn!("Docker is unavailable; cannot run compose demo");
            return Ok(());
        }
        Err(err) => return Err(anyhow::Error::new(err)).context("deploying compose stack failed"),
    };

    if !runner.context().telemetry().is_configured() {
        warn!(
            "metrics querying is disabled; set LOGOS_BLOCKCHAIN_METRICS_QUERY_URL to enable PromQL queries"
        );
    }

    info!("running scenario");
    runner
        .run(&mut plan)
        .await
        .context("running compose scenario failed")?;
    Ok(())
}