diff options
Diffstat (limited to 'src/modules/proxy.rs')
| -rw-r--r-- | src/modules/proxy.rs | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/modules/proxy.rs b/src/modules/proxy.rs index ce72f65..925e456 100644 --- a/src/modules/proxy.rs +++ b/src/modules/proxy.rs @@ -1,8 +1,9 @@ use super::{Node, NodeContext, NodeKind, NodeRequest, NodeResponse}; -use crate::{helper::TokioIo, ServiceError}; +use crate::ServiceError; use futures::Future; use http_body_util::BodyExt; use hyper::{http::HeaderValue, upgrade::OnUpgrade, StatusCode}; +use hyper_util::rt::TokioIo; use log::{debug, warn}; use serde::Deserialize; use serde_yaml::Value; @@ -42,7 +43,7 @@ impl Node for Proxy { let _limit_guard = context.state.l_outgoing.try_acquire()?; debug!("\tforwarding to {}", self.backend); let mut resp = { - let client_stream = TokioIo( + let client_stream = TokioIo::new( TcpStream::connect(self.backend) .await .map_err(|_| ServiceError::CantConnect)?, @@ -75,8 +76,8 @@ impl Node for Proxy { (Ok(upgraded_upstream), Ok(upgraded_downstream)) => { debug!("upgrade successful"); match tokio::io::copy_bidirectional( - &mut TokioIo(upgraded_downstream), - &mut TokioIo(upgraded_upstream), + &mut TokioIo::new(upgraded_downstream), + &mut TokioIo::new(upgraded_upstream), ) .await { |