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: module receive command through rx

Browse Source
This commit is contained in:
appflowy 2021-06-27 01:24:00 +08:00
parent 85e8e4b8af
commit d2b53fe1f3
2 changed files with 22 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
rust-lib/flowy-sys/src/module/module.rs
View File

@ -37,7 +37,7 @@ pub type CommandServiceFactory = BoxServiceFactory<(), ServiceRequest, ServiceRe
pub struct Module {
name: String,
data: DataContainer,
factory_map: HashMap<Command, CommandServiceFactory>,
service_map: HashMap<Command, CommandServiceFactory>,
req_tx: UnboundedSender<FlowyRequest>,
req_rx: UnboundedReceiver<FlowyRequest>,
resp_tx: UnboundedSender<FlowyResponse>,
@ -49,7 +49,7 @@ impl Module {
Self {
name: "".to_owned(),
data: DataContainer::new(),
factory_map: HashMap::new(),
service_map: HashMap::new(),
req_tx,
req_rx,
resp_tx,
@ -74,11 +74,11 @@ impl Module {
R: Future + 'static,
R::Output: Responder + 'static,
{
self.factory_map.insert(command, factory(HandlerService::new(handler)));
self.service_map.insert(command, factory(HandlerService::new(handler)));
self
}
pub fn can_handle(&self, cmd: &Command) -> bool { self.factory_map.contains_key(cmd) }
pub fn can_handle(&self, cmd: &Command) -> bool { self.service_map.contains_key(cmd) }
pub fn req_tx(&self) -> UnboundedSender<FlowyRequest> { self.req_tx.clone() }
@ -90,6 +90,13 @@ impl Module {
},
}
}
pub fn service_sender_map(&self) -> HashMap<Command, UnboundedSender<FlowyRequest>> {
self.service_map
.keys()
.map(|key| (key.clone(), self.req_tx()))
.collect::<HashMap<_, _>>()
}
}
impl Future for Module {
@ -98,7 +105,7 @@ impl Future for Module {
loop {
match ready!(Pin::new(&mut self.req_rx).poll_recv(cx)) {
None => return Poll::Ready(()),
Some(request) => match self.factory_map.get(request.get_cmd()) {
Some(request) => match self.service_map.get(request.get_cmd()) {
Some(factory) => {
let fut = ModuleServiceFuture {
request,

24
rust-lib/flowy-sys/src/rt/system.rs
View File

@ -1,12 +1,13 @@
use crate::{
module::{Command, Module},
module::{Command, CommandServiceFactory, Module},
request::FlowyRequest,
response::FlowyResponse,
rt::Runtime,
service::BoxServiceFactory,
};
use futures_core::{future::LocalBoxFuture, ready, task::Context};
use futures_util::{future, pin_mut};
use std::{cell::RefCell, future::Future, io, sync::Arc};
use std::{cell::RefCell, collections::HashMap, future::Future, io, sync::Arc};
use tokio::{
macros::support::{Pin, Poll},
sync::{
@ -21,7 +22,7 @@ thread_local!(
pub struct FlowySystem {
resp_tx: UnboundedSender<FlowyResponse>,
modules: Vec<Module>,
sender_map: HashMap<Command, UnboundedSender<FlowyRequest>>,
}
impl FlowySystem {
@ -37,31 +38,26 @@ impl FlowySystem {
let mut system = Self {
resp_tx: resp_tx.clone(),
modules: vec![],
sender_map: HashMap::default(),
};
let factory = module_factory(resp_tx.clone());
factory.into_iter().for_each(|m| {
system.sender_map.extend(m.service_sender_map());
runtime.spawn(m);
// system.add_module(m);
});
FlowySystem::set_current(system);
let runner = SystemRunner { rt: runtime, stop_rx };
runner
}
pub fn handle_command(&self, cmd: Command, request: FlowyRequest) {
self.modules.iter().for_each(|m| {
if m.can_handle(&cmd) {
m.handle(request.clone());
}
})
if let Some(sender) = self.sender_map.get(&cmd) {
sender.send(request);
}
}
pub fn add_module(&mut self, module: Module) { self.modules.push(module); }
#[doc(hidden)]
pub fn set_current(sys: FlowySystem) {
CURRENT.with(|cell| {
@ -91,7 +87,7 @@ impl Future for SystemController {
match ready!(Pin::new(&mut self.resp_rx).poll_recv(cx)) {
None => return Poll::Ready(()),
Some(resp) => {
// FFI
// FF
println!("Receive response: {:?}", resp);
},
}