diff --git a/src/proxy/proxy_main.rs b/src/proxy/proxy_main.rs index 1a8c707..22434d5 100644 --- a/src/proxy/proxy_main.rs +++ b/src/proxy/proxy_main.rs @@ -61,10 +61,12 @@ where self.globals.runtime_handle.clone().spawn(async move { tokio::time::timeout( self.globals.timeout + Duration::from_secs(1), - server.serve_connection( - stream, - service_fn(move |req: Request| self.clone().handle_request(req, peer_addr)), - ), + server + .serve_connection( + stream, + service_fn(move |req: Request| self.clone().handle_request(req, peer_addr)), + ) + .with_upgrades(), ) .await .ok();