WIP: libdelimiter #172
@ -18,66 +18,66 @@
 | 
			
		||||
 */
 | 
			
		||||
 | 
			
		||||
use std::{
 | 
			
		||||
    io::{ Read, Result },
 | 
			
		||||
    mem,
 | 
			
		||||
	io::{ Read, Result },
 | 
			
		||||
	mem,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
const BUFFER_SIZE: usize = 4096;
 | 
			
		||||
 | 
			
		||||
pub struct Delimited<T: Read> {
 | 
			
		||||
    delimiter: Vec<u8>,
 | 
			
		||||
    buffer: Vec<u8>,
 | 
			
		||||
    stream: T,
 | 
			
		||||
	delimiter: Vec<u8>,
 | 
			
		||||
	buffer: Vec<u8>,
 | 
			
		||||
	stream: T,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T> Delimited<T> where T: Read {
 | 
			
		||||
	pub fn new(stream: T, delimiter: &[u8]) -> Self {
 | 
			
		||||
        Delimited {
 | 
			
		||||
            stream,
 | 
			
		||||
            delimiter: delimiter.to_vec(),
 | 
			
		||||
            buffer: Vec::with_capacity(BUFFER_SIZE),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
		Delimited {
 | 
			
		||||
			stream,
 | 
			
		||||
			delimiter: delimiter.to_vec(),
 | 
			
		||||
			buffer: Vec::with_capacity(BUFFER_SIZE),
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl<T> Iterator for Delimited<T> where T: Read {
 | 
			
		||||
    type Item = Result<Vec<u8>>;
 | 
			
		||||
	type Item = Result<Vec<u8>>;
 | 
			
		||||
 | 
			
		||||
    fn next(&mut self) -> Option<Self::Item> {
 | 
			
		||||
        let mut buf = [0; BUFFER_SIZE];
 | 
			
		||||
	fn next(&mut self) -> Option<Self::Item> {
 | 
			
		||||
		let mut buf = [0; BUFFER_SIZE];
 | 
			
		||||
 | 
			
		||||
        loop {
 | 
			
		||||
            if let Some(p) = find_subslice(&self.buffer, &self.delimiter) {
 | 
			
		||||
                let chunk = self.buffer.drain(..p).collect::<Vec<_>>();
 | 
			
		||||
		loop {
 | 
			
		||||
			if let Some(p) = find_subslice(&self.buffer, &self.delimiter) {
 | 
			
		||||
				let chunk = self.buffer.drain(..p).collect::<Vec<_>>();
 | 
			
		||||
				
 | 
			
		||||
                let _ = self.buffer.drain(..self.delimiter.len());
 | 
			
		||||
				let _ = self.buffer.drain(..self.delimiter.len());
 | 
			
		||||
 | 
			
		||||
                return Some(Ok(chunk));
 | 
			
		||||
            }
 | 
			
		||||
				return Some(Ok(chunk));
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
            match self.stream.read(&mut buf) {
 | 
			
		||||
                Ok(0) => {
 | 
			
		||||
                    let _ = self.buffer.is_empty() && return None;
 | 
			
		||||
			match self.stream.read(&mut buf) {
 | 
			
		||||
				Ok(0) => {
 | 
			
		||||
					let _ = self.buffer.is_empty() && return None;
 | 
			
		||||
 | 
			
		||||
                    return Some(Ok(mem::take(&mut self.buffer)));
 | 
			
		||||
                },
 | 
			
		||||
                Ok(n) => {
 | 
			
		||||
                    self.buffer.extend_from_slice(&buf[..n]);
 | 
			
		||||
                },
 | 
			
		||||
                Err(e) => {
 | 
			
		||||
                    return Some(Err(e));
 | 
			
		||||
                },
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
					return Some(Ok(mem::take(&mut self.buffer)));
 | 
			
		||||
				},
 | 
			
		||||
				Ok(n) => {
 | 
			
		||||
					self.buffer.extend_from_slice(&buf[..n]);
 | 
			
		||||
				},
 | 
			
		||||
				Err(e) => {
 | 
			
		||||
					return Some(Err(e));
 | 
			
		||||
				},
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn find_subslice(stack: &[u8], key: &[u8]) -> Option<usize> {
 | 
			
		||||
    match key.len() {
 | 
			
		||||
    	/* TODO: is this optimization necessary? */
 | 
			
		||||
        1 => stack.iter().position(|&b| b == key[0]),
 | 
			
		||||
        _ => stack.windows(key.len()).position(|w| w == key),
 | 
			
		||||
    }
 | 
			
		||||
	match key.len() {
 | 
			
		||||
		/* TODO: is this optimization necessary? */
 | 
			
		||||
		1 => stack.iter().position(|&b| b == key[0]),
 | 
			
		||||
		_ => stack.windows(key.len()).position(|w| w == key),
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user