How to print more log? #563
-
pub async fn list_dataset(Json(req): Json<ListDatasetRequest>) -> Result<Json<ListDatasetResponse>, StatusCode>I have written a handler, and I run the server on my Macbook. It works. However, when I compose it with nginx, it doesn't work. How to print more log? I want to know what request body is before deserializing.
|
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 6 replies
-
|
Try this https://github.com/tokio-rs/axum/blob/main/examples/print-request-response/src/main.rs |
Beta Was this translation helpful? Give feedback.
-
|
I copy the code and some errors occurs. error[E0277]: the trait bound `impl Future: IntoResponse` is not satisfied
--> src/bin/server.rs:27:10
|
27 | .layer(
| ^^^^^ the trait `IntoResponse` is not implemented for `impl Future`
|
= note: required because of the requirements on the impl of `Service<Request<hyper::Body>>` for `HandleError<AndThen<AsyncFilter<Route, fn(Request<hyper::Body>) -> impl Future {map_request}>, fn(Response<http_body::combinators::box_body::UnsyncBoxBody<axum::body::Bytes, axum::Error>>) -> impl Future {map_response}>, [closure@src/bin/server.rs:29:46: 34:18], hyper::Body>`
For more information about this error, try `rustc --explain E0277`.
error: could not compile `lab_mt_ops` due to previous error
|
Beta Was this translation helpful? Give feedback.
-
|
Here is my code. https://github.com/lz1998/atest Cargo.toml [package]
name = "atest"
version = "0.1.0"
edition = "2018"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]
axum = "0.3.4"
tokio = { version = "1.0", features = ["full"] }
tracing = "0.1"
tracing-subscriber = { version="0.3", features = ["env-filter"] }
tower = { version = "0.4", features = ["util", "filter"] }
hyper = { version = "0.14", features = ["full"] }src/main.rs //! Run with
//!
//! ```not_rust
//! cargo run -p example-print-request-response
//! ```
use axum::{
body::{Body, BoxBody, Bytes},
error_handling::HandleErrorLayer,
http::{Request, Response, StatusCode},
routing::post,
Router,
};
use std::net::SocketAddr;
use tower::{filter::AsyncFilterLayer, util::AndThenLayer, BoxError, ServiceBuilder};
#[tokio::main]
async fn main() {
// Set the RUST_LOG, if it hasn't been explicitly defined
if std::env::var_os("RUST_LOG").is_none() {
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 /`" }))
.layer(
ServiceBuilder::new()
.layer(HandleErrorLayer::new(|error| async move {
(
StatusCode::INTERNAL_SERVER_ERROR,
format!("Unhandled internal error: {}", error),
)
}))
.layer(AndThenLayer::new(map_response))
.layer(AsyncFilterLayer::new(map_request)),
);
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)
} |
Beta Was this translation helpful? Give feedback.
-
|
Ah sorry my bad. We've made some changes on |
Beta Was this translation helpful? Give feedback.
-
If there are lots of layer, main.rs will be too large, so I want to define pub fn new_debug_layer() -> ServiceBuilder<?> {
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))
}
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)
}
|
Beta Was this translation helpful? Give feedback.
-
|
Are we on to a different problem now? Did you manage to solve the previous thing? About what you asked. It's possible but the generics make it fairly difficult. Honestly I wouldn't bother extracting layers like this to functions. You could try |
Beta Was this translation helpful? Give feedback.
-
|
@davidpdrsn do you think axum could expose or expose directly an axum function, something like |
Beta Was this translation helpful? Give feedback.
-
|
Nah not sure we should do that. I'd like to avoid re-exports unless they're very common, which I don't think this is. Running |
Beta Was this translation helpful? Give feedback.
Try this https://github.com/tokio-rs/axum/blob/main/examples/print-request-response/src/main.rs