1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66
use bytes::{Bytes, BytesMut}; use futures::Poll; use never::Never; use std::io; use BufStream; impl BufStream for Vec<u8> { type Item = io::Cursor<Vec<u8>>; type Error = Never; fn poll_buf(&mut self) -> Poll<Option<Self::Item>, Self::Error> { if self.is_empty() { return Ok(None.into()); } poll_bytes(self) } } impl BufStream for &'static [u8] { type Item = io::Cursor<&'static [u8]>; type Error = Never; fn poll_buf(&mut self) -> Poll<Option<Self::Item>, Self::Error> { if self.is_empty() { return Ok(None.into()); } poll_bytes(self) } } impl BufStream for Bytes { type Item = io::Cursor<Bytes>; type Error = Never; fn poll_buf(&mut self) -> Poll<Option<Self::Item>, Self::Error> { if self.is_empty() { return Ok(None.into()); } poll_bytes(self) } } impl BufStream for BytesMut { type Item = io::Cursor<BytesMut>; type Error = Never; fn poll_buf(&mut self) -> Poll<Option<Self::Item>, Self::Error> { if self.is_empty() { return Ok(None.into()); } poll_bytes(self) } } fn poll_bytes<T: Default>(buf: &mut T) -> Poll<Option<io::Cursor<T>>, Never> { use std::mem; let bytes = mem::replace(buf, Default::default()); let buf = io::Cursor::new(bytes); Ok(Some(buf).into()) }