Abstract over sd_notify

src/main.rs

@@ -5,7 +5,7 @@
 use std::process::ExitCode;
 
 use clap::Parser;
-use tracing::error;
+use tracing::{error, info};
 
 mod api;
 mod cli;
@@ -41,6 +41,31 @@ fn version() -> String {
     }
 }
 
+#[derive(Debug, Clone, Copy)]
+enum ApplicationState {
+    Ready,
+    Stopping,
+}
+
+fn set_application_state(state: ApplicationState) {
+    info!(?state, "Application state changed");
+
+    #[cfg(feature = "systemd")]
+    {
+        use sd_notify::NotifyState;
+
+        fn notify(states: &[NotifyState<'_>]) {
+            sd_notify::notify(false, states)
+                .expect("should be able to notify systemd");
+        }
+
+        match state {
+            ApplicationState::Ready => notify(&[NotifyState::Ready]),
+            ApplicationState::Stopping => notify(&[NotifyState::Stopping]),
+        };
+    }
+}
+
 #[tokio::main]
 async fn main() -> ExitCode {
     let args = cli::Args::parse();