8ba9a8d2c4
Adds utility functions to `hyper::body` to help asynchronously collecting all the buffers of some `HttpBody` into one. - `aggregate` will collect all into an `impl Buf` without copying the contents. This is ideal if you don't need a contiguous buffer. - `to_bytes` will copy all the data into a single contiguous `Bytes` buffer.
54 lines
1.4 KiB
Rust
54 lines
1.4 KiB
Rust
#![deny(warnings)]
|
|
#![warn(rust_2018_idioms)]
|
|
use std::env;
|
|
|
|
use hyper::{body::HttpBody as _, Client};
|
|
use tokio::io::{self, AsyncWriteExt as _};
|
|
|
|
// A simple type alias so as to DRY.
|
|
type Result<T> = std::result::Result<T, Box<dyn std::error::Error + Send + Sync>>;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
pretty_env_logger::init();
|
|
|
|
// Some simple CLI args requirements...
|
|
let url = match env::args().nth(1) {
|
|
Some(url) => url,
|
|
None => {
|
|
println!("Usage: client <url>");
|
|
return Ok(());
|
|
}
|
|
};
|
|
|
|
// HTTPS requires picking a TLS implementation, so give a better
|
|
// warning if the user tries to request an 'https' URL.
|
|
let url = url.parse::<hyper::Uri>().unwrap();
|
|
if url.scheme_str() != Some("http") {
|
|
println!("This example only works with 'http' URLs.");
|
|
return Ok(());
|
|
}
|
|
|
|
fetch_url(url).await
|
|
}
|
|
|
|
async fn fetch_url(url: hyper::Uri) -> Result<()> {
|
|
let client = Client::new();
|
|
|
|
let mut res = client.get(url).await?;
|
|
|
|
println!("Response: {}", res.status());
|
|
println!("Headers: {:#?}\n", res.headers());
|
|
|
|
// Stream the body, writing each chunk to stdout as we get it
|
|
// (instead of buffering and printing at the end).
|
|
while let Some(next) = res.body_mut().data().await {
|
|
let chunk = next?;
|
|
io::stdout().write_all(&chunk).await?;
|
|
}
|
|
|
|
println!("\n\nDone!");
|
|
|
|
Ok(())
|
|
}
|