fix upgrade for http1.1 and 2

2022-07-06 14:13:55 +09:00 · 2022-07-06 14:13:55 +09:00 · 6d5c904f12
commit 6d5c904f12
parent aa95ab9c16
1 changed files with 6 additions and 4 deletions
--- 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(
+        server
+          .serve_connection(
            stream,
            service_fn(move |req: Request<Body>| self.clone().handle_request(req, peer_addr)),
-        ),
+          )
+          .with_upgrades(),
      )
      .await
      .ok();