Astro Run Server

astro-run-server is an extension of astro-run that allows it to be served as a service and accept runners via gRPC requests.

astro-run-server forwards run requests, plugin events, logs, etc. from astro-run to runners using gRPC streams, enabling remote invocation and multi-runner scheduling.

Example

Add astro-run and astro-run-server as dependencies in your Cargo.toml:

[dependencies]

astro-run = "0.1"

astro-run-server = "0.1"

Astro Run Server

use astro_run::{AstroRun, Result, Workflow};
use astro_run_server::AstroRunServer;

#[tokio::main]
async fn main() -> Result<()> {
    let server = AstroRunServer::new();

    let cloned_server = server.clone();

    // Start the server in the background
    let handle = tokio::spawn(async move {
        // Server listening address
        cloned_server.serve("127.0.0.1:5001").await.unwrap();
    });

    let astro_run = AstroRun::builder().runner(server).build();

    let workflow = r#"
    jobs:
      test:
        name: Test Job
        steps:
          - timeout: 60m
            continue-on-error: false
            run: Hello World
    "#;

    let workflow = Workflow::builder()
        .config(workflow)
        .build(&astro_run)
        .unwrap();

    let ctx = astro_run.execution_context().build();

    // Run the workflow
    let _res = workflow.run(ctx).await;

    // Wait for the server to stop
    handle.await.unwrap();

    Ok(())
}

Runner

use astro_run::{stream, Context, Result, RunResult};
use astro_run_server::AstroRunRunner;

// Simulated implementation of a Runner
struct Runner {}

impl Runner {
    fn new() -> Self {
        Runner {}
    }
}

impl astro_run::Runner for Runner {
    fn run(&self, ctx: Context) -> astro_run::RunResponse {
        let (tx, rx) = stream();

        tx.log(ctx.command.run);
        tx.end(RunResult::Succeeded);

        Ok(rx)
    }
}

#[tokio::main]
async fn main() -> Result<()> {
    let runner = Runner::new();

    let mut astro_run_runner = AstroRunRunner::builder()
        .runner(runner)
        // Remote server address
        .url("http://127.0.0.1:5001")
        // Runner ID
        .id("test-runner")
        .build()
        .await
        .unwrap();

    astro_run_runner.start().await.unwrap();

    Ok(())
}

In the above example, you can replace the runner with a specific implementation from astro-runner.