impl dashmap

3 files changed, 9 insertions, 11 deletions

diff --git a/src/main.rs b/src/main.rs
index 545d8fe..762a817 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,12 +1,11 @@
-use std::collections::HashMap;
 use std::env;
 use std::sync::Arc;
 use axum::{Router, routing::post};
 use axum::routing::{get, put};
+use dashmap::DashMap;
 use sqlx::PgPool;
 use time::util::local_offset;
 use tokio::sync::broadcast::{channel, Sender};
-use tokio::sync::Mutex;
 use tracing::{info, level_filters::LevelFilter};
 use tracing_subscriber::{EnvFilter, fmt::{self, time::LocalTime}, prelude::*};
 use crate::config::SETTINGS;
@@ -51,9 +50,9 @@ async fn main() {
 
     let (tx, _) = channel(32);
 
-    let ping_map: HashMap<String, (String, bool)> = HashMap::new();
+    let ping_map: DashMap<String, (String, bool)> = DashMap::new();
     
-    let shared_state = Arc::new(AppState { db, ping_send: tx, ping_map: Arc::new(Mutex::new(ping_map)) });
+    let shared_state = Arc::new(AppState { db, ping_send: tx, ping_map: Arc::new(ping_map) });
 
     let app = Router::new()
         .route("/start", post(start))
diff --git a/src/routes/start.rs b/src/routes/start.rs
index b1c8a73..5b73281 100644
--- a/src/routes/start.rs
+++ b/src/routes/start.rs
@@ -47,7 +47,7 @@ pub async fn start(State(state): State<Arc<crate::AppState>>, headers: HeaderMap
             let uuid_gencc = uuid_gen.clone();
             tokio::spawn(async move{
                 debug!("Init ping service");
-                state.ping_map.lock().await.insert(uuid_gen, ("192.168.178.94".to_string(), false));
+                state.ping_map.insert(uuid_gen, ("192.168.178.94".to_string(), false));
 
                 warn!("{:?}", state.ping_map);
 
diff --git a/src/services/ping.rs b/src/services/ping.rs
index 6835fc0..ed848fc 100644
--- a/src/services/ping.rs
+++ b/src/services/ping.rs
@@ -1,16 +1,15 @@
 use std::borrow::Cow;
-use std::collections::HashMap;
 use std::sync::Arc;
 
 use axum::extract::{ws::WebSocket};
 use axum::extract::ws::{CloseFrame, Message};
+use dashmap::DashMap;
 use tokio::sync::broadcast::{Sender};
-use tokio::sync::Mutex;
 use tracing::{debug, trace, warn};
 
 use crate::error::WebolError;
 
-pub type PingMap = Arc<Mutex<HashMap<String, (String, bool)>>>;
+pub type PingMap = Arc<DashMap<String, (String, bool)>>;
 
 pub async fn spawn(tx: Sender<BroadcastCommands>, ip: String, uuid: String, ping_map: PingMap) -> Result<(), WebolError> {
     let payload = [0; 8];
@@ -41,11 +40,11 @@ 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!("sending pingsuccess message");
-    ping_map.lock().await.insert(uuid.clone(), (ip.clone(), true));
+    ping_map.insert(uuid.clone(), (ip.clone(), true));
     let _ = tx.send(BroadcastCommands::PingSuccess(ip));
     tokio::time::sleep(tokio::time::Duration::from_secs(60)).await;
     trace!("remove {} from ping_map", uuid);
-    ping_map.lock().await.remove(&uuid);
+    ping_map.remove(&uuid);
 }
 
 #[derive(Clone, Debug)]
@@ -63,7 +62,7 @@ pub async fn status_websocket(mut socket: WebSocket, tx: Sender<BroadcastCommand
     trace!("Search for uuid: {:?}", uuid);
 
     // TODO: Handle Error
-    let device = ping_map.lock().await.get(&uuid).unwrap().to_owned();
+    let device = ping_map.get(&uuid).unwrap().to_owned();
 
     trace!("got device: {:?}", device);