Function run

components/vmworker/src/lib.rs:17–72

(config: WorkerConfig)

mod metrics_forwarder;

pub async fn run(config: WorkerConfig) -> Result<(), Box<dyn std::error::Error>> {
    common::setup_tracing(&config.common, "vmworker");

    let discord_config = common::fetch_discord_config(config.common.discord_token.clone())
        .await
        .expect("failed fetching discord config");

    info!("worker starting");

    #[cfg(target_family = "unix")]
    let (scheduler_tx, scheduler_rx) =
        connect_scheduler("/tmp/botloader_scheduler_workers", config.worker_id).await;

    #[cfg(target_family = "windows")]
    let (scheduler_tx, scheduler_rx) = connect_scheduler("localhost:7885", config.worker_id).await;

    metrics::set_global_recorder(metrics_forwarder::MetricsForwarder {
        tx: scheduler_tx.clone(),
    })
    .expect("set metrics recorder");

    let postgres_store = Db::new_with_url(&config.common.database_url).await.unwrap();

    // suppress signals for now
    // TODO: remove this? do we need signals here?
    // ideally we wanna manage this through the parent
    tokio::spawn(common::shutdown::wait_shutdown_signal());

    let logger = {
        let builder =
            guild_logger::GuildLoggerBuilder::new().add_backend(Arc::new(GuildLogForwarder {
                tx: scheduler_tx.clone(),
            }));

        builder.run()
    };

    let broker_client = dbrokerapi::state_client::Client::new(config.broker_api_addr);

    vm::init_v8_platform();

    let worker = Worker::new(
        scheduler_rx,
        scheduler_tx,
        postgres_store,
        logger,
        discord_config,
        config.common.user_script_http_proxy.clone(),
        broker_client,
    );

    worker.run().await;
    info!("worker shutting down");

    Ok(())
}

#[derive(Clone, clap::Parser)]