From e7e8e1a18b6e3bcdf190931869527489704296a4 Mon Sep 17 00:00:00 2001
From: Kingtous <kingtous@qq.com>
Date: Sun, 29 Jan 2023 22:23:18 +0800
Subject: [PATCH] opt: send audio frame when connected

---
 src/client/io_loop.rs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/client/io_loop.rs b/src/client/io_loop.rs
index bac1e5d23..f16c9af72 100644
--- a/src/client/io_loop.rs
+++ b/src/client/io_loop.rs
@@ -90,7 +90,6 @@ impl<T: InvokeUiSession> Remote<T> {
 
     pub async fn io_loop(&mut self, key: &str, token: &str) {
         let stop_clipboard = self.start_clipboard();
-        let stop_client_audio = self.start_client_audio();
         let mut last_recv_time = Instant::now();
         let mut received = false;
         let conn_type = if self.handler.is_file_transfer() {
@@ -114,6 +113,8 @@ impl<T: InvokeUiSession> Remote<T> {
                 SERVER_FILE_TRANSFER_ENABLED.store(true, Ordering::SeqCst);
                 self.handler.set_connection_type(peer.is_secured(), direct); // flutter -> connection_ready
                 self.handler.set_connection_info(direct, false);
+                // Start client audio when connection is established.
+                let stop_client_audio = self.start_client_audio();
 
                 // just build for now
                 #[cfg(not(windows))]
@@ -218,6 +219,10 @@ impl<T: InvokeUiSession> Remote<T> {
                     }
                 }
                 log::debug!("Exit io_loop of id={}", self.handler.id);
+                // Stop client audio server.
+                if let Some(stop) = stop_client_audio {
+                    stop.send(()).ok();
+                }
             }
             Err(err) => {
                 self.handler
@@ -227,9 +232,6 @@ impl<T: InvokeUiSession> Remote<T> {
         if let Some(stop) = stop_clipboard {
             stop.send(()).ok();
         }
-        if let Some(stop) = stop_client_audio {
-            stop.send(()).ok();
-        }
         SERVER_KEYBOARD_ENABLED.store(false, Ordering::SeqCst);
         SERVER_CLIPBOARD_ENABLED.store(false, Ordering::SeqCst);
         SERVER_FILE_TRANSFER_ENABLED.store(false, Ordering::SeqCst);