@@ -414,10 +414,8 @@ async fn async_test(cfg: __TestConfig) {
414
414
async move {
415
415
let stream = TcpStream :: connect ( addr) . await . unwrap ( ) ;
416
416
417
- // TODO: cleaner way?
418
417
let res = if http2_only {
419
- let ( mut sender, conn) = hyper:: client:: conn:: http2:: Builder :: new ( )
420
- . handshake :: < TcpStream , Body > ( stream)
418
+ let ( mut sender, conn) = hyper:: client:: conn:: http2:: handshake :: < TcpStream > ( stream)
421
419
. await
422
420
. unwrap ( ) ;
423
421
@@ -428,8 +426,7 @@ async fn async_test(cfg: __TestConfig) {
428
426
} ) ;
429
427
sender. send_request ( req) . await . unwrap ( )
430
428
} else {
431
- let ( mut sender, conn) = hyper:: client:: conn:: http1:: Builder :: new ( )
432
- . handshake :: < TcpStream , Body > ( stream)
429
+ let ( mut sender, conn) = hyper:: client:: conn:: http1:: handshake :: < TcpStream > ( stream)
433
430
. await
434
431
. unwrap ( ) ;
435
432
@@ -516,28 +513,27 @@ async fn naive_proxy(cfg: ProxyConfig) -> (SocketAddr, impl Future<Output = ()>)
516
513
let resp = if http2_only {
517
514
let builder = hyper:: client:: conn:: http2:: Builder :: new ( ) ;
518
515
let ( mut sender, conn) = builder. handshake ( stream) . await . unwrap ( ) ;
519
-
516
+
520
517
tokio:: task:: spawn ( async move {
521
518
if let Err ( err) = conn. await {
522
519
panic ! ( "{:?}" , err) ;
523
520
}
524
521
} ) ;
525
-
522
+
526
523
sender. send_request ( req) . await ?
527
524
} else {
528
525
let builder = hyper:: client:: conn:: http1:: Builder :: new ( ) ;
529
526
let ( mut sender, conn) = builder. handshake ( stream) . await . unwrap ( ) ;
530
-
527
+
531
528
tokio:: task:: spawn ( async move {
532
529
if let Err ( err) = conn. await {
533
530
panic ! ( "{:?}" , err) ;
534
531
}
535
532
} ) ;
536
-
533
+
537
534
sender. send_request ( req) . await ?
538
535
} ;
539
536
540
-
541
537
let ( mut parts, body) = resp. into_parts ( ) ;
542
538
543
539
// Remove the Connection header for HTTP/1.1 proxy connections.
0 commit comments