OpenSource/teloxide
OpenSource/teloxide
1
0
Fork 0
mirror of https://github.com/teloxide/teloxide.git synced 2025-01-03 17:52:12 +01:00
Code Issues Projects Releases Packages Wiki Activity

Refactor dptree usage

Browse source 
Former-commit-id: fe3aa2e798
This commit is contained in:
Hirrolot 2022-10-07 10:14:36 +06:00
parent 760f843010
commit 744679419a
2 changed files with 14 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/dispatching/dialogue.rs
View file

@ -220,18 +220,17 @@ where
Upd: GetChatId + Clone + Send + Sync + 'static, Upd: GetChatId + Clone + Send + Sync + 'static,
Output: Send + Sync + 'static, Output: Send + Sync + 'static,
{ {
dptree::entry() dptree::filter_map(|storage: Arc<S>, upd: Upd| {
.chain(dptree::filter_map(|storage: Arc<S>, upd: Upd| { let chat_id = upd.chat_id()?;
let chat_id = upd.chat_id()?; Some(Dialogue::new(storage, chat_id))
Some(Dialogue::new(storage, chat_id)) })
})) .filter_map_async(|dialogue: Dialogue<D, S>| async move {
.chain(dptree::filter_map_async(|dialogue: Dialogue<D, S>| async move { match dialogue.get_or_default().await {
match dialogue.get_or_default().await { Ok(dialogue) => Some(dialogue),
Ok(dialogue) => Some(dialogue), Err(err) => {
Err(err) => { log::error!("dialogue.get_or_default() failed: {:?}", err);
log::error!("dialogue.get_or_default() failed: {:?}", err); None
None
}
} }
})) }
})
} }

4
src/dispatching/handler_ext.rs
View file

@ -88,8 +88,8 @@ where
C: BotCommands + Send + Sync + 'static, C: BotCommands + Send + Sync + 'static,
Output: Send + Sync + 'static, Output: Send + Sync + 'static,
{ {
dptree::entry().chain(dptree::filter_map(move |message: Message, me: Me| { dptree::filter_map(move |message: Message, me: Me| {
let bot_name = me.user.username.expect("Bots must have a username"); let bot_name = me.user.username.expect("Bots must have a username");
message.text().and_then(|text| C::parse(text, &bot_name).ok()) message.text().and_then(|text| C::parse(text, &bot_name).ok())
})) })
} }