dora-cli 1.0.0-rc.2

`dora` goal is to be a low latency, composable, and distributed data flow.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use crate::{
    command::{Executable, default_tracing},
    common::{CoordinatorOptions, resolve_dataflow_identifier_interactive, send_control_request},
};
use dora_message::{cli_to_coordinator::ControlRequest, coordinator_to_cli::ControlRequestReply};
use eyre::{Context, bail};
use std::path::PathBuf;

/// Add a node to a running dataflow.
#[derive(Debug, clap::Args)]
pub struct Add {
    #[clap(flatten)]
    coordinator: CoordinatorOptions,
    /// Dataflow name or UUID to add the node to
    #[clap(long, short = 'd', value_name = "NAME_OR_UUID")]
    dataflow: Option<String>,
    #[clap(long = "from-yaml")]
    from_yaml: PathBuf,
}

impl Executable for Add {
    fn execute(self) -> eyre::Result<()> {
        default_tracing()?;
        let session = self.coordinator.connect()?;
        let dataflow_id =
            resolve_dataflow_identifier_interactive(&session, self.dataflow.as_deref())?;

        let yaml_content = std::fs::read_to_string(&self.from_yaml)
            .wrap_err_with(|| format!("failed to read {}", self.from_yaml.display()))?;
        let node: dora_message::descriptor::Node =
            serde_yaml::from_str(&yaml_content).wrap_err("failed to parse node YAML")?;

        let reply = send_control_request(&session, &ControlRequest::AddNode { dataflow_id, node })?;

        match reply {
            ControlRequestReply::NodeAdded { node_id, .. } => {
                println!("Node `{node_id}` added to dataflow {dataflow_id}");
            }
            ControlRequestReply::Error(err) => bail!("failed to add node: {err}"),
            other => bail!("unexpected reply: {other:?}"),
        }
        Ok(())
    }
}