snoyberg · December 29, 2020 16:33
diff --git a/main.rs b/main.rs
 use hyper::{Client, Server, Request, Response, Body};
 use anyhow::*;
 use std::net::SocketAddr;
 use hyper::service::{make_service_fn, service_fn};
 use std::sync::{Arc, RwLock};

 fn mutate_request(req: &mut Request<Body>) -> Result<()> {
    for key in &["content-length", "transfer-encoding", "accept-encoding", "content-encoding"] {
        req.headers_mut().remove(*key);
    }

    let uri = req.uri();
    let uri_string = match uri.query() {
        None => format!("https://www.snoyman.com{}", uri.path()),
        Some(query) => format!("https://www.snoyman.com{}?{}", uri.path(), query),
    };
    *req.uri_mut() = uri_string.parse().context("Parsing URI in mutate_request")?;
    Ok(())
 }

 #[derive(Debug)]
 struct Stats {
    proxied: usize,
 }

 #[tokio::main]
 async fn main() -> Result<()> {
    let https = hyper_rustls::HttpsConnector::with_native_roots();
    let client: Client<_, hyper::Body> = Client::builder().build(https);
    let client: Arc<Client<_, hyper::Body>> = Arc::new(client);
    let stats: Arc<RwLock<Stats>> = Arc::new(RwLock::new(Stats {
        proxied: 0,
    }));

    let addr = SocketAddr::from(([0, 0, 0, 0], 3000));

    let make_svc = make_service_fn(move |_| {
        let client = Arc::clone(&client);
        let stats = Arc::clone(&stats);
        async move {
            Ok::<_, Error>(service_fn(move |mut req| {
                let client = Arc::clone(&client);
                let stats = Arc::clone(&stats);
                async move {
                    if req.uri().path() == "/status" {
                        let stats: &Stats = &*stats.read().unwrap();
                        let body: Body = format!("{:?}", stats).into();
                        Ok(Response::new(body))
                    } else {
                        println!("Proxied: {}", req.uri().path());
                        stats.write().unwrap().proxied += 1;
                        mutate_request(&mut req)?;
                        client.request(req).await.context("Making request to backend server")
                    }
                }
            }))
        }
    });
    Server::bind(&addr).serve(make_svc).await.context("Running server")?;
    Ok::<(), anyhow::Error>(())
 }
	use hyper::{Client, Server, Request, Response, Body};
	use anyhow::*;
	use std::net::SocketAddr;
	use hyper::service::{make_service_fn, service_fn};
	use std::sync::{Arc, RwLock};

	fn mutate_request(req: &mut Request<Body>) -> Result<()> {
	for key in &["content-length", "transfer-encoding", "accept-encoding", "content-encoding"] {
	req.headers_mut().remove(*key);
	}

	let uri = req.uri();
	let uri_string = match uri.query() {
	None => format!("https://www.snoyman.com{}", uri.path()),
	Some(query) => format!("https://www.snoyman.com{}?{}", uri.path(), query),
	};
	*req.uri_mut() = uri_string.parse().context("Parsing URI in mutate_request")?;
	Ok(())
	}

	#[derive(Debug)]
	struct Stats {
	proxied: usize,
	}

	#[tokio::main]
	async fn main() -> Result<()> {
	let https = hyper_rustls::HttpsConnector::with_native_roots();
	let client: Client<_, hyper::Body> = Client::builder().build(https);
	let client: Arc<Client<_, hyper::Body>> = Arc::new(client);
	let stats: Arc<RwLock<Stats>> = Arc::new(RwLock::new(Stats {
	proxied: 0,
	}));

	let addr = SocketAddr::from(([0, 0, 0, 0], 3000));

	let make_svc = make_service_fn(move \|_\| {
	let client = Arc::clone(&client);
	let stats = Arc::clone(&stats);
	async move {
	Ok::<_, Error>(service_fn(move \|mut req\| {
	let client = Arc::clone(&client);
	let stats = Arc::clone(&stats);
	async move {
	if req.uri().path() == "/status" {
	let stats: &Stats = &*stats.read().unwrap();
	let body: Body = format!("{:?}", stats).into();
	Ok(Response::new(body))
	} else {
	println!("Proxied: {}", req.uri().path());
	stats.write().unwrap().proxied += 1;
	mutate_request(&mut req)?;
	client.request(req).await.context("Making request to backend server")
	}
	}
	}))
	}
	});
	Server::bind(&addr).serve(make_svc).await.context("Running server")?;
	Ok::<(), anyhow::Error>(())
	}