mirror of
https://github.com/tokio-rs/bytes.git
synced 2026-08-24 00:00:14 +02:00
Implementation of Buf for VecDeque (#249)
This commit is contained in:
committed by
Carl Lerche
parent
e13d2a783e
commit
0e8b440650
@@ -24,6 +24,7 @@ mod into_buf;
|
|||||||
mod iter;
|
mod iter;
|
||||||
mod reader;
|
mod reader;
|
||||||
mod take;
|
mod take;
|
||||||
|
mod vec_deque;
|
||||||
mod writer;
|
mod writer;
|
||||||
|
|
||||||
pub use self::buf::Buf;
|
pub use self::buf::Buf;
|
||||||
|
|||||||
@@ -0,0 +1,39 @@
|
|||||||
|
use std::collections::VecDeque;
|
||||||
|
|
||||||
|
use super::Buf;
|
||||||
|
|
||||||
|
impl Buf for VecDeque<u8> {
|
||||||
|
fn remaining(&self) -> usize {
|
||||||
|
self.len()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn bytes(&self) -> &[u8] {
|
||||||
|
let (s1, s2) = self.as_slices();
|
||||||
|
if s1.is_empty() {
|
||||||
|
s2
|
||||||
|
} else {
|
||||||
|
s1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn advance(&mut self, cnt: usize) {
|
||||||
|
self.drain(..cnt);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hello_world() {
|
||||||
|
let mut buffer: VecDeque<u8> = VecDeque::new();
|
||||||
|
buffer.extend(b"hello world");
|
||||||
|
assert_eq!(11, buffer.remaining());
|
||||||
|
assert_eq!(b"hello world", buffer.bytes());
|
||||||
|
buffer.advance(6);
|
||||||
|
assert_eq!(b"world", buffer.bytes());
|
||||||
|
buffer.extend(b" piece");
|
||||||
|
assert_eq!(b"world piece" as &[u8], &buffer.collect::<Vec<u8>>()[..]);
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user