fixed broadcast and cleanup

2 files changed, 16 insertions, 12 deletions

diff --git a/src/routes/start.rs b/src/routes/start.rs
index 9cd358b..271f924 100644
--- a/src/routes/start.rs
+++ b/src/routes/start.rs
@@ -4,7 +4,7 @@ use serde::{Deserialize, Serialize};
 use std::sync::Arc;
 use axum::extract::State;
 use serde_json::{json, Value};
-use tracing::{debug, info, warn};
+use tracing::{debug, info};
 use uuid::Uuid;
 use crate::auth::auth;
 use crate::config::SETTINGS;
@@ -16,7 +16,6 @@ use crate::services::ping::PingValue;
 #[axum_macros::debug_handler]
 pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap, Json(payload): Json<StartPayload>) -> Result<Json<Value>, WebolError> {
     info!("POST request");
-    warn!("{:?}", state.ping_map);
     let secret = headers.get("authorization");
     let authorized = auth(secret).map_err(WebolError::Auth)?;
     if authorized {
@@ -46,7 +45,7 @@ pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap
             let uuid_gen = Uuid::new_v4().to_string();
             let uuid_genc = uuid_gen.clone();
             tokio::spawn(async move {
-                debug!("Init ping service");
+                debug!("init ping service");
                 state.ping_map.insert(uuid_gen.clone(), PingValue { ip: device.ip.clone(), online: false });
 
                 crate::services::ping::spawn(state.ping_send.clone(), device.ip, uuid_gen.clone(), &state.ping_map).await
diff --git a/src/services/ping.rs b/src/services/ping.rs
index a26dacc..d900acb 100644
--- a/src/services/ping.rs
+++ b/src/services/ping.rs
@@ -41,7 +41,7 @@ pub async fn spawn(tx: Sender<BroadcastCommands>, ip: String, uuid: String, ping
             }
         } else {
             let (_, duration) = ping.map_err(|err| error!("{}", err.to_string())).expect("fatal error");
-            debug!("Ping took {:?}", duration);
+            debug!("ping took {:?}", duration);
             cont = false;
             handle_broadcast_send(&tx, ip.clone(), ping_map, uuid.clone()).await;
         };
@@ -50,10 +50,12 @@ pub async fn spawn(tx: Sender<BroadcastCommands>, ip: String, uuid: String, ping
 
 async fn handle_broadcast_send(tx: &Sender<BroadcastCommands>, ip: String, ping_map: &PingMap, uuid: String) {
     debug!("send pingsuccess message");
-    ping_map.insert(uuid.clone(), PingValue { ip: ip.clone(), online: true });
     let _ = tx.send(BroadcastCommands::PingSuccess(uuid.clone()));
+    trace!("sent message");
+    ping_map.insert(uuid.clone(), PingValue { ip: ip.clone(), online: true });
+    trace!("updated ping_map");
     tokio::time::sleep(tokio::time::Duration::from_secs(60)).await;
-    trace!("remove {} from ping_map after success", uuid);
+    debug!("remove {} from ping_map after success", uuid);
     ping_map.remove(&uuid);
 }
 
@@ -70,12 +72,12 @@ pub async fn status_websocket(mut socket: WebSocket, state: Arc<AppState>) {
 
     trace!("Search for uuid: {:?}", uuid);
 
-    match state.ping_map.get(&uuid) {
-        Some(device) => {
-            debug!("got device: {} (online: {})", device.ip, device.online);
-            let _ = socket.send(process_device(state.clone(), uuid, device.to_owned()).await).await;
+    let device_exists = state.ping_map.contains_key(&uuid);
+    match device_exists {
+        true => {
+            let _ = socket.send(process_device(state.clone(), uuid).await).await;
         },
-        None => {
+        false => {
             debug!("didn't find any device");
             let _ = socket.send(Message::Text(format!("notfound_{}", uuid))).await;
         },
@@ -84,7 +86,10 @@ pub async fn status_websocket(mut socket: WebSocket, state: Arc<AppState>) {
     let _ = socket.close().await;
 }
 
-async fn process_device(state: Arc<AppState>, uuid: String, device: PingValue) -> Message {
+async fn process_device(state: Arc<AppState>, uuid: String) -> Message {
+    let pm = state.ping_map.clone().into_read_only();
+    let device = pm.get(&uuid).expect("fatal error");
+    debug!("got device: {} (online: {})", device.ip, device.online);
     match device.online {
         true => {
             debug!("already started");