github-mirrors/AppFlowy
1
0
Fork 0
mirror of https://github.com/AppFlowy-IO/AppFlowy.git synced 2025-04-25 07:07:32 -04:00
Code Issues Projects Releases Packages Wiki Activity

mock websocket

Browse source
This commit is contained in:
appflowy 2021-12-10 22:18:44 +08:00
parent 34441ee076
commit aeb448c4de
11 changed files with 175 additions and 56 deletions
Download patch file Download diff file Expand all files Collapse all files

24
frontend/rust-lib/flowy-sdk/src/deps_resolve/document_deps.rs
View file

@ -29,7 +29,7 @@ impl DocumentDepsResolver {
user: self.user_session.clone(),
});
let ws_manager = Arc::new(WsDocumentManager::new(sender));
let ws_handler = Arc::new(WsDocumentReceiver {
let ws_handler = Arc::new(DocumentWsMessageReceiver {
inner: ws_manager.clone(),
});
self.user_session.add_ws_handler(ws_handler);
@ -74,15 +74,13 @@ struct WsSenderImpl {
impl DocumentWebSocket for WsSenderImpl {
fn send(&self, data: WsDocumentData) -> Result<(), DocError> {
if cfg!(feature = "http_server") {
let bytes: Bytes = data.try_into().unwrap();
let msg = WsMessage {
module: WsModule::Doc,
data: bytes.to_vec(),
};
let sender = self.user.ws_sender().map_err(internal_error)?;
sender.send_msg(msg).map_err(internal_error)?;
}
let bytes: Bytes = data.try_into().unwrap();
let msg = WsMessage {
module: WsModule::Doc,
data: bytes.to_vec(),
};
let sender = self.user.ws_sender().map_err(internal_error)?;
sender.send(msg).map_err(internal_error)?;
Ok(())
}
@ -90,15 +88,15 @@ impl DocumentWebSocket for WsSenderImpl {
fn state_notify(&self) -> WsStateReceiver { self.user.ws_state_notifier() }
}
struct WsDocumentReceiver {
struct DocumentWsMessageReceiver {
inner: Arc<WsDocumentManager>,
}
impl WsMessageHandler for WsDocumentReceiver {
impl WsMessageHandler for DocumentWsMessageReceiver {
fn source(&self) -> WsModule { WsModule::Doc }
fn receive_message(&self, msg: WsMessage) {
let data = Bytes::from(msg.data);
self.inner.handle_ws_data(data);
self.inner.did_receive_ws_data(data);
}
}