Keep uptime kuma connection alive between syncs to reduce load on uptime kuma

Author: BigBoot

autokuma/src/sync.rs

@@ -18,6 +18,7 @@ pub struct Sync {
     app_state: Arc<AppState>,
     auth_token: Option<String>,
     sources: Vec<Box<dyn Source>>,
+    client: Option<Arc<Client>>,
 }
 
 impl Sync {
@@ -27,6 +28,7 @@ impl Sync {
             app_state: state.clone(),
             auth_token: state.config.kuma.auth_token.clone(),
             sources: crate::sources::get_sources(state),
+            client: None,
         })
     }
 
@@ -199,12 +201,21 @@ impl Sync {
         })
     }
 
+    async fn get_connection(&mut self) -> Result<Arc<Client>> {
+        if self.client.is_none() {
+            let kuma_config = kuma_client::Config {
+                auth_token: self.auth_token.clone(),
+                ..self.app_state.config.kuma.clone()
+            };
+            let kuma = Client::connect(kuma_config).await?;
+            self.client = Some(Arc::new(kuma));
+        }
+
+        Ok(self.client.as_ref().unwrap().clone())
+    }
+
     async fn do_sync(&mut self) -> Result<()> {
-        let kuma_config = kuma_client::Config {
-            auth_token: self.auth_token.clone(),
-            ..self.app_state.config.kuma.clone()
-        };
-        let kuma = Client::connect(kuma_config).await?;
+        let kuma = self.get_connection().await?;
 
         crate::migrations::migrate(&self.app_state, &kuma).await?;
 

kuma-client/src/client.rs

@@ -235,6 +235,16 @@ impl Worker {
         Ok(())
     }
 
+    async fn on_delete_monitor_from_list(self: &Arc<Self>, monitor_id: i32) -> Result<()> {
+        self.monitors.lock().await.remove(&monitor_id.to_string());
+        Ok(())
+    }
+
+    async fn on_update_monitor_into_list(self: &Arc<Self>, monitors: MonitorList) -> Result<()> {
+        self.monitors.lock().await.extend(monitors);
+        Ok(())
+    }
+
     async fn on_event(self: &Arc<Self>, event: Event, payload: Value) -> Result<()> {
         match event {
             Event::MonitorList => {
@@ -280,6 +290,20 @@ impl Worker {
             Event::Info => self.on_info().await?,
             Event::AutoLogin => self.on_auto_login().await?,
             Event::LoginRequired => self.on_login_required().await?,
+            Event::UpdateMonitorIntoList => {
+                self.on_update_monitor_into_list(
+                    serde_json::from_value(payload)
+                        .log_error(module_path!(), |_| {
+                            "Failed to deserialize UpdateMonitorIntoList"
+                        })
+                        .unwrap(),
+                )
+                .await?
+            }
+            Event::DeleteMonitorFromList => {
+                self.on_delete_monitor_from_list(payload.as_i64().unwrap().try_into().unwrap())
+                    .await?
+            }
             _ => {}
         }
 

kuma-client/src/models/event.rs

@@ -22,4 +22,6 @@ pub(crate) enum Event {
     StatusPageList,
     Uptime,
     LoginRequired,
+    UpdateMonitorIntoList,
+    DeleteMonitorFromList,
 }