Mirrored_Repos/AppFlowy
1
0
Fork 0
mirror of https://github.com/AppFlowy-IO/AppFlowy.git synced 2024-08-30 18:12:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix listen on websocket error

Browse Source
This commit is contained in:
appflowy 2021-12-20 12:06:17 +08:00
parent 936e133624
commit 16c702b1e1
2 changed files with 33 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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) {

9
frontend/rust-lib/flowy-sdk/src/lib.rs
View File

@ -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;
});