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): add proxy request timeout

Browse Source
This commit is contained in:
xfy
2025-06-15 21:28:48 +08:00
parent 4d795d1a62
commit a6ddba3a2b
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/http/reverse_proxy.rs
View File

@ -1,3 +1,5 @@
use std::time::Duration;
use axum::{ use axum::{
body::Body, body::Body,
extract::{Path, Request}, extract::{Path, Request},
@ -35,11 +37,11 @@ pub async fn serve(
let parent_path = resolve_parent_path(&req_uri, path.as_ref()); let parent_path = resolve_parent_path(&req_uri, path.as_ref());
tracing::debug!("parent path: {:?}", parent_path); tracing::debug!("parent path: {:?}", parent_path);
let proxy_pass = route_map let proxy_config = route_map
.get(&parent_path) .get(&parent_path)
.ok_or(RouteError::RouteNotFound())?; .ok_or(RouteError::RouteNotFound())?;
tracing::debug!("proxy pass: {:?}", proxy_pass); tracing::debug!("proxy pass: {:?}", proxy_config);
let Some(ref proxy_pass) = proxy_pass.proxy_pass else { let Some(ref proxy_pass) = proxy_config.proxy_pass else {
// return custom_not_found!(host_route, request).await; // return custom_not_found!(host_route, request).await;
return Err(RouteError::RouteNotFound()); return Err(RouteError::RouteNotFound());
}; };
@ -47,9 +49,13 @@ pub async fn serve(
tracing::debug!("reverse proxy uri: {:?}", &uri); tracing::debug!("reverse proxy uri: {:?}", &uri);
*req.uri_mut() = Uri::try_from(uri.clone()).map_err(|_| RouteError::InternalError())?; *req.uri_mut() = Uri::try_from(uri.clone()).map_err(|_| RouteError::InternalError())?;
let timeout = proxy_config.proxy_timeout;
// forward request headers // forward request headers
let client = Client::new(); let client = Client::new();
let mut forward_req = client.request(req.method().clone(), uri); let mut forward_req = client
.request(req.method().clone(), uri)
.timeout(Duration::from_secs(timeout.into()));
for (name, value) in req.headers().iter() { for (name, value) in req.headers().iter() {
if !is_exclude_header(name) { if !is_exclude_header(name) {
forward_req = forward_req.header(name.clone(), value.clone()); forward_req = forward_req.header(name.clone(), value.clone());