wewo

src/calls/call_connection.rs

@@ -133,7 +133,7 @@ impl CallConnection {
                 let mut user_info = JsonValue::new_object();
                 let _ = user_info.insert("state", JsonValue::from("muted"));
                 let _ = user_info.insert("streaming", JsonValue::from(false));
-                let _ = users.insert(&caller.user_id.to_string(), user_info);
+                let _ = users.insert(&caller.user_id.read().await.to_string(), user_info);
             }
         }
         response = response.add_data(DataTypes::about, users);
@@ -249,8 +249,7 @@ impl CallConnection {
                         .to_json()
                         .to_string(),
                 );
-                group.lock().await.get_member(uid).await;
-                group.lock().await.remove_member(uid);
+                group.lock().await.disconnect_member(uid).await;
             }
             call_manager::remove_inactive().await;
         }

src/calls/call_group.rs

@@ -21,19 +21,21 @@ impl CallGroup {
         self.callers
             .insert(user_id, Arc::new(Caller::new(user_id, tx)));
     }
-    pub fn disconnect_member(&mut self, user_id: Uuid) {
-        if let Some(caller) = self.callers.get(&user_id) {
-            caller.disconnect();
-        }
-    }
-
     pub fn remove_member(&mut self, user_id: Uuid) {
         self.callers.remove(&user_id);
     }
+    pub fn get_member(&self, user_id: &Uuid) -> Option<&Arc<Caller>> {
+        self.callers.get(user_id)
+    }
+    pub async fn disconnect_member(&mut self, user_id: Uuid) {
+        if let Some(caller) = self.callers.get(&user_id) {
+            caller.disconnect().await;
+        }
+    }
 
-    pub fn is_empty(&self) -> bool {
+    pub async fn is_empty(&self) -> bool {
         for caller in self.callers.values() {
-            if !caller.is_connected() {
+            if !caller.is_connected().await {
                 return false;
             }
         }

src/calls/call_manager.rs

@@ -46,6 +46,7 @@ pub async fn remove_inactive() {
             .lock()
             .await
             .is_empty()
+            .await
         {
             rem.push(cg.clone());
         }

src/calls/caller.rs

@@ -1,15 +1,15 @@
 use std::sync::Arc;
 
 use futures::lock::Mutex;
-use tokio::sync::mpsc::UnboundedSender;
+use tokio::sync::{RwLock, mpsc::UnboundedSender};
 use tungstenite::Utf8Bytes;
 use uuid::Uuid;
 
 pub struct Caller {
-    pub user_id: Mutex<Uuid>,
+    pub user_id: RwLock<Uuid>,
     pub tx: Mutex<Option<UnboundedSender<Utf8Bytes>>>,
-    pub user_state: Mutex<CallUserState>,
-    pub streaming: Mutex<bool>,
+    pub user_state: RwLock<CallUserState>,
+    pub streaming: RwLock<bool>,
 }
 #[derive(Clone)]
 pub enum CallUserState {
@@ -21,26 +21,26 @@ pub enum CallUserState {
 impl Caller {
     pub fn new(user_id: Uuid, tx: UnboundedSender<Utf8Bytes>) -> Self {
         Self {
-            user_id: Mutex::new(user_id),
+            user_id: RwLock::new(user_id),
             tx: Mutex::new(Some(tx)),
-            user_state: Mutex::new(CallUserState::Active),
-            streaming: Mutex::new(false),
+            user_state: RwLock::new(CallUserState::Active),
+            streaming: RwLock::new(false),
         }
     }
 
-    pub fn send(&self, msg: impl Into<Utf8Bytes>) {
-        if let Some(tx) = &self.tx.lock().await {
+    pub async fn send(&self, msg: impl Into<Utf8Bytes>) {
+        if let Some(tx) = &*self.tx.lock().await {
             let _ = tx.send(msg.into());
         }
     }
 
-    pub fn disconnect(self: Arc<Self>) {
-        self.user_state = CallUserState::Disconnected;
-        self.tx = None;
+    pub async fn disconnect(self: &Arc<Self>) {
+        *self.user_state.write().await = CallUserState::Disconnected;
+        *self.tx.lock().await = None;
     }
 
-    pub fn is_connected(&self) -> bool {
-        if let CallUserState::Disconnected = self.user_state {
+    pub async fn is_connected(&self) -> bool {
+        if let CallUserState::Disconnected = *self.user_state.read().await {
             false
         } else {
             true