@@ -8,10 +8,9 @@ use std::{
88} ;
99use DigestAlgorithm :: { Sha1 , Sha256 , Sha384 , Sha512 } ;
1010
11- use futures:: future:: { FutureExt , TryFutureExt } ;
1211use ring:: digest;
13- use rustls:: ClientConfig ;
1412use rustls:: pki_types:: ServerName ;
13+ use rustls:: ClientConfig ;
1514use tokio:: io:: { AsyncRead , AsyncWrite , ReadBuf } ;
1615use tokio_postgres:: tls:: { ChannelBinding , MakeTlsConnect , TlsConnect } ;
1716use tokio_rustls:: { client:: TlsStream , TlsConnector } ;
@@ -71,11 +70,12 @@ where
7170 fn connect ( self , stream : S ) -> Self :: Future {
7271 match self . 0 {
7372 None => Box :: pin ( core:: future:: ready ( Err ( io:: ErrorKind :: InvalidInput . into ( ) ) ) ) ,
74- Some ( c) => c
75- . connector
76- . connect ( c. hostname , stream)
77- . map_ok ( |s| RustlsStream ( Box :: pin ( s) ) )
78- . boxed ( ) ,
73+ Some ( c) => Box :: pin ( async move {
74+ c. connector
75+ . connect ( c. hostname , stream)
76+ . await
77+ . map ( |s| RustlsStream ( Box :: pin ( s) ) )
78+ } ) ,
7979 }
8080 }
8181}
@@ -151,13 +151,12 @@ where
151151#[ cfg( test) ]
152152mod tests {
153153 use super :: * ;
154- use futures :: future :: TryFutureExt ;
154+ use rustls :: pki_types :: { CertificateDer , UnixTime } ;
155155 use rustls:: {
156156 client:: danger:: ServerCertVerifier ,
157157 client:: danger:: { HandshakeSignatureValid , ServerCertVerified } ,
158158 Error , SignatureScheme ,
159159 } ;
160- use rustls:: pki_types:: { CertificateDer , UnixTime } ;
161160
162161 #[ derive( Debug ) ]
163162 struct AcceptAllVerifier { }
@@ -220,7 +219,7 @@ mod tests {
220219 )
221220 . await
222221 . expect ( "connect" ) ;
223- tokio:: spawn ( conn. map_err ( |e| panic ! ( "{:?}" , e) ) ) ;
222+ tokio:: spawn ( async move { conn. await . map_err ( |e| panic ! ( "{:?}" , e) ) } ) ;
224223 let stmt = client. prepare ( "SELECT 1" ) . await . expect ( "prepare" ) ;
225224 let _ = client. query ( & stmt, & [ ] ) . await . expect ( "query" ) ;
226225 }
0 commit comments