use crate::{
cli::clap::{flag_arg, parse_matches, path_option, string_option, value_arg},
cli::defaults::{default_icp, local_network},
cli::globals::{internal_icp_arg, internal_network_arg},
metrics::{MetricsCommandError, model::MetricsKind},
};
use clap::Command as ClapCommand;
use std::{ffi::OsString, path::PathBuf};
const DEFAULT_LIMIT: u64 = 1_000;
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct MetricsOptions {
pub fleet: String,
pub kind: MetricsKind,
pub role: Option<String>,
pub canister: Option<String>,
pub nonzero: bool,
pub limit: u64,
pub json: bool,
pub out: Option<PathBuf>,
pub network: String,
pub icp: String,
}
impl MetricsOptions {
pub fn parse<I>(args: I) -> Result<Self, MetricsCommandError>
where
I: IntoIterator<Item = OsString>,
{
let matches = parse_matches(metrics_command(), args)
.map_err(|_| MetricsCommandError::Usage(usage()))?;
let kind = string_option(&matches, "kind")
.map(|value| MetricsKind::parse(&value))
.transpose()?
.unwrap_or(MetricsKind::Core);
let limit = string_option(&matches, "limit")
.and_then(|value| value.parse::<u64>().ok())
.filter(|limit| *limit > 0)
.unwrap_or(DEFAULT_LIMIT);
Ok(Self {
fleet: string_option(&matches, "fleet").expect("clap requires fleet"),
kind,
role: string_option(&matches, "role"),
canister: string_option(&matches, "canister"),
nonzero: matches.get_flag("nonzero"),
limit,
json: matches.get_flag("json"),
out: path_option(&matches, "out"),
network: string_option(&matches, "network").unwrap_or_else(local_network),
icp: string_option(&matches, "icp").unwrap_or_else(default_icp),
})
}
}
pub(super) fn usage() -> String {
let mut command = metrics_command();
command.render_help().to_string()
}
fn metrics_command() -> ClapCommand {
ClapCommand::new("metrics")
.bin_name("canic metrics")
.about("Query Canic runtime telemetry")
.disable_help_flag(true)
.arg(
value_arg("fleet")
.value_name("fleet")
.required(true)
.help("Installed fleet name to inspect"),
)
.arg(
value_arg("kind")
.long("kind")
.value_name("kind")
.help("Metrics tier to query; defaults to core"),
)
.arg(
value_arg("role")
.long("role")
.value_name("role")
.help("Only query one registry role"),
)
.arg(
value_arg("canister")
.long("canister")
.value_name("id")
.help("Only query one canister principal"),
)
.arg(
value_arg("limit")
.long("limit")
.value_name("entries")
.help("Maximum metric rows to fetch per canister; defaults to 1000"),
)
.arg(flag_arg("nonzero").long("nonzero"))
.arg(flag_arg("json").long("json"))
.arg(value_arg("out").long("out").value_name("file"))
.arg(internal_network_arg())
.arg(internal_icp_arg())
}