Support for systemd notify

Signed-off-by: Sebastian Schildt <[email protected]>
eclipse · Mar 5, 2024 · cfd068b · cfd068b
1 parent 6936959
commit cfd068b
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/kuksa_databroker/databroker/Cargo.toml b/kuksa_databroker/databroker/Cargo.toml
@@ -63,6 +63,10 @@ futures = { version = "0.3.28", optional = true }
 chrono = { version = "0.4.31", optional = true, features = ["std"] }
 uuid = { version = "1.4.1", optional = true, features = ["v4"] }
 
+# systemd related dependency, only relevant on linux systems
+[target.'cfg(target_os = "linux")'.dependencies]
+sd-notify = "0.4.1"
+
 [features]
 default = ["tls"]
 tls = ["tonic/tls"]

diff --git a/kuksa_databroker/databroker/src/grpc/server.rs b/kuksa_databroker/databroker/src/grpc/server.rs
@@ -101,6 +101,24 @@ where
 {
     let socket_addr = addr.into();
     let listener = TcpListener::bind(socket_addr).await?;
+
+    /* On Linux systems try to notify daemon readiness to systemd.
+     * This function determines whether the a system is using systemd
+     * or not, so it is safe to use on non-systemd systems as well.
+     */
+    #[cfg(target_os = "linux")]
+    {
+        match sd_notify::booted() {
+            Ok(true) => {
+                info!("Notifying systemd that the service is ready");
+                sd_notify::notify(false, &[sd_notify::NotifyState::Ready])?;
+            }
+            _ => {
+                debug!("System is not using systemd, will not try to notify");
+            }
+        }
+    }
+
     serve_with_incoming_shutdown(
         listener,
         broker,