bdbaf80073It is not uncommon that users who are new to tracing look at the examples in the `master` branch of the repository and find that they don't compile. This is because they are examples which compile with the code from the master branch, which is for the as yet unreleased tracing 0.2.0 ecosystem. Users should instead go to the `v0.1.x` branch to find examples compatible with the crates published on crates.io. This change adds a doc-comment to the beginning of every example file informing the user of this fact and suggesting that they check out the `v0.1.x` branch instead.
110 lines
3.7 KiB
Rust
110 lines
3.7 KiB
Rust
//! NOTE: This is pre-release documentation for the upcoming tracing 0.2.0 ecosystem. For the
|
|
//! release examples, please see the `v0.1.x` branch instead.
|
|
#![deny(rust_2018_idioms)]
|
|
|
|
use http::{Method, Request, Response, StatusCode};
|
|
use hyper::{
|
|
service::{make_service_fn, service_fn},
|
|
Body, Server,
|
|
};
|
|
use std::str;
|
|
use tracing::{debug, info, span, Instrument as _, Level};
|
|
|
|
async fn echo(req: Request<Body>) -> Result<Response<Body>, hyper::Error> {
|
|
let span = span!(
|
|
Level::INFO,
|
|
"request",
|
|
method = ?req.method(),
|
|
uri = ?req.uri(),
|
|
headers = ?req.headers()
|
|
);
|
|
async move {
|
|
info!("received request");
|
|
let mut response = Response::new(Body::empty());
|
|
|
|
match (req.method(), req.uri().path()) {
|
|
// Serve some instructions at /
|
|
(&Method::GET, "/") => {
|
|
const BODY: &str = "Try POSTing data to /echo";
|
|
*response.body_mut() = Body::from(BODY);
|
|
info!(body = %(&BODY), "response",);
|
|
Ok(response)
|
|
}
|
|
|
|
// Simply echo the body back to the client.
|
|
(&Method::POST, "/echo") => {
|
|
info!(response_kind = %"echo", "response");
|
|
*response.body_mut() = req.into_body();
|
|
Ok(response)
|
|
}
|
|
|
|
// Convert to uppercase before sending back to client.
|
|
(&Method::POST, "/echo/uppercase") => {
|
|
let body = hyper::body::to_bytes(req).await?;
|
|
let upper = body
|
|
.iter()
|
|
.map(|byte| byte.to_ascii_uppercase())
|
|
.collect::<Vec<u8>>();
|
|
debug!(
|
|
body = ?str::from_utf8(&body[..]),
|
|
uppercased = ?str::from_utf8(&upper[..]),
|
|
"uppercased request body"
|
|
);
|
|
|
|
info!(response_kind = %"uppercase", "response");
|
|
*response.body_mut() = Body::from(upper);
|
|
Ok(response)
|
|
}
|
|
|
|
// Reverse the entire body before sending back to the client.
|
|
(&Method::POST, "/echo/reversed") => {
|
|
async move {
|
|
let body = hyper::body::to_bytes(req).await?;
|
|
let reversed = body.iter().rev().cloned().collect::<Vec<u8>>();
|
|
debug!(
|
|
body = ?str::from_utf8(&body[..]),
|
|
"reversed request body"
|
|
);
|
|
*response.body_mut() = Body::from(reversed);
|
|
info!(body = ?(&response.body()), "response");
|
|
Ok(response)
|
|
}
|
|
.instrument(span!(Level::TRACE, "response", response_kind = %"reversed"))
|
|
.await
|
|
}
|
|
|
|
// The 404 Not Found route...
|
|
_ => {
|
|
*response.status_mut() = StatusCode::NOT_FOUND;
|
|
info!(
|
|
body = ?(),
|
|
status = ?StatusCode::NOT_FOUND,
|
|
"response",
|
|
);
|
|
Ok(response)
|
|
}
|
|
}
|
|
}
|
|
.instrument(span)
|
|
.await
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
tracing_subscriber::fmt()
|
|
.with_max_level(Level::TRACE)
|
|
.init();
|
|
|
|
let local_addr: std::net::SocketAddr = ([127, 0, 0, 1], 3000).into();
|
|
let server_span = span!(Level::TRACE, "server", %local_addr);
|
|
|
|
let service = make_service_fn(|_| async { Ok::<_, hyper::Error>(service_fn(echo)) });
|
|
let server = Server::bind(&local_addr)
|
|
.serve(service)
|
|
.instrument(server_span.clone());
|
|
|
|
info!(parent: &server_span, "listening...");
|
|
server.await?;
|
|
|
|
Ok(())
|
|
}
|