From d8eaa32485277d049342b93d9ec5f892d0a2b342 Mon Sep 17 00:00:00 2001
From: Will <LMJW@users.noreply.github.com>
Date: Tue, 25 May 2021 00:31:45 +1000
Subject: [PATCH] Update example code to use write_all_buf (#408)

---
 aws/sdk/examples/polly-synthesize-speech/src/main.rs | 10 ++++------
 rust-runtime/smithy-http/Cargo.toml                  |  2 +-
 rust-runtime/smithy-http/src/byte_stream.rs          |  5 +----
 3 files changed, 6 insertions(+), 11 deletions(-)

diff --git a/aws/sdk/examples/polly-synthesize-speech/src/main.rs b/aws/sdk/examples/polly-synthesize-speech/src/main.rs
index 1fa8064113..6c8de62a0a 100644
--- a/aws/sdk/examples/polly-synthesize-speech/src/main.rs
+++ b/aws/sdk/examples/polly-synthesize-speech/src/main.rs
@@ -10,7 +10,6 @@ use polly::{Client, Config, Region};
 
 use aws_types::region::{EnvironmentProvider, ProvideRegion};
 
-use bytes::Buf;
 use structopt::StructOpt;
 use tokio::io::AsyncWriteExt;
 use tracing_subscriber::fmt::format::FmtSpan;
@@ -90,9 +89,8 @@ async fn main() {
     let mut file = tokio::fs::File::create(out_file)
         .await
         .expect("failed to create file");
-    while blob.has_remaining() {
-        file.write_buf(&mut blob)
-            .await
-            .expect("failed to write to file");
-    }
+
+    file.write_all_buf(&mut blob)
+        .await
+        .expect("failed to write to file");
 }
diff --git a/rust-runtime/smithy-http/Cargo.toml b/rust-runtime/smithy-http/Cargo.toml
index f9e03af6e3..a18b8672b4 100644
--- a/rust-runtime/smithy-http/Cargo.toml
+++ b/rust-runtime/smithy-http/Cargo.toml
@@ -25,5 +25,5 @@ futures-core = "0.3.14"
 [dev-dependencies]
 proptest = "1"
 base64 = "0.13.0"
-tokio = { version = "1", features = ["macros", "rt", "fs", "io-util"]}
+tokio = {version = "1.6", features = ["macros", "rt", "fs", "io-util"]}
 tokio-stream = "0.1.5"
diff --git a/rust-runtime/smithy-http/src/byte_stream.rs b/rust-runtime/smithy-http/src/byte_stream.rs
index b26527ec91..354ab254b4 100644
--- a/rust-runtime/smithy-http/src/byte_stream.rs
+++ b/rust-runtime/smithy-http/src/byte_stream.rs
@@ -11,7 +11,6 @@
 //!
 //! ### Writing a ByteStream into a file:
 //! ```rust
-//! use bytes::Buf;
 //! use smithy_http::byte_stream::ByteStream;
 //! use std::error::Error;
 //! use tokio::fs::File;
@@ -25,9 +24,7 @@
 //! ) -> Result<(), Box<dyn Error + Send + Sync>> {
 //!     let mut buf = output.audio_stream.collect().await?;
 //!     let mut file = File::open("audio.mp3").await?;
-//!     while buf.has_remaining() {
-//!         file.write_buf(&mut buf).await?;
-//!     }
+//!     file.write_all_buf(&mut buf).await?;
 //!     Ok(())
 //! }
 //! ```