docs(server): update `with_graceful_shutdown doc test

Signed-off-by: Weihang Lo <me@weihanglo.tw>

src/server/mod.rs

@@ -164,34 +164,32 @@ where
     /// # Example
     ///
     /// ```
-    /// # extern crate hyper;
+    /// # #![feature(async_await)]
-    /// # extern crate futures;
-    /// # use futures::Future;
     /// # fn main() {}
     /// # #[cfg(feature = "runtime")]
-    /// # fn run() {
+    /// # async fn run() {
-    /// # use hyper::{Body, Response, Server};
+    /// # use hyper::{Body, Response, Server, Error};
-    /// # use hyper::service::service_fn_ok;
+    /// # use hyper::service::{make_service_fn, service_fn};
-    /// # let new_service = || {
+    /// # let make_service = make_service_fn(|_| async {
-    /// #     service_fn_ok(|_req| {
+    /// #     Ok::<_, Error>(service_fn(|_req| async {
-    /// #         Response::new(Body::from("Hello World"))
+    /// #         Ok::<_, Error>(Response::new(Body::from("Hello World")))
-    /// #     })
+    /// #     }))
-    /// # };
+    /// # });
-    ///
     /// // Make a server from the previous examples...
     /// let server = Server::bind(&([127, 0, 0, 1], 3000).into())
-    ///     .serve(new_service);
+    ///     .serve(make_service);
-    ///
-    /// // Prepare some signal for when the server should start
-    /// // shutting down...
-    /// let (tx, rx) = futures::sync::oneshot::channel::<()>();
     ///
+    /// // Prepare some signal for when the server should start shutting down...
+    /// let (tx, rx) = tokio::sync::oneshot::channel::<()>();
     /// let graceful = server
-    ///     .with_graceful_shutdown(rx)
+    ///     .with_graceful_shutdown(async {
-    ///     .map_err(|err| eprintln!("server error: {}", err));
+    ///         rx.await.ok();
+    ///     });
     ///
-    /// // Spawn `server` onto an Executor...
+    /// // Await the `server` receiving the signal...
-    /// hyper::rt::spawn(graceful);
+    /// if let Err(e) = graceful.await {
+    ///     eprintln!("server error: {}", e);
+    /// }
     ///
     /// // And later, trigger the signal by calling `tx.send(())`.
     /// let _ = tx.send(());