refactor: improve flow

This commit is contained in:
chesedo 2022-05-09 11:45:00 +02:00 committed by Felipe Noronha
parent ae854b395f
commit 87f1ed675a

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