[pdm-devel] [PATCH proxmox-datacenter-manager 05/12] unprivileged api daemon: tasks: add remote update refresh task

[Lukas Wagner]

This commit adds a very simple refresh tasks that runs at relatively
long interval. The task polls all remotes for available updates and puts
the results in the update cache.

Signed-off-by: Lukas Wagner <l.wagner at proxmox.com>
---
 server/src/bin/proxmox-datacenter-api/main.rs |  1 +
 .../bin/proxmox-datacenter-api/tasks/mod.rs   |  1 +
 .../tasks/remote_updates.rs                   | 44 +++++++++++++++++++
 3 files changed, 46 insertions(+)
 create mode 100644 server/src/bin/proxmox-datacenter-api/tasks/remote_updates.rs

diff --git a/server/src/bin/proxmox-datacenter-api/main.rs b/server/src/bin/proxmox-datacenter-api/main.rs
index c0f0e3a4..420a3b4d 100644
--- a/server/src/bin/proxmox-datacenter-api/main.rs
+++ b/server/src/bin/proxmox-datacenter-api/main.rs
@@ -377,6 +377,7 @@ async fn run(debug: bool) -> Result<(), Error> {
     tasks::remote_node_mapping::start_task();
     resource_cache::start_task();
     tasks::remote_tasks::start_task()?;
+    tasks::remote_updates::start_task()?;
 
     server.await?;
     log::info!("server shutting down, waiting for active workers to complete");
diff --git a/server/src/bin/proxmox-datacenter-api/tasks/mod.rs b/server/src/bin/proxmox-datacenter-api/tasks/mod.rs
index a6b1f439..f4d1d3a1 100644
--- a/server/src/bin/proxmox-datacenter-api/tasks/mod.rs
+++ b/server/src/bin/proxmox-datacenter-api/tasks/mod.rs
@@ -1,2 +1,3 @@
 pub mod remote_node_mapping;
 pub mod remote_tasks;
+pub mod remote_updates;
diff --git a/server/src/bin/proxmox-datacenter-api/tasks/remote_updates.rs b/server/src/bin/proxmox-datacenter-api/tasks/remote_updates.rs
new file mode 100644
index 00000000..bd9e178d
--- /dev/null
+++ b/server/src/bin/proxmox-datacenter-api/tasks/remote_updates.rs
@@ -0,0 +1,44 @@
+use anyhow::Error;
+
+use server::{remote_updates, task_utils};
+
+const REFRESH_TIME: u64 = 6 * 3600;
+
+/// Start the remote task fetching task
+pub fn start_task() -> Result<(), Error> {
+    tokio::spawn(async move {
+        let task_scheduler = std::pin::pin!(RemoteUpdateRefreshTask {}.run());
+        let abort_future = std::pin::pin!(proxmox_daemon::shutdown_future());
+        futures::future::select(task_scheduler, abort_future).await;
+    });
+
+    Ok(())
+}
+
+struct RemoteUpdateRefreshTask {}
+
+impl RemoteUpdateRefreshTask {
+    async fn run(self) {
+        loop {
+            self.refresh().await;
+            self.wait_for_refresh().await;
+        }
+    }
+
+    async fn refresh(&self) {
+        if let Err(err) = self.do_refresh().await {
+            log::error!("could not refresh remote update cache: {err:#}");
+        }
+    }
+
+    async fn do_refresh(&self) -> Result<(), Error> {
+        let (config, _digest) = tokio::task::spawn_blocking(pdm_config::remotes::config).await??;
+        remote_updates::refresh_update_summary_cache(config.into_iter().map(|(_, r)| r).collect())
+            .await
+    }
+
+    async fn wait_for_refresh(&self) {
+        let instant = task_utils::next_aligned_instant(REFRESH_TIME);
+        tokio::time::sleep_until(instant.into()).await;
+    }
+}
-- 
2.47.3