fix listen on websocket error

frontend/rust-lib/flowy-net/src/services/ws/manager.rs

@@ -24,7 +24,6 @@ impl WsManager {
             local_web_socket()
         };
         let (status_notifier, _) = broadcast::channel(10);
-        listen_on_websocket(ws.clone());
         WsManager {
             inner: ws,
             connect_type: RwLock::new(NetworkType::default()),
@@ -36,7 +35,6 @@ impl WsManager {
     pub async fn start(&self, token: String) -> Result<(), FlowyError> {
         let addr = format!("{}/{}", self.addr, token);
         self.inner.stop_connect().await?;
-
         let _ = self.inner.start_connect(addr).await?;
         Ok(())
     }
@@ -79,28 +77,33 @@ impl WsManager {
     pub fn ws_sender(&self) -> Result<Arc<dyn FlowyWsSender>, FlowyError> { self.inner.ws_sender() }
 }
 
-#[tracing::instrument(level = "debug", skip(ws))]
-fn listen_on_websocket(ws: Arc<dyn FlowyWebSocket>) {
-    let mut notify = ws.subscribe_connect_state();
-    let _ = tokio::spawn(async move {
-        loop {
-            match notify.recv().await {
-                Ok(state) => {
-                    tracing::info!("Websocket state changed: {}", state);
-                    match state {
-                        WSConnectState::Init => {},
-                        WSConnectState::Connected => {},
-                        WSConnectState::Connecting => {},
-                        WSConnectState::Disconnected => retry_connect(ws.clone(), 100).await,
-                    }
-                },
-                Err(e) => {
-                    tracing::error!("Websocket state notify error: {:?}", e);
-                    break;
-                },
+#[tracing::instrument(level = "debug", skip(manager))]
+pub fn listen_on_websocket(manager: Arc<WsManager>) {
+    if cfg!(feature = "http_server") {
+        let ws = manager.inner.clone();
+        let mut notify = manager.inner.subscribe_connect_state();
+        let _ = tokio::spawn(async move {
+            loop {
+                match notify.recv().await {
+                    Ok(state) => {
+                        tracing::info!("Websocket state changed: {}", state);
+                        match state {
+                            WSConnectState::Init => {},
+                            WSConnectState::Connected => {},
+                            WSConnectState::Connecting => {},
+                            WSConnectState::Disconnected => retry_connect(ws.clone(), 100).await,
+                        }
+                    },
+                    Err(e) => {
+                        tracing::error!("Websocket state notify error: {:?}", e);
+                        break;
+                    },
+                }
             }
-        }
-    });
+        });
+    } else {
+        // do nothing
+    };
 }
 
 async fn retry_connect(ws: Arc<dyn FlowyWebSocket>, count: usize) {

frontend/rust-lib/flowy-sdk/src/lib.rs

@@ -5,7 +5,10 @@ use crate::deps_resolve::{DocumentDepsResolver, WorkspaceDepsResolver};
 use backend_service::configuration::ClientServerConfiguration;
 use flowy_core::{errors::FlowyError, module::init_core, prelude::CoreContext};
 use flowy_document::module::FlowyDocument;
-use flowy_net::{entities::NetworkType, services::ws::WsManager};
+use flowy_net::{
+    entities::NetworkType,
+    services::ws::{listen_on_websocket, WsManager},
+};
 use flowy_user::{
     prelude::UserStatus,
     services::user::{UserSession, UserSessionConfig},
@@ -113,8 +116,10 @@ fn _init(
 
     dispatch.spawn(async move {
         user_session.init();
-        _listen_user_status(ws_manager, subscribe_user_status, core.clone()).await;
+        listen_on_websocket(ws_manager.clone());
+        _listen_user_status(ws_manager.clone(), subscribe_user_status, core.clone()).await;
     });
+
     dispatch.spawn(async move {
         _listen_network_status(subscribe_network_type, cloned_core).await;
     });