feat: add output of downtime info

Author: c-git

src/heartbeat.rs

@@ -1,4 +1,10 @@
-use crate::{db::save_kv, model::schedule::UnixTimestamp};
+use std::time::Duration;
+
+use crate::{
+    db::{load_kv, save_kv},
+    model::schedule::UnixTimestamp,
+};
+use human_time::ToHumanTimeString;
 use tracing::{error, info};
 
 const KEY: &str = "HEARTBEAT";
@@ -20,7 +26,39 @@ pub fn start_heartbeat(db_pool: sqlx::PgPool) {
     });
 }
 
-pub fn time_since_last_heartbeat(db_pool: sqlx::PgPool) -> String {
-    // TODO 1: Get last heartbeat from DB
-    "First run".to_string()
+pub async fn last_heartbeat_info(db_pool: sqlx::PgPool) -> String {
+    match load_kv(&db_pool, KEY).await {
+        Some(db_value) => match UnixTimestamp::from_db_fmt(&db_value) {
+            Ok(last_heartbeat) => {
+                let Ok(now) = UnixTimestamp::now() else {
+                    return format!(
+                        "Last Heartbeat: {last_heartbeat} but Failed to get current timestamp"
+                    );
+                };
+                let seconds_since_last_heartbeat = now.0 - last_heartbeat.0;
+                if seconds_since_last_heartbeat < 0 {
+                    return format!(
+                        "Last heartbeat in the future?! Last heartbeat: {last_heartbeat}, Now: {now}"
+                    );
+                }
+                let Ok(seconds_since_last_heartbeat) = seconds_since_last_heartbeat.try_into()
+                else {
+                    // Invalid u64
+                    return format!(
+                        "Invalid u64!!! Seconds since heartbeat: {seconds_since_last_heartbeat},  Last heartbeat: {last_heartbeat}, Now: {now}"
+                    );
+                };
+                let downtime = Duration::from_secs(seconds_since_last_heartbeat);
+                format!(
+                    "Downtime: {}\nLast Heartbeat: {last_heartbeat}\nNow: {now}",
+                    downtime.to_human_time_string()
+                )
+            }
+            Err(err) => {
+                error!(?err);
+                "Error Loading Last Heartbeat".to_string()
+            }
+        },
+        None => "First run".to_string(),
+    }
 }

src/main.rs

@@ -84,9 +84,9 @@ async fn main(
                     error!("Development run detected but no guild ID found so slash commands NOT registered");
                 }
                 let connect_msg = format!(
-                    "{} is connected! Version: {} [{}]", 
+                    "{} is connected! Version: {}\n{}", 
                     ready.user.name, version!(),
-                    heartbeat::time_since_last_heartbeat(db_pool.clone()),
+                    heartbeat::last_heartbeat_info(db_pool.clone()).await,
                 );
                 info!("{connect_msg}");
                 if let Some(channel) = shared_config.channel_bot_status{

src/model/schedule.rs

@@ -13,7 +13,7 @@ pub mod protected_ops;
 pub type ScheduledTaskId = OneBasedId;
 
 #[derive(Debug, serde::Serialize, serde::Deserialize, Default, Clone, Copy)]
-pub struct UnixTimestamp(i32);
+pub struct UnixTimestamp(pub i32);
 impl UnixTimestamp {
     pub fn new(value: i32) -> Self {
         Self(value)