micropelago/hyper-reverse-proxy
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: improve flow

Browse Source
This commit is contained in:
chesedo 2022-05-09 11:45:00 +02:00 committed by Felipe Noronha
parent ae854b395f
commit 87f1ed675a
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/lib.rs
View File

@ -412,6 +412,8 @@ pub async fn call<'a, T: hyper::client::connect::Connect + Clone + Send + Sync +
.expect("response does not have an upgrade extension") .expect("response does not have an upgrade extension")
.await?; .await?;
debug!("Responding to a connection upgrade response");
tokio::spawn(async move { tokio::spawn(async move {
let mut request_upgraded = request_upgraded let mut request_upgraded = request_upgraded
.expect("request does not have an upgrade extension") .expect("request does not have an upgrade extension")
@ -423,14 +425,14 @@ pub async fn call<'a, T: hyper::client::connect::Connect + Clone + Send + Sync +
.expect("coping between upgraded connections failed"); .expect("coping between upgraded connections failed");
}); });
return Ok(response); Ok(response)
} } else {
let proxied_response = create_proxied_response(response); let proxied_response = create_proxied_response(response);
debug!("Responding to call with response"); debug!("Responding to call with response");
Ok(proxied_response) Ok(proxied_response)
} }
}
pub struct ReverseProxy<T: hyper::client::connect::Connect + Clone + Send + Sync + 'static> { pub struct ReverseProxy<T: hyper::client::connect::Connect + Clone + Send + Sync + 'static> {
client: Client<T>, client: Client<T>,