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

update iovec

Browse Source
This commit is contained in:
quininer 2019-05-18 23:44:29 +08:00
parent 4cc374fd4c
commit 4d673f9a72
2 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Cargo.toml
View File

@ -17,9 +17,7 @@ appveyor = { repository = "quininer/tokio-rustls" }
[dependencies]
smallvec = "*"
futures = { package = "futures-preview", version = "0.3.0-alpha.15" }
bytes = "0.4"
iovec = "0.1"
futures = { package = "futures-preview", version = "0.3.0-alpha.16" }
rustls = "0.15"
webpki = "0.19"

11
src/common/mod.rs
View File

@ -7,7 +7,7 @@ use std::io::{ self, Read };
use rustls::Session;
use rustls::WriteV;
use futures::task::Context;
use futures::io::{ AsyncRead, AsyncWrite, IoVec };
use futures::io::{ AsyncRead, AsyncWrite, IoSlice };
use smallvec::SmallVec;
@ -159,13 +159,10 @@ impl<'a, IO: AsyncRead + AsyncWrite + Unpin, S: Session> WriteTls<IO, S> for Str
fn writev(&mut self, vbytes: &[&[u8]]) -> io::Result<usize> {
let vbytes = vbytes
.into_iter()
.try_fold(SmallVec::<[&'_ IoVec; 16]>::new(), |mut sum, next| {
sum.push(IoVec::from_bytes(next)?);
Some(sum)
})
.unwrap_or_default();
.map(|v| IoSlice::new(v))
.collect::<SmallVec<[IoSlice<'_>; 64]>>();
match Pin::new(&mut self.io).poll_vectored_write(self.cx, &vbytes) {
match Pin::new(&mut self.io).poll_write_vectored(self.cx, &vbytes) {
Poll::Ready(result) => result,
Poll::Pending => Err(io::ErrorKind::WouldBlock.into())
}