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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

use clap::Args;

use crate::{
    command::{Executable, default_tracing},
    common::{CoordinatorOptions, resolve_dataflow_identifier_interactive, send_control_request},
    formatting::OutputFormat,
};
use dora_message::{
    cli_to_coordinator::ControlRequest, coordinator_to_cli::ControlRequestReply, id::NodeId,
};
use eyre::bail;

/// List all runtime parameters for a node.
///
/// Examples:
///
/// List parameters for a node:
///   dora param list camera_node
///
/// List for a specific dataflow:
///   dora param list camera_node -d my-dataflow
///
/// Output as JSON:
///   dora param list camera_node --format json
#[derive(Debug, Args)]
#[clap(verbatim_doc_comment)]
pub struct List {
    /// Node ID
    #[clap(value_name = "NODE")]
    pub node: String,

    /// Filter by dataflow name or UUID
    #[clap(long, short = 'd', value_name = "NAME_OR_UUID")]
    pub dataflow: Option<String>,

    /// Output format: table or json
    #[clap(long, default_value = "table")]
    pub format: OutputFormat,

    #[clap(flatten)]
    coordinator: CoordinatorOptions,
}

impl Executable for List {
    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 node_id: NodeId = self
            .node
            .parse::<NodeId>()
            .map_err(|e| eyre::eyre!("invalid node ID: {e}"))?;

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

        match reply {
            ControlRequestReply::ParamList { params } => {
                if self.format == OutputFormat::Json {
                    let map: serde_json::Map<String, serde_json::Value> =
                        params.into_iter().collect();
                    println!("{}", serde_json::to_string_pretty(&map)?);
                } else if params.is_empty() {
                    println!("No parameters set for node `{}`", self.node);
                } else {
                    println!("Parameters for node `{}`:", self.node);
                    for (key, value) in &params {
                        println!("  {key} = {value}");
                    }
                }
                Ok(())
            }
            ControlRequestReply::Error(err) => bail!("{err}"),
            other => bail!("unexpected reply: {other:?}"),
        }
    }
}