tokio-rs/axum
1
0
Fork 0
mirror of https://github.com/tokio-rs/axum.git synced 2026-09-09 00:00:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add reverse-proxy example (#389)

Browse Source
This commit is contained in:
Eray Karatay
2021-10-19 22:52:19 +02:00
committed by GitHub
parent e71ab44bd5
commit a724af3d0a
2 changed files with 74 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/reverse-proxy/Cargo.toml
+11
View File
@@ -0,0 +1,11 @@
[package]
name = "example-reverse-proxy"
version = "0.1.0"
edition = "2018"
[dependencies]
axum = { path = "../.." }
hyper = { version = "0.14", features = ["full"] }
tokio = { version = "1", features = ["full"] }
tracing = "0.1"
tracing-subscriber = "0.2"
examples/reverse-proxy/src/main.rs
+63
View File
@@ -0,0 +1,63 @@
//! Reverse proxy listening in "localhost:4000" will proxy all requests to "localhost:3000"
//! endpoint.
//!
//! Run with
//!
//! ```not_rust
//! cargo run -p example-reverse-proxy
//! ```
use axum::{
extract::Extension,
handler::get,
http::{uri::Uri, Request, Response},
AddExtensionLayer, Router,
};
use hyper::{client::HttpConnector, Body};
use std::{convert::TryFrom, net::SocketAddr};
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))
.layer(AddExtensionLayer::new(client));
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();
}
async fn handler(Extension(client): Extension<Client>, mut req: Request<Body>) -> Response<Body> {
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();
}