#![deny(warnings)] use std::net::SocketAddr; use bytes::Buf; use hyper::server::conn::Http; use hyper::service::service_fn; use hyper::{header, Body, Method, Request, Response, StatusCode}; use tokio::net::{TcpListener, TcpStream}; type GenericError = Box; type Result = std::result::Result; static INDEX: &[u8] = b"test.html"; static INTERNAL_SERVER_ERROR: &[u8] = b"Internal Server Error"; static NOTFOUND: &[u8] = b"Not Found"; static POST_DATA: &str = r#"{"original": "data"}"#; static URL: &str = "http://127.0.0.1:1337/json_api"; async fn client_request_response() -> Result> { let req = Request::builder() .method(Method::POST) .uri(URL) .header(header::CONTENT_TYPE, "application/json") .body(POST_DATA.into()) .unwrap(); let host = req.uri().host().expect("uri has no host"); let port = req.uri().port_u16().expect("uri has no port"); let stream = TcpStream::connect(format!("{}:{}", host, port)).await?; let (mut sender, conn) = hyper::client::conn::handshake(stream).await?; tokio::task::spawn(async move { if let Err(err) = conn.await { println!("Connection error: {:?}", err); } }); let web_res = sender.send_request(req).await?; let res_body = web_res.into_body(); Ok(Response::new(res_body)) } async fn api_post_response(req: Request) -> Result> { // Aggregate the body... let whole_body = hyper::body::aggregate(req).await?; // Decode as JSON... let mut data: serde_json::Value = serde_json::from_reader(whole_body.reader())?; // Change the JSON... data["test"] = serde_json::Value::from("test_value"); // And respond with the new JSON. let json = serde_json::to_string(&data)?; let response = Response::builder() .status(StatusCode::OK) .header(header::CONTENT_TYPE, "application/json") .body(Body::from(json))?; Ok(response) } async fn api_get_response() -> Result> { let data = vec!["foo", "bar"]; let res = match serde_json::to_string(&data) { Ok(json) => Response::builder() .header(header::CONTENT_TYPE, "application/json") .body(Body::from(json)) .unwrap(), Err(_) => Response::builder() .status(StatusCode::INTERNAL_SERVER_ERROR) .body(INTERNAL_SERVER_ERROR.into()) .unwrap(), }; Ok(res) } async fn response_examples(req: Request) -> Result> { match (req.method(), req.uri().path()) { (&Method::GET, "/") | (&Method::GET, "/index.html") => Ok(Response::new(INDEX.into())), (&Method::GET, "/test.html") => client_request_response().await, (&Method::POST, "/json_api") => api_post_response(req).await, (&Method::GET, "/json_api") => api_get_response().await, _ => { // Return 404 not found response. Ok(Response::builder() .status(StatusCode::NOT_FOUND) .body(NOTFOUND.into()) .unwrap()) } } } #[tokio::main] async fn main() -> Result<()> { pretty_env_logger::init(); let addr: SocketAddr = "127.0.0.1:1337".parse().unwrap(); let listener = TcpListener::bind(&addr).await?; println!("Listening on http://{}", addr); loop { let (stream, _) = listener.accept().await?; tokio::task::spawn(async move { let service = service_fn(move |req| response_examples(req)); if let Err(err) = Http::new().serve_connection(stream, service).await { println!("Failed to serve connection: {:?}", err); } }); } }