book: fix example code

Author: djc

Cargo.lock

@@ -10,6 +10,7 @@ categories.workspace = true
 
 [dependencies]
 anyhow.workspace = true
+bytes = { workspace = true }
 quinn = { version = "0.11.7", path = "../../quinn" }
 rcgen.workspace = true
 rustls.workspace = true

docs/book/Cargo.toml

@@ -1,40 +1,41 @@
+use bytes::Bytes;
 use quinn::Connection;
 
 async fn open_bidirectional_stream(connection: Connection) -> anyhow::Result<()> {
-    let (mut send, recv) = connection.open_bi().await?;
+    let (mut send, mut recv) = connection.open_bi().await?;
     send.write_all(b"test").await?;
-    send.finish().await?;
+    send.finish()?;
     let received = recv.read_to_end(10).await?;
     Ok(())
 }
 
 async fn receive_bidirectional_stream(connection: Connection) -> anyhow::Result<()> {
-    while let Ok((mut send, recv)) = connection.accept_bi().await {
+    while let Ok((mut send, mut recv)) = connection.accept_bi().await {
         // Because it is a bidirectional stream, we can both send and receive.
         println!("request: {:?}", recv.read_to_end(50).await?);
         send.write_all(b"response").await?;
-        send.finish().await?;
+        send.finish()?;
     }
     Ok(())
 }
 
 async fn open_unidirectional_stream(connection: Connection) -> anyhow::Result<()> {
     let mut send = connection.open_uni().await?;
     send.write_all(b"test").await?;
-    send.finish().await?;
+    send.finish()?;
     Ok(())
 }
 
 async fn receive_unidirectional_stream(connection: Connection) -> anyhow::Result<()> {
-    while let Ok(recv) = connection.accept_uni().await {
+    while let Ok(mut recv) = connection.accept_uni().await {
         // Because it is a unidirectional stream, we can only receive not send back.
         println!("{:?}", recv.read_to_end(50).await?);
     }
     Ok(())
 }
 
 async fn send_unreliable(connection: Connection) -> anyhow::Result<()> {
-    connection.send_datagram(b"test".into()).await?;
+    connection.send_datagram(Bytes::from(&b"test"[..]))?;
     Ok(())
 }
 

docs/book/src/quinn/data-transfer.rs

@@ -28,7 +28,7 @@ async fn server(config: ServerConfig) -> Result<(), Box<dyn Error>> {
 
 async fn client() -> Result<(), Box<dyn Error>> {
     // Bind this endpoint to a UDP socket on the given client address.
-    let mut endpoint = Endpoint::client(client_addr());
+    let mut endpoint = Endpoint::client(client_addr())?;
 
     // Connect to the server passing in the server name which is supposed to be in the server certificate.
     let connection = endpoint.connect(server_addr(), SERVER_NAME)?.await?;