Make Chunk flexible in the type of backing buffer used

This commit is contained in:
Tangent 128 2017-09-25 00:22:41 -04:00
parent 0a18cb408f
commit 27642f7b14
1 changed files with 8 additions and 7 deletions

View File

@ -1,8 +1,9 @@
use std::rc::Rc; use std::rc::Rc;
pub enum Chunk { #[derive(Clone)]
pub enum Chunk<B: AsRef<[u8]> = Vec<u8>> {
Headers { Headers {
bytes: Rc<Vec<u8>> bytes: Rc<B>
}, },
ClusterHead { ClusterHead {
keyframe: bool, keyframe: bool,
@ -12,16 +13,16 @@ pub enum Chunk {
bytes: [u8;16] bytes: [u8;16]
}, },
ClusterBody { ClusterBody {
bytes: Rc<Vec<u8>> bytes: Rc<B>
} }
} }
impl AsRef<[u8]> for Chunk { impl<B: AsRef<[u8]>> AsRef<[u8]> for Chunk<B> {
fn as_ref(&self) -> &[u8] { fn as_ref(&self) -> &[u8] {
match self { match self {
&Chunk::Headers {ref bytes, ..} => &bytes, &Chunk::Headers {ref bytes, ..} => bytes.as_ref().as_ref(),
&Chunk::ClusterHead {ref bytes, ..} => bytes, &Chunk::ClusterHead {ref bytes, ..} => bytes,
&Chunk::ClusterBody {ref bytes, ..} => &bytes &Chunk::ClusterBody {ref bytes, ..} => bytes.as_ref().as_ref()
} }
} }
} }
@ -35,7 +36,7 @@ mod tests {
#[test] #[test]
fn enough_space_for_header() { fn enough_space_for_header() {
let mut chunk = Chunk::ClusterHead { let mut chunk: Chunk = Chunk::ClusterHead {
keyframe: false, keyframe: false,
start: 0, start: 0,
end: 0, end: 0,