micropelago/tokio-rustls
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

move sleep

Browse Source
This commit is contained in:
quininer 2019-10-10 22:52:31 +08:00
parent 369c13d6a5
commit d8235071cd
1 changed files with 9 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/early-data.rs
View File

@ -10,7 +10,6 @@ use std::task::{ Context, Poll };
use std::time::Duration; use std::time::Duration;
use tokio::prelude::*; use tokio::prelude::*;
use tokio::net::TcpStream; use tokio::net::TcpStream;
use tokio::io::split;
use tokio::timer::delay_for; use tokio::timer::delay_for;
use futures_util::{ future, ready }; use futures_util::{ future, ready };
use rustls::ClientConfig; use rustls::ClientConfig;
@ -41,22 +40,17 @@ async fn send(config: Arc<ClientConfig>, addr: SocketAddr, data: &[u8])
stream.write_all(data).await?; stream.write_all(data).await?;
stream.flush().await?; stream.flush().await?;
let (r, mut w) = split(stream);
let fut = Read1(r);
let fut2 = async move {
// sleep 3s // sleep 3s
// //
// see https://www.mail-archive.com/openssl-users@openssl.org/msg84451.html // see https://www.mail-archive.com/openssl-users@openssl.org/msg84451.html
delay_for(Duration::from_secs(3)).await; let sleep3 = delay_for(Duration::from_secs(3));
w.shutdown().await?; let mut stream = match future::select(Read1(stream), sleep3).await {
Ok(w) as io::Result<_> future::Either::Right((_, Read1(stream))) => stream,
future::Either::Left((Err(err), _)) => return Err(err),
future::Either::Left((Ok(_), _)) => unreachable!(),
}; };
let stream = match future::select(fut, fut2.boxed()).await { stream.shutdown().await?;
future::Either::Left(_) => unreachable!(),
future::Either::Right((Ok(w), Read1(r))) => r.unsplit(w),
future::Either::Right((Err(err), _)) => return Err(err)
};
Ok(stream) Ok(stream)
} }