Unverified Commit d8eaa324 authored by Will's avatar Will Committed by GitHub
Browse files

Update example code to use write_all_buf (#408)

parent b5a2e4ca
Loading
Loading
Loading
Loading
+4 −6
Original line number Diff line number Diff line
@@ -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)

    file.write_all_buf(&mut blob)
        .await
        .expect("failed to write to file");
}
}
+1 −1
Original line number Diff line number Diff line
@@ -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"
+1 −4
Original line number Diff line number Diff line
@@ -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(())
//! }
//! ```