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

Add a bit of trace! logging to polling for easier debug

Browse source
This commit is contained in:
Maybe Waffle 2023-09-25 19:25:31 +04:00
parent c7feb38995
commit 836080523f
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
crates/teloxide/src/update_listeners/polling.rs
View file

@ -333,6 +333,7 @@ impl<B: Requester> Stream for PollingStream<'_, B> {
type Item = Result<Update, B::Err>; type Item = Result<Update, B::Err>;
fn poll_next(mut self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Option<Self::Item>> { fn poll_next(mut self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Option<Self::Item>> {
log::trace!("polling polling stream");
let mut this = self.as_mut().project(); let mut this = self.as_mut().project();
if *this.force_stop { if *this.force_stop {
@ -342,6 +343,7 @@ impl<B: Requester> Stream for PollingStream<'_, B> {
// Poll in-flight future until completion // Poll in-flight future until completion
if let Some(in_flight) = this.in_flight.as_mut().as_pin_mut() { if let Some(in_flight) = this.in_flight.as_mut().as_pin_mut() {
let res = ready!(in_flight.poll(cx)); let res = ready!(in_flight.poll(cx));
log::trace!("in-flight completed");
this.in_flight.set(None); this.in_flight.set(None);
match res { match res {
@ -380,7 +382,10 @@ impl<B: Requester> Stream for PollingStream<'_, B> {
// //
// When stopping we set `timeout = 0` and `limit = 1` so that `get_updates()` // When stopping we set `timeout = 0` and `limit = 1` so that `get_updates()`
// set last seen update (offset) and return immediately // set last seen update (offset) and return immediately
(true, _) => (*this.offset, Some(1), Some(0)), (true, _) => {
log::trace!("graceful shutdown `get_updates` call");
(*this.offset, Some(1), Some(0))
}
// Drop pending updates // Drop pending updates
(_, true) => (-1, Some(1), Some(0)), (_, true) => (-1, Some(1), Some(0)),
}; };