fluencelabs/rust-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/rust-libp2p synced 2025-06-23 14:51:34 +00:00
Code Issues Projects Releases Wiki Activity

Fix Transport::and_then (#690)

Browse Source
This commit is contained in:
Pierre Krieger
2018-11-27 19:13:34 +01:00
committed by GitHub
parent e12795d1e6
commit 6cde480c29
1 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
core/src/transport/and_then.rs
View File

@ -149,16 +149,18 @@ where
type Error = T::Error; type Error = T::Error;
fn poll(&mut self) -> Poll<Self::Item, Self::Error> { fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
let future = match self.inner { loop {
Either::A(ref mut future) => { let future = match self.inner {
let item = try_ready!(future.poll()); Either::A(ref mut future) => {
let (f, a) = self.args.take().expect("AndThenFuture has already finished."); let item = try_ready!(future.poll());
f(item, a).into_future() let (f, a) = self.args.take().expect("AndThenFuture has already finished.");
} f(item, a).into_future()
Either::B(ref mut future) => return future.poll() }
}; Either::B(ref mut future) => return future.poll()
self.inner = Either::B(future); };
Ok(Async::NotReady)
self.inner = Either::B(future);
}
} }
} }