use hyper::{header, Body, Request, Response, Server, StatusCode};
use routerify::{RequestInfo, Router, RouterService};
use std::io;
use std::net::SocketAddr;
async fn home_handler(_: Request<Body>) -> Result<Response<Body>, io::Error> {
Err(io::Error::new(io::ErrorKind::Other, "Some errors"))
}
async fn about_handler(_: Request<Body>) -> Result<Response<Body>, io::Error> {
Ok(Response::new(Body::from("About page")))
}
async fn error_handler(err: routerify::RouteError, req_info: RequestInfo) -> Response<Body> {
eprintln!("{}", err);
let cookie = req_info.headers().get(header::COOKIE).unwrap().to_str().unwrap();
Response::builder()
.header(header::SET_COOKIE, cookie)
.status(StatusCode::INTERNAL_SERVER_ERROR)
.body(Body::from(format!("Something went wrong: {}", err)))
.unwrap()
}
fn router() -> Router<Body, io::Error> {
Router::builder()
.get("/", home_handler)
.get("/about", about_handler)
.err_handler_with_info(error_handler)
.build()
.unwrap()
}
#[tokio::main]
async fn main() {
let router = router();
let service = RouterService::new(router).unwrap();
let addr = SocketAddr::from(([127, 0, 0, 1], 3001));
let server = Server::bind(&addr).serve(service);
println!("App is running on: {}", addr);
if let Err(err) = server.await {
eprintln!("Server error: {}", err);
}
}