refactor: remove unncessary wrapping

Author: Devdutt Shenoi

src/handlers/http/health_check.rs

@@ -53,7 +53,7 @@ pub async fn check_shutdown_middleware(
     }
 }
 
-pub async fn handle_signals(shutdown_signal: Arc<Mutex<Option<oneshot::Sender<()>>>>) {
+pub async fn handle_signals(shutdown_signal: oneshot::Sender<()>) {
     #[cfg(windows)]
     {
         tokio::select! {
@@ -80,19 +80,10 @@ pub async fn handle_signals(shutdown_signal: Arc<Mutex<Option<oneshot::Sender<()
     }
 }
 
-async fn shutdown(shutdown_signal: Arc<Mutex<Option<oneshot::Sender<()>>>>) {
-    // Set the shutdown flag to true
-    let mut shutdown_flag = SIGNAL_RECEIVED.lock().await;
-    *shutdown_flag = true;
-
-    // Sync to local
-    crate::event::STREAM_WRITERS.unset_all();
-
-    // Trigger graceful shutdown
-    if let Some(shutdown_sender) = shutdown_signal.lock().await.take() {
-        let _ = shutdown_sender.send(());
-    }
+async fn shutdown(shutdown_signal: oneshot::Sender<()>) {
+    shutdown_signal.send(()).unwrap();
 }
+
 pub async fn readiness() -> HttpResponse {
     // Check the object store connection
     if CONFIG.storage().get_object_store().check().await.is_ok() {

src/handlers/http/modal/mod.rs

@@ -38,7 +38,7 @@ use openid::Discovered;
 use serde::Deserialize;
 use serde::Serialize;
 use ssl_acceptor::get_ssl_acceptor;
-use tokio::sync::{oneshot, Mutex};
+use tokio::sync::oneshot;
 use tracing::{error, info, warn};
 
 use super::audit;
@@ -110,14 +110,10 @@ pub trait ParseableServer {
 
         // Create a channel to trigger server shutdown
         let (shutdown_trigger, shutdown_rx) = oneshot::channel::<()>();
-        let server_shutdown_signal = Arc::new(Mutex::new(Some(shutdown_trigger)));
-
-        // Clone the shutdown signal for the signal handler
-        let shutdown_signal = server_shutdown_signal.clone();
 
         // Spawn the signal handler task
         let signal_task = tokio::spawn(async move {
-            health_check::handle_signals(shutdown_signal).await;
+            health_check::handle_signals(shutdown_trigger).await;
             println!("Received shutdown signal, notifying server to shut down...");
         });