2021-10-19 23:52:19 +03:00
|
|
|
//! Reverse proxy listening in "localhost:4000" will proxy all requests to "localhost:3000"
|
|
|
|
|
//! endpoint.
|
|
|
|
|
//!
|
|
|
|
|
//! Run with
|
|
|
|
|
//!
|
|
|
|
|
//! ```not_rust
|
2022-04-29 18:53:41 +02:00
|
|
|
//! cd examples && cargo run -p example-reverse-proxy
|
2021-10-19 23:52:19 +03:00
|
|
|
//! ```
|
|
|
|
|
|
|
|
|
|
use axum::{
|
|
|
|
|
extract::Extension,
|
|
|
|
|
http::{uri::Uri, Request, Response},
|
2021-10-24 22:05:16 +02:00
|
|
|
routing::get,
|
2022-03-01 00:39:22 +01:00
|
|
|
Router,
|
2021-10-19 23:52:19 +03:00
|
|
|
};
|
|
|
|
|
use hyper::{client::HttpConnector, Body};
|
2022-06-17 20:36:49 +02:00
|
|
|
use std::net::SocketAddr;
|
2021-10-19 23:52:19 +03:00
|
|
|
|
|
|
|
|
type Client = hyper::client::Client<HttpConnector, Body>;
|
|
|
|
|
|
|
|
|
|
#[tokio::main]
|
|
|
|
|
async fn main() {
|
|
|
|
|
tokio::spawn(server());
|
|
|
|
|
|
|
|
|
|
let client = Client::new();
|
|
|
|
|
|
|
|
|
|
let app = Router::new()
|
|
|
|
|
.route("/", get(handler))
|
2022-03-01 00:39:22 +01:00
|
|
|
.layer(Extension(client));
|
2021-10-19 23:52:19 +03:00
|
|
|
|
|
|
|
|
let addr = SocketAddr::from(([127, 0, 0, 1], 4000));
|
|
|
|
|
println!("reverse proxy listening on {}", addr);
|
|
|
|
|
axum::Server::bind(&addr)
|
|
|
|
|
.serve(app.into_make_service())
|
|
|
|
|
.await
|
|
|
|
|
.unwrap();
|
|
|
|
|
}
|
|
|
|
|
|
2022-03-03 22:33:33 +01:00
|
|
|
async fn handler(
|
|
|
|
|
Extension(client): Extension<Client>,
|
|
|
|
|
// NOTE: Make sure to put the request extractor last because once the request
|
|
|
|
|
// is extracted, extensions can't be extracted anymore.
|
|
|
|
|
mut req: Request<Body>,
|
|
|
|
|
) -> Response<Body> {
|
2021-10-19 23:52:19 +03:00
|
|
|
let path = req.uri().path();
|
|
|
|
|
let path_query = req
|
|
|
|
|
.uri()
|
|
|
|
|
.path_and_query()
|
|
|
|
|
.map(|v| v.as_str())
|
|
|
|
|
.unwrap_or(path);
|
|
|
|
|
|
|
|
|
|
let uri = format!("http://127.0.0.1:3000{}", path_query);
|
|
|
|
|
|
|
|
|
|
*req.uri_mut() = Uri::try_from(uri).unwrap();
|
|
|
|
|
|
|
|
|
|
client.request(req).await.unwrap()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async fn server() {
|
|
|
|
|
let app = Router::new().route("/", get(|| async { "Hello, world!" }));
|
|
|
|
|
|
|
|
|
|
let addr = SocketAddr::from(([127, 0, 0, 1], 3000));
|
|
|
|
|
println!("server listening on {}", addr);
|
|
|
|
|
axum::Server::bind(&addr)
|
|
|
|
|
.serve(app.into_make_service())
|
|
|
|
|
.await
|
|
|
|
|
.unwrap();
|
|
|
|
|
}
|
