mirror of
https://github.com/actix/examples
synced 2025-03-15 01:06:25 +01:00
91 lines
2.6 KiB
Rust
91 lines
2.6 KiB
Rust
use std::net::ToSocketAddrs;
|
|
|
|
use actix_web::{
|
|
error, middleware, web, App, Error, HttpRequest, HttpResponse, HttpServer,
|
|
};
|
|
use awc::Client;
|
|
use clap::StructOpt;
|
|
use url::Url;
|
|
|
|
async fn forward(
|
|
req: HttpRequest,
|
|
payload: web::Payload,
|
|
url: web::Data<Url>,
|
|
client: web::Data<Client>,
|
|
) -> Result<HttpResponse, Error> {
|
|
let mut new_url = url.get_ref().clone();
|
|
new_url.set_path(req.uri().path());
|
|
new_url.set_query(req.uri().query());
|
|
|
|
// TODO: This forwarded implementation is incomplete as it only handles the unofficial
|
|
// X-Forwarded-For header but not the official Forwarded one.
|
|
let forwarded_req = client
|
|
.request_from(new_url.as_str(), req.head())
|
|
.no_decompress();
|
|
let forwarded_req = match req.head().peer_addr {
|
|
Some(addr) => {
|
|
forwarded_req.insert_header(("x-forwarded-for", format!("{}", addr.ip())))
|
|
}
|
|
None => forwarded_req,
|
|
};
|
|
|
|
let res = forwarded_req
|
|
.send_stream(payload)
|
|
.await
|
|
.map_err(error::ErrorInternalServerError)?;
|
|
|
|
let mut client_resp = HttpResponse::build(res.status());
|
|
// Remove `Connection` as per
|
|
// https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Connection#Directives
|
|
for (header_name, header_value) in
|
|
res.headers().iter().filter(|(h, _)| *h != "connection")
|
|
{
|
|
client_resp.insert_header((header_name.clone(), header_value.clone()));
|
|
}
|
|
|
|
Ok(client_resp.streaming(res))
|
|
}
|
|
|
|
#[derive(clap::Parser, Debug)]
|
|
struct CliArguments {
|
|
listen_addr: String,
|
|
listen_port: u16,
|
|
forward_addr: String,
|
|
forward_port: u16,
|
|
}
|
|
|
|
#[actix_web::main]
|
|
async fn main() -> std::io::Result<()> {
|
|
env_logger::init_from_env(env_logger::Env::new().default_filter_or("info"));
|
|
|
|
let args = CliArguments::parse();
|
|
|
|
let forward_socket_addr = (args.forward_addr, args.forward_port)
|
|
.to_socket_addrs()?
|
|
.next()
|
|
.expect("given forwarding address was not valid");
|
|
|
|
let forward_url = format!("http://{}", forward_socket_addr);
|
|
let forward_url = Url::parse(&forward_url).unwrap();
|
|
|
|
log::info!(
|
|
"starting HTTP server at http://{}:{}",
|
|
&args.listen_addr,
|
|
args.listen_port
|
|
);
|
|
|
|
log::info!("forwarding to {forward_url}");
|
|
|
|
HttpServer::new(move || {
|
|
App::new()
|
|
.app_data(web::Data::new(Client::default()))
|
|
.app_data(web::Data::new(forward_url.clone()))
|
|
.wrap(middleware::Logger::default())
|
|
.default_service(web::to(forward))
|
|
})
|
|
.bind((args.listen_addr, args.listen_port))?
|
|
.workers(2)
|
|
.run()
|
|
.await
|
|
}
|