OpenSource/teloxide
OpenSource/teloxide
1
0
Fork 0
mirror of https://github.com/teloxide/teloxide.git synced 2024-12-22 14:35:36 +01:00
Code Issues Projects Releases Packages Wiki Activity

Improve StopFlag's implementation

Browse source
This commit is contained in:
Maybe Waffle 2022-09-09 21:39:39 +04:00
parent 9527f82608
commit 0807eb57e1
1 changed files with 5 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/stop.rs
View file

@ -5,7 +5,7 @@
//! [flag]: StopFlag
//! [listeners]: crate::dispatching::update_listeners
use std::{future::Future, pin::Pin, task};
use std::{convert::Infallible, future::Future, pin::Pin, task};
use futures::future::{pending, AbortHandle, Abortable, Pending};
@ -31,7 +31,7 @@ pub struct StopToken(AbortHandle);
/// [`is_stopped`]: StopFlag::is_stopped
#[pin_project::pin_project]
#[derive(Clone)]
pub struct StopFlag(#[pin] Abortable<Pending<()>>);
pub struct StopFlag(#[pin] Abortable<Pending<Infallible>>);
impl StopToken {
/// "Stops" the flag associated with this token.
@ -56,12 +56,9 @@ impl Future for StopFlag {
type Output = ();
fn poll(self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> task::Poll<Self::Output> {
self.project().0.poll(cx).map(|res| {
debug_assert!(
res.is_err(),
"Pending Future can't ever be resolved, so Abortable is only resolved when \
canceled"
);
self.project().0.poll(cx).map(|res| match res {
Err(_aborted) => (),
Ok(unreachable) => match unreachable {},
})
}
}