postrust_proxy/vendored/
forwarder.rs

1//! Vendored forwarder client from rpxy-lib: forwarder/client.rs
2//!
3//! This module handles forwarding requests to upstream backends.
4
5use crate::config::Backend;
6use crate::vendored::hyper_ext::ProxyBody;
7use crate::vendored::types::ProxyError;
8use hyper::body::Incoming;
9use hyper::{Request, Response};
10use hyper_util::client::legacy::Client;
11use hyper_util::rt::TokioExecutor as HyperTokioExecutor;
12use std::time::Duration;
13
14/// HTTP connector type alias.
15type HttpConnector = hyper_util::client::legacy::connect::HttpConnector;
16
17/// Forwarder client for sending requests to upstream backends.
18pub struct ForwarderClient {
19    /// HTTP client
20    http_client: Client<HttpConnector, ProxyBody>,
21    /// Request timeout
22    timeout: Duration,
23}
24
25impl ForwarderClient {
26    /// Create a new forwarder client.
27    pub fn new(timeout: Duration) -> Self {
28        let http_connector = HttpConnector::new();
29        let http_client = Client::builder(HyperTokioExecutor::new())
30            .pool_idle_timeout(Duration::from_secs(30))
31            .pool_max_idle_per_host(32)
32            .build(http_connector);
33
34        Self {
35            http_client,
36            timeout,
37        }
38    }
39
40    /// Forward a request to a backend.
41    pub async fn forward(
42        &self,
43        backend: &Backend,
44        mut request: Request<ProxyBody>,
45    ) -> Result<Response<Incoming>, ProxyError> {
46        // Build the upstream URL
47        let uri = request.uri();
48        let path_and_query = uri.path_and_query().map(|pq| pq.as_str()).unwrap_or("/");
49
50        let upstream_uri = format!("{}://{}{}", backend.scheme, backend.address, path_and_query);
51
52        // Update request URI
53        *request.uri_mut() = upstream_uri
54            .parse()
55            .map_err(|e| ProxyError::Request(format!("Invalid upstream URI: {}", e)))?;
56
57        // Send request with timeout
58        let response = tokio::time::timeout(self.timeout, self.http_client.request(request))
59            .await
60            .map_err(|_| ProxyError::Timeout)?
61            .map_err(|e| ProxyError::Connection(e.to_string()))?;
62
63        Ok(response)
64    }
65}
66
67impl Default for ForwarderClient {
68    fn default() -> Self {
69        Self::new(Duration::from_secs(30))
70    }
71}
postrust_proxy/vendored/forwarder.rs

postrust_proxy/vendored/
forwarder.rs
