Update example code to use write_all_buf (#408)

This commit is contained in:
Will 2021-05-25 00:31:45 +10:00 committed by GitHub
parent b5a2e4ca8c
commit d8eaa32485
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 11 deletions

View File

@ -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");
}

View File

@ -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"

View File

@ -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(())
//! }
//! ```