feat: ✨ destroy chat after timeout

configs/server_config.txt

@@ -1,3 +1,4 @@
 axum_address:192.168.1.2
 port:2334
 max_message:150
+chat_cleaning_timeout:600

src/chat.rs

@@ -1,4 +1,4 @@
-use std::{collections::VecDeque, time::Duration};
+use std::collections::VecDeque;
 
 use chrono::Utc;
 use serde::Serialize;
@@ -15,6 +15,7 @@ pub struct Message {
 pub struct Chat {
     pub room_id: String,
     pub messages: VecDeque<Message>,
+    pub last_interaction: u64,
 }
 
 impl Chat {
@@ -22,6 +23,7 @@ impl Chat {
         Chat {
             room_id,
             messages: vec![].into(),
+            last_interaction: Utc::now().timestamp() as u64,
         }
     }
 
@@ -29,6 +31,7 @@ impl Chat {
         message.calculate_hash();
         if !self.is_message_exists(message.clone()) {
             self.messages.push_back(message);
+            self.last_interaction = Utc::now().timestamp() as u64;
             if self.messages.len() > max_message_count as usize {
                 let _ = self.messages.pop_front();
             }
@@ -45,15 +48,6 @@ impl Chat {
         }
         false
     }
-
-    pub async fn message_cleaner(&mut self, max_message_count: u64) {
-        loop {
-            tokio::time::sleep(Duration::from_secs(1)).await;
-            if self.messages.len() > max_message_count as usize {
-                self.messages.clear();
-            }
-        }
-    }
 }
 
 impl Message {

src/lib.rs

@@ -1,6 +1,7 @@
-use std::{net::IpAddr, sync::Arc};
+use std::{net::IpAddr, sync::Arc, time::Duration};
 
 use chat::Chat;
+use chrono::Utc;
 use tokio::sync::Mutex;
 
 pub mod chat;
@@ -12,12 +13,14 @@ pub struct ServerConfig {
     pub ip_address: IpAddr,
     pub port: u16,
     pub max_message_counter: u16,
+    pub chat_cleaning_timeout: u16,
 }
 
 #[derive(Debug, Clone)]
 pub struct AppState {
     pub chats: Arc<Mutex<Vec<Chat>>>,
     pub max_message_counter: u16,
+    pub chat_cleaning_timeout: u16,
 }
 
 impl AppState {
@@ -30,4 +33,29 @@ impl AppState {
         }
         None
     }
+
+    pub async fn chat_destroyer(
+        chats: Arc<Mutex<Vec<Chat>>>,
+        room_id: String,
+        chat_cleaning_timeout: u16,
+    ) {
+        let timeout_control = (chat_cleaning_timeout / 10) as u64;
+        loop {
+            tokio::time::sleep(Duration::from_secs(timeout_control)).await;
+            let mut chats = chats.lock().await;
+            let current = Utc::now().timestamp() as u64;
+            for i in 0..chats.len() {
+                if chats[i].room_id == room_id {
+                    if chats[i].last_interaction < current - chat_cleaning_timeout as u64 {
+                        chats.remove(i);
+                        return;
+                    }
+                }
+            }
+            // if chats[chat_index].last_interaction < current - chat_cleaning_timeout as u64 {
+            //     chats.remove(chat_index);
+            //     return;
+            // }
+        }
+    }
 }

src/main.rs

@@ -18,6 +18,7 @@ async fn main() {
     let state = AppState {
         chats: Arc::new(Mutex::new(vec![])),
         max_message_counter: server_config.max_message_counter,
+        chat_cleaning_timeout: server_config.chat_cleaning_timeout,
     };
     let app = routing(axum::extract::State(state)).await;
     let addr = SocketAddr::new(server_config.ip_address, server_config.port);

src/routing.rs

@@ -53,7 +53,14 @@ async fn receive_message(
             let mut new_chat = Chat::new(room_id);
             new_chat.add_message(message, state.max_message_counter);
             let mut chats = state.chats.lock().await;
+            let room_id = new_chat.room_id.clone();
             chats.push(new_chat);
+            drop(chats);
+            tokio::spawn(AppState::chat_destroyer(
+                state.chats.clone(),
+                room_id,
+                state.chat_cleaning_timeout,
+            ));
         }
     }
 }

src/utils.rs

@@ -21,10 +21,13 @@ pub fn read_server_config() -> ServerConfig {
     let max_message_counter: Vec<&str> = configs_uncleaned[2].split(':').collect();
     let max_message_counter = max_message_counter[1].parse().unwrap();
 
+    let chat_cleaning_timeout: Vec<&str> = configs_uncleaned[3].split(':').collect();
+    let chat_cleaning_timeout = chat_cleaning_timeout[1].parse().unwrap();
     ServerConfig {
         ip_address,
         port,
         max_message_counter,
+        chat_cleaning_timeout,
     }
 }