Simplify Streamer
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
asonix 2023-08-23 19:12:59 -05:00
parent 95637fdfe5
commit b28cc38c38

View file

@ -139,12 +139,9 @@ impl<S> Streamer<S> {
where
S: Stream + Unpin,
{
let opt = match self.0 {
Some(ref mut stream) => {
std::future::poll_fn(|cx| Pin::new(&mut *stream).poll_next(cx)).await
}
None => None,
};
let stream = self.0.as_mut().take()?;
let opt = std::future::poll_fn(|cx| Pin::new(&mut *stream).poll_next(cx)).await;
if opt.is_none() {
self.0.take();