Axum proxy? #704

frederikhors · 2024-01-20T16:05:38Z

frederikhors
Jan 20, 2024

After hyper v1 revolution (they removed the native hyper Client moving it in "legacy" crate in another hyper-util crate) I would like to use ureq (instead of reqwest or others).

I need to use it in multiple places but one of them is driving me crazy (along with #705).

I'm using "axum" and I need to "proxy" some requests, but I don't understand how to do it.

The actual code is:

use axum::{response::Html, routing::get, Router};

#[tokio::main]
async fn main() {
  let app = Router::new()
    .route("/", get(handler))
    .route("/custom/*key", get(proxy).post(proxy));

  let listener = tokio::net::TcpListener::bind("127.0.0.1:3000").await.unwrap();

  axum::serve(listener, app).await.unwrap();
}

pub async fn proxy(mut req: Request<Body>) -> Result<Response<Body>, StatusCode> {
  let path_query = req
    .uri()
    .path_and_query()
    .map_or_else(|| req.uri().path(), |v| v.as_str());

  let uri = format!("http://localhost:9000/{}", path_query);

  *req.uri_mut() = Uri::try_from(uri).unwrap();

  // This is not re-created each time in real code
  let hyper_client = HyperClient::new();

  let res = hyper_client.request(req).await.unwrap();

  if res.status().is_server_error() {
    return Err(StatusCode::INTERNAL_SERVER_ERROR);
  }

  Ok(res)
}

And this works because axum 0.6 uses the same Request and Response of hyper pre v1.

Now with ureq I would like to know if something like this is possible.

I tried with:

pub async fn proxy(mut req: Request<Body>) -> Result<Response<Body>, StatusCode> {
  let path_query = req
    .uri()
    .path_and_query()
    .map_or_else(|| req.uri().path(), |v| v.as_str());

  let proxy = ureq::Proxy::new("localhost:9000").unwrap();

  let agent = ureq::AgentBuilder::new().proxy(proxy).build();

  let res = agent
    .request(req.method().as_str(), req.uri())
    .call()
    .map_err(|err| {
      dbg!(err);
      StatusCode::INTERNAL_SERVER_ERROR
    })
    .unwrap();

  Ok(res.into_string().unwrap().into_response())
}

but obviously this doesn't work because for example if the localhost:9000 answers with a redirect the status code is wrong, the body too and so on.

I'm trying to understand if this is the right way to do it...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Axum proxy? #704

{{title}}

{{editor}}'s edit

{{editor}}'s edit

Replies: 0 comments

Select a reply

Axum proxy? #704

frederikhors Jan 20, 2024

Replies: 0 comments

frederikhors
Jan 20, 2024