[pdm-devel] [PATCH proxmox-datacenter-manager 23/25] api: metric-collection: add status endpoint

[Lukas Wagner]

This adds a new API endpoint at /metric-collection/status, returning the
status of the last metric collection attempt from each remote.
For now, this contains the timestamp (last-collected) and any
error that occurred (error).

Signed-off-by: Lukas Wagner <l.wagner at proxmox.com>
---
 lib/pdm-api-types/src/metric_collection.rs | 15 +++++++++++++++
 server/src/api/metric_collection.rs        | 16 +++++++++++++++-
 server/src/metric_collection/mod.rs        | 22 ++++++++++++++++++++++
 3 files changed, 52 insertions(+), 1 deletion(-)

diff --git a/lib/pdm-api-types/src/metric_collection.rs b/lib/pdm-api-types/src/metric_collection.rs
index 99dac9b..954e1bd 100644
--- a/lib/pdm-api-types/src/metric_collection.rs
+++ b/lib/pdm-api-types/src/metric_collection.rs
@@ -178,3 +178,18 @@ impl CollectionSettings {
             .unwrap_or(DEFAULT_MIN_CONNECTION_DELAY)
     }
 }
+
+#[api]
+#[derive(Clone, Deserialize, Serialize)]
+#[serde(rename_all = "kebab-case")]
+/// Per-remote collection status.
+pub struct MetricCollectionStatus {
+    /// The remote's name.
+    pub remote: String,
+    /// Any error that occured during the last collection attempt.
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub error: Option<String>,
+    /// Timestamp of last successful collection.
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub last_collection: Option<i64>,
+}
diff --git a/server/src/api/metric_collection.rs b/server/src/api/metric_collection.rs
index 8b6748d..f89e6a0 100644
--- a/server/src/api/metric_collection.rs
+++ b/server/src/api/metric_collection.rs
@@ -5,7 +5,11 @@ use proxmox_rrd_api_types::{RrdMode, RrdTimeframe};
 use proxmox_schema::api;
 use proxmox_sortable_macro::sortable;
 
-use pdm_api_types::{remotes::REMOTE_ID_SCHEMA, rrddata::FullCollectionDatapoint};
+use pdm_api_types::{
+    remotes::REMOTE_ID_SCHEMA, rrddata::FullCollectionDatapoint, MetricCollectionStatus,
+};
+
+use crate::metric_collection;
 
 use super::rrd_common::{self, DataPoint};
 
@@ -21,6 +25,10 @@ const SUBDIRS: SubdirMap = &sorted!([
         "rrddata",
         &Router::new().get(&API_METHOD_GET_METRIC_COLLECTION_RRD_DATA)
     ),
+    (
+        "status",
+        &Router::new().get(&API_METHOD_GET_METRIC_COLLECTION_STATUS)
+    ),
 ]);
 
 #[api(
@@ -83,3 +91,9 @@ fn get_metric_collection_rrd_data(
     let base = "local/metric-collection";
     rrd_common::create_datapoints_from_rrd(base, timeframe, cf)
 }
+
+#[api]
+/// Read metric collection RRD data.
+fn get_metric_collection_status() -> Result<Vec<MetricCollectionStatus>, Error> {
+    metric_collection::get_status()
+}
diff --git a/server/src/metric_collection/mod.rs b/server/src/metric_collection/mod.rs
index 6551f2b..1069c40 100644
--- a/server/src/metric_collection/mod.rs
+++ b/server/src/metric_collection/mod.rs
@@ -7,6 +7,8 @@ use tokio::sync::mpsc::{self, Sender};
 
 use proxmox_sys::fs::CreateOptions;
 
+use pdm_api_types::MetricCollectionStatus;
+
 mod collection_task;
 pub mod rrd_cache;
 mod rrd_task;
@@ -88,3 +90,23 @@ pub async fn trigger_metric_collection() -> Result<(), Error> {
 
     Ok(())
 }
+
+/// Get each remote's metric collection status.
+pub fn get_status() -> Result<Vec<MetricCollectionStatus>, Error> {
+    let (remotes, _) = pdm_config::remotes::config()?;
+    let state = collection_task::load_state()?;
+
+    let mut result = Vec::new();
+
+    for remote in remotes.order {
+        if let Some(status) = state.get_status(&remote) {
+            result.push(MetricCollectionStatus {
+                remote,
+                error: status.error.clone(),
+                last_collection: status.last_collection,
+            })
+        }
+    }
+
+    Ok(result)
+}
-- 
2.39.5