tuxmain/rust-rpxy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: separeted forwarder definition for more flexibility

Browse source
This commit is contained in:
Jun Kurihara 2023-08-08 17:59:20 +09:00
commit 43b004cf6e
No known key found for this signature in database
GPG key ID: D992B3E3DE1DED23
5 changed files with 84 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

58
rpxy-lib/src/handler/forwarder.rs Normal file
View file

@ -0,0 +1,58 @@
use crate::error::RpxyError;
use async_trait::async_trait;
use derive_builder::Builder;
use hyper::{
body::{Body, HttpBody},
client::{connect::Connect, HttpConnector},
Client, Request, Response,
};
use hyper_rustls::HttpsConnector;
#[async_trait]
/// Definition of the forwarder that simply forward requests from downstream client to upstream app servers.
pub trait ForwardRequest<B> {
type Error;
async fn request(&self, req: Request<B>) -> Result<Response<Body>, Self::Error>;
}
#[derive(Builder, Clone)]
/// Forwarder struct
pub struct Forwarder<C, B = Body>
where
C: Connect + Clone + Sync + Send + 'static,
{
// TODO: need `h2c` or http/2-only client separately
inner: Client<C, B>,
}
#[async_trait]
impl<C, B> ForwardRequest<B> for Forwarder<C, B>
where
B: HttpBody + Send + 'static,
B::Data: Send,
B::Error: Into<Box<dyn std::error::Error + Send + Sync>>,
C: Connect + Clone + Sync + Send + 'static,
{
type Error = RpxyError;
async fn request(&self, req: Request<B>) -> Result<Response<Body>, Self::Error> {
// TODO:
// TODO: Implement here a client that handles `h2c` requests
// TODO:
self.inner.request(req).await.map_err(RpxyError::Hyper)
}
}
impl Forwarder<HttpsConnector<HttpConnector>, Body> {
pub async fn new() -> Self {
// let connector = TrustDnsResolver::default().into_rustls_webpki_https_connector();
let connector = hyper_rustls::HttpsConnectorBuilder::new()
.with_webpki_roots()
.https_or_http()
.enable_http1()
.enable_http2()
.build();
let inner = Client::builder().build::<_, Body>(connector);
Self { inner }
}
}