use hyper::{ header::{self, HeaderValue}, Body, Request, Response, Server, }; // Import the routerify prelude traits. use routerify::prelude::*; use routerify::{Middleware, RequestInfo, Router, RouterService}; use std::io; use std::net::SocketAddr; // A handler for "/" page. async fn home_handler(_: Request) -> Result, io::Error> { Ok(Response::new(Body::from("Home page"))) } // A handler for "/about" page. async fn about_handler(_: Request) -> Result, io::Error> { Ok(Response::new(Body::from("About page"))) } // Define a pre middleware handler which will be executed on every request and // logs some meta. async fn logger_middleware(req: Request) -> Result, io::Error> { println!("{} {} {}", req.remote_addr(), req.method(), req.uri().path()); Ok(req) } // Define a post middleware handler which will be executed on every request and // adds a header to the response. async fn my_custom_header_adder_middleware(mut res: Response) -> Result, io::Error> { res.headers_mut() .insert("x-custom-header", HeaderValue::from_static("some value")); Ok(res) } // Define a post middleware handler which will be executed on every request and // accesses request information and adds the session cookies to manage session. async fn my_session_middleware(mut res: Response, req_info: RequestInfo) -> Result, io::Error> { // Access a cookie. let cookie = req_info .headers() .get(header::COOKIE) .and_then(|v| v.to_str().ok()) .unwrap_or(""); res.headers_mut() .insert(header::SET_COOKIE, HeaderValue::from_str(cookie).unwrap()); Ok(res) } fn router() -> Router { // Create a router and specify the the handlers. Router::builder() // Create a pre middleware using `Middleware::pre()` method // and attach it to the router. .middleware(Middleware::pre(logger_middleware)) // Create a post middleware using `Middleware::post()` method // and attach it to the router. .middleware(Middleware::post(my_custom_header_adder_middleware)) // Create a post middleware which will require request info using `Middleware::post_with_info()` method // and attach it to the router. .middleware(Middleware::post_with_info(my_session_middleware)) .get("/", home_handler) .get("/about", about_handler) .build() .unwrap() } #[tokio::main] async fn main() { let router = router(); // Create a Service from the router above to handle incoming requests. let service = RouterService::new(router).unwrap(); // The address on which the server will be listening. let addr = SocketAddr::from(([127, 0, 0, 1], 3001)); // Create a server by passing the created service to `.serve` method. let server = Server::bind(&addr).serve(service); println!("App is running on: {}", addr); if let Err(err) = server.await { eprintln!("Server error: {}", err); } }