Update graceful shutdown example on type as well.

Author: bittrance

axum/src/json.rs

@@ -208,6 +208,19 @@ where
     }
 }
 
+impl<T> From<Json<T>> for Bytes
+where
+    T: Serialize,
+{
+    fn from(v: Json<T>) -> Bytes {
+        let mut buf = BytesMut::with_capacity(128).writer();
+        match serde_json::to_writer(&mut buf, &v.0) {
+            Ok(()) => buf.into_inner().freeze(),
+            Err(err) => todo!(""),
+        }
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;

axum/src/serve.rs

@@ -287,21 +287,22 @@ impl<M, S, F> WithGracefulShutdown<M, S, F> {
     /// # Example
     /// ```
     /// use axum::{Router, routing::get};
+    /// use std::{future::IntoFuture, sync::Arc};
+    /// use tokio::sync::oneshot;
     ///
     /// # async {
     /// let router = Router::new().route("/", get(|| async { "Hello, World!" }));
     ///
     /// let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await.unwrap();
-    /// axum::serve(listener, router)
-    ///     .with_graceful_shutdown(shutdown_signal())
-    ///     .tcp_nodelay(true)
-    ///     .await
-    ///     .unwrap();
+    /// let (tx, rx) = oneshot::channel();
+    /// tokio::spawn(
+    ///     axum::serve(listener, router)
+    ///         .with_graceful_shutdown(async move { rx.await.unwrap_or(()) })
+    ///         .into_future(),
+    /// );
+    /// // ...
+    /// tx.send(()).unwrap();
     /// # };
-    ///
-    /// async fn shutdown_signal() {
-    ///     // ...
-    /// }
     /// ```
     pub fn tcp_nodelay(self, nodelay: bool) -> Self {
         Self {