Update lib to std-future

src/lib.rs

@@ -81,6 +81,7 @@
 #![doc(html_root_url = "https://docs.rs/h2/0.1.25")]
 #![deny(missing_debug_implementations, missing_docs)]
 #![cfg_attr(test, deny(warnings))]
+#![feature(async_await)]
 
 macro_rules! proto_err {
     (conn: $($msg:tt)+) => {
@@ -91,9 +92,9 @@ macro_rules! proto_err {
     };
 }
 
-mod error;
 #[cfg_attr(feature = "unstable", allow(missing_docs))]
 mod codec;
+mod error;
 mod hpack;
 mod proto;
 
@@ -109,7 +110,48 @@ pub mod server;
 mod share;
 
 pub use crate::error::{Error, Reason};
-pub use crate::share::{SendStream, StreamId, RecvStream, ReleaseCapacity, PingPong, Ping, Pong};
+pub use crate::share::{Ping, PingPong, Pong, RecvStream, ReleaseCapacity, SendStream, StreamId};
 
 #[cfg(feature = "unstable")]
 pub use codec::{Codec, RecvError, SendError, UserError};
+
+use std::task::Poll;
+
+// TODO: Get rid of this trait once https://github.com/rust-lang/rust/pull/63512
+// is stablized.
+trait PollExt<T, E> {
+    /// Changes the success value of this `Poll` with the closure provided.
+    fn map_ok_<U, F>(self, f: F) -> Poll<Option<Result<U, E>>>
+    where
+        F: FnOnce(T) -> U;
+    /// Changes the error value of this `Poll` with the closure provided.
+    fn map_err_<U, F>(self, f: F) -> Poll<Option<Result<T, U>>>
+    where
+        F: FnOnce(E) -> U;
+}
+
+impl<T, E> PollExt<T, E> for Poll<Option<Result<T, E>>> {
+    fn map_ok_<U, F>(self, f: F) -> Poll<Option<Result<U, E>>>
+    where
+        F: FnOnce(T) -> U,
+    {
+        match self {
+            Poll::Ready(Some(Ok(t))) => Poll::Ready(Some(Ok(f(t)))),
+            Poll::Ready(Some(Err(e))) => Poll::Ready(Some(Err(e))),
+            Poll::Ready(None) => Poll::Ready(None),
+            Poll::Pending => Poll::Pending,
+        }
+    }
+
+    fn map_err_<U, F>(self, f: F) -> Poll<Option<Result<T, U>>>
+    where
+        F: FnOnce(E) -> U,
+    {
+        match self {
+            Poll::Ready(Some(Ok(t))) => Poll::Ready(Some(Ok(t))),
+            Poll::Ready(Some(Err(e))) => Poll::Ready(Some(Err(f(e)))),
+            Poll::Ready(None) => Poll::Ready(None),
+            Poll::Pending => Poll::Pending,
+        }
+    }
+}