xfy/candy
1
0
Fork 0
mirror of https://github.com/DefectingCat/candy synced 2025-07-15 16:51:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(reverse): forward request body

Browse Source
This commit is contained in:
xfy
2025-06-11 20:50:13 +08:00
parent 10160ab5e5
commit 70c4b29e7f
1 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/http/reverse_proxy.rs
View File

@ -19,7 +19,7 @@ pub async fn serve(
req_uri: Uri, req_uri: Uri,
path: Option<Path<String>>, path: Option<Path<String>>,
Host(host): Host, Host(host): Host,
mut req: Request, mut req: Request<Body>,
) -> RouteResult<impl IntoResponse> { ) -> RouteResult<impl IntoResponse> {
let req_path = req.uri().path(); let req_path = req.uri().path();
let path_query = req let path_query = req
@ -56,11 +56,26 @@ pub async fn serve(
} }
} }
// forward request body
let body = req.into_body();
// TODO: set body size limit
let bytes = axum::body::to_bytes(body, 2048).await.map_err(|err| {
tracing::error!("Failed to proxy request: {}", err);
RouteError::InternalError()
})?;
let body_str = String::from_utf8(bytes.to_vec()).map_err(|err| {
tracing::error!("Failed to proxy request: {}", err);
RouteError::InternalError()
})?;
forward_req = forward_req.body(body_str);
// send reverse proxy request
let reqwest_response = forward_req.send().await.map_err(|e| { let reqwest_response = forward_req.send().await.map_err(|e| {
tracing::error!("Failed to proxy request: {}", e); tracing::error!("Failed to proxy request: {}", e);
RouteError::BadRequest() RouteError::BadRequest()
})?; })?;
// response from reverse proxy server
let mut response_builder = Response::builder().status(reqwest_response.status()); let mut response_builder = Response::builder().status(reqwest_response.status());
copy_headers( copy_headers(
reqwest_response.headers(), reqwest_response.headers(),