some improvments

shiroedev2024 · Apr 23, 2024 · 71d1bb7 · 71d1bb7
1 parent 4cd2a9d
commit 71d1bb7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/doh/src/run.rs b/doh/src/run.rs
@@ -68,15 +68,15 @@ pub fn run_doh(config: Config) -> DohResult<()> {
 pub fn shutdown_doh() -> DohResult<()> {
     debug!("stopping doh");
 
-    if let Some(sender) = SHUTDOWN_SENDER.lock().unwrap().as_ref() {
+    let mut sender_option = SHUTDOWN_SENDER.lock().map_err(|e| io::Error::new(io::ErrorKind::Other, e.to_string()))?;
+    if let Some(sender) = sender_option.take() {
         sender.blocking_send(()).map_err(|e| io::Error::new(io::ErrorKind::Other, e))?;
-        *SHUTDOWN_SENDER.lock().unwrap() = None;
     }
 
     Ok(())
 }
 
 /// is running
 pub fn is_doh_running() -> bool {
-    SHUTDOWN_SENDER.lock().unwrap().as_ref().is_some()
+    SHUTDOWN_SENDER.lock().map_or(false, |guard| guard.is_some())
 }