A simple reverse proxy for use with Hyper and Tokio
6e9aa4501f
* chore: upgrade hyper/tokio libraries * updates tokio reactor version * doc: updates documentation in README and rust code doc |
||
---|---|---|
src | ||
.gitignore | ||
.travis.yml | ||
Cargo.toml | ||
LICENSE | ||
README.md |
hyper-reverse-proxy
A simple reverse proxy, to be used with Hyper.
The implementation ensures that Hop-by-hop headers are stripped correctly in both directions,
and adds the client's IP address to a comma-space-separated list of forwarding addresses in the
X-Forwarded-For
header.
The implementation is based on Go's httputil.ReverseProxy
.
Example
Add these dependencies to your Cargo.toml
file.
[dependencies]
hyper-reverse-proxy = "0.5"
hyper = { version = "0.14", features = ["full"] }
tokio = { version = "1", features = ["full"] }
The following example will set up a reverse proxy listening on 127.0.0.1:13900
,
and will proxy these calls:
"/target/first"
will be proxied tohttp://127.0.0.1:13901
"/target/second"
will be proxied tohttp://127.0.0.1:13902
- All other URLs will be handled by
debug_request
function, that will display request information.
use hyper::server::conn::AddrStream;
use hyper::{Body, Request, Response, Server};
use hyper::service::{service_fn, make_service_fn};
use futures::future::{self, Future};
type BoxFut = Box<Future<Item=Response<Body>, Error=hyper::Error> + Send>;
fn debug_request(req: Request<Body>) -> BoxFut {
let body_str = format!("{:?}", req);
let response = Response::new(Body::from(body_str));
Box::new(future::ok(response))
}
#[tokio::main]
fn main() {
// This is our socket address...
let addr = ([127, 0, 0, 1], 13900).into();
// A `Service` is needed for every connection.
let make_svc = make_service_fn(|socket: &AddrStream| {
let remote_addr = socket.remote_addr();
service_fn(move |req: Request<Body>| { // returns BoxFut
if req.uri().path().starts_with("/target/first") {
// will forward requests to port 13901
return hyper_reverse_proxy::call(remote_addr.ip(), "http://127.0.0.1:13901", req)
} else if req.uri().path().starts_with("/target/second") {
// will forward requests to port 13902
return hyper_reverse_proxy::call(remote_addr.ip(), "http://127.0.0.1:13902", req)
} else {
debug_request(req)
}
})
});
let server = Server::bind(&addr).serve(make_svc);
// Run this server for... forever!
println!("Running server on {:?}", addr);
if let Err(e) = server.await {
eprintln!("server error: {}", e);
}
}