2021-08-20 09:26:31 +02:00
|
|
|
//! Run with
|
|
|
|
|
//!
|
|
|
|
|
//! ```not_rust
|
|
|
|
|
//! cargo run -p example-print-request-response
|
|
|
|
|
//! ```
|
|
|
|
|
|
|
|
|
|
use axum::{
|
|
|
|
|
body::{Body, BoxBody, Bytes},
|
2021-10-24 19:33:03 +02:00
|
|
|
error_handling::HandleErrorLayer,
|
|
|
|
|
http::{Request, Response, StatusCode},
|
2021-10-24 22:05:16 +02:00
|
|
|
routing::post,
|
2021-08-20 09:26:31 +02:00
|
|
|
Router,
|
|
|
|
|
};
|
|
|
|
|
use std::net::SocketAddr;
|
2021-10-24 19:33:03 +02:00
|
|
|
use tower::{filter::AsyncFilterLayer, util::AndThenLayer, BoxError, ServiceBuilder};
|
2021-08-20 09:26:31 +02:00
|
|
|
|
|
|
|
|
#[tokio::main]
|
|
|
|
|
async fn main() {
|
|
|
|
|
// Set the RUST_LOG, if it hasn't been explicitly defined
|
2021-09-12 18:39:43 +03:00
|
|
|
if std::env::var_os("RUST_LOG").is_none() {
|
2021-08-20 09:26:31 +02:00
|
|
|
std::env::set_var(
|
|
|
|
|
"RUST_LOG",
|
|
|
|
|
"example_print_request_response=debug,tower_http=debug",
|
|
|
|
|
)
|
|
|
|
|
}
|
|
|
|
|
tracing_subscriber::fmt::init();
|
|
|
|
|
|
|
|
|
|
let app = Router::new()
|
|
|
|
|
.route("/", post(|| async move { "Hello from `POST /`" }))
|
2021-10-24 19:33:03 +02:00
|
|
|
.layer(
|
|
|
|
|
ServiceBuilder::new()
|
|
|
|
|
.layer(HandleErrorLayer::new(|error| {
|
|
|
|
|
(
|
|
|
|
|
StatusCode::INTERNAL_SERVER_ERROR,
|
|
|
|
|
format!("Unhandled internal error: {}", error),
|
|
|
|
|
)
|
|
|
|
|
}))
|
|
|
|
|
.layer(AndThenLayer::new(map_response))
|
|
|
|
|
.layer(AsyncFilterLayer::new(map_request)),
|
|
|
|
|
);
|
2021-08-20 09:26:31 +02:00
|
|
|
|
|
|
|
|
let addr = SocketAddr::from(([127, 0, 0, 1], 3000));
|
|
|
|
|
tracing::debug!("listening on {}", addr);
|
|
|
|
|
axum::Server::bind(&addr)
|
|
|
|
|
.serve(app.into_make_service())
|
|
|
|
|
.await
|
|
|
|
|
.unwrap();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async fn map_request(req: Request<Body>) -> Result<Request<Body>, BoxError> {
|
|
|
|
|
let (parts, body) = req.into_parts();
|
|
|
|
|
let bytes = buffer_and_print("request", body).await?;
|
|
|
|
|
let req = Request::from_parts(parts, Body::from(bytes));
|
|
|
|
|
Ok(req)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async fn map_response(res: Response<BoxBody>) -> Result<Response<Body>, BoxError> {
|
|
|
|
|
let (parts, body) = res.into_parts();
|
|
|
|
|
let bytes = buffer_and_print("response", body).await?;
|
|
|
|
|
let res = Response::from_parts(parts, Body::from(bytes));
|
|
|
|
|
Ok(res)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async fn buffer_and_print<B>(direction: &str, body: B) -> Result<Bytes, BoxError>
|
|
|
|
|
where
|
|
|
|
|
B: axum::body::HttpBody<Data = Bytes>,
|
|
|
|
|
B::Error: Into<BoxError>,
|
|
|
|
|
{
|
|
|
|
|
let bytes = hyper::body::to_bytes(body).await.map_err(Into::into)?;
|
|
|
|
|
if let Ok(body) = std::str::from_utf8(&bytes) {
|
|
|
|
|
tracing::debug!("{} body = {:?}", direction, body);
|
|
|
|
|
}
|
|
|
|
|
Ok(bytes)
|
|
|
|
|
}
|
