Function run_as_worker

cake-cli/src/main.rs:243–278

Worker setup: zero-config discovery or manual topology. Used by `run` (no model + cluster key) subcommand.

(mut args: Args)

/// Worker setup: zero-config discovery or manual topology.
/// Used by `run` (no model + cluster key) subcommand.
async fn run_as_worker(mut args: Args) -> Result<()> {
    // Zero-config: wait for master assignment + model data
    let listener_override = if let (Some(key), None) = (&args.cluster_key, &args.topology) {
        if args.name.is_none() {
            args.name = Some("worker".to_string());
        }
        let worker_name = args.name.as_deref().unwrap();
        let cache_dir = cache_base_dir();
        let (layers, model_path, listener) = cake::sharding::worker_setup(
            worker_name,
            key,
            &args.address,
            &cache_dir,
        )
        .await?;
        args.model = model_path.to_string_lossy().to_string();
        args.topology_override = Some(build_worker_topology(
            worker_name,
            &args.address,
            &layers,
        ));
        Some(listener)
    } else {
        None
    };

    let mut ctx = Context::from_args(args)?;
    if let Some(listener) = listener_override {
        *ctx.listener_override.lock().unwrap() = Some(listener);
    }
    let ret = run_worker(&mut ctx).await;
    if ret.is_err() {
        println!();
    }
    ret
}

/// Shared master setup: zero-config discovery + dispatch.
/// Used by `run` and `serve` subcommands.