OpenSource/teloxide
OpenSource/teloxide
1
0
Fork 0
mirror of https://github.com/teloxide/teloxide.git synced 2024-10-24 01:47:08 +02:00
Code Issues Projects Releases Packages Wiki Activity

When creating Polling assert that it's an UpdateListener

Browse source
This commit is contained in:
Maybe Waffle 2022-07-04 23:39:04 +04:00
parent ee52bb28b4
commit 08da55f54f
2 changed files with 18 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/dispatching/update_listeners.rs
View file

@ -126,3 +126,11 @@ pub trait AsUpdateStream<'a, E> {
/// [`Stream`]: AsUpdateStream::Stream
fn as_stream(&'a mut self) -> Self::Stream;
}
#[inline(always)]
pub(crate) fn assert_update_listener<L, E>(listener: L) -> L
where
L: UpdateListener<E>,
{
listener
}

14
src/dispatching/update_listeners/polling.rs
View file

@ -15,7 +15,7 @@ use futures::{ready, stream::Stream};
use crate::{
dispatching::{
stop_token::{AsyncStopFlag, AsyncStopToken},
update_listeners::{AsUpdateStream, UpdateListener},
update_listeners::{assert_update_listener, AsUpdateStream, UpdateListener},
},
requests::{HasPayload, Request, Requester},
types::{AllowedUpdate, Update},
@ -97,7 +97,10 @@ where
pub fn build(self) -> Polling<R> {
let Self { bot, timeout, limit, allowed_updates, drop_pending_updates } = self;
let (token, flag) = AsyncStopToken::new_pair();
Polling { bot, timeout, limit, allowed_updates, drop_pending_updates, flag, token }
let polling =
Polling { bot, timeout, limit, allowed_updates, drop_pending_updates, flag, token };
assert_update_listener(polling)
}
}
@ -113,7 +116,10 @@ where
R: Requester + Send + 'static,
<R as Requester>::GetUpdates: Send,
{
Polling::builder(bot).timeout(Duration::from_secs(10)).delete_webhook().await.build()
let polling =
Polling::builder(bot).timeout(Duration::from_secs(10)).delete_webhook().await.build();
assert_update_listener(polling)
}
/// Returns a long polling update listener with some additional options.
@ -132,7 +138,7 @@ where
builder.timeout = timeout;
builder.limit = limit;
builder.allowed_updates = allowed_updates;
builder.build()
assert_update_listener(builder.build())
}
async fn delete_webhook_if_setup<R>(requester: &R)