Add buffers
This commit is contained in:
parent
da80468266
commit
fdca7ada0b
209
src/buffers.rs
Normal file
209
src/buffers.rs
Normal file
@ -0,0 +1,209 @@
|
|||||||
|
struct ReadState {
|
||||||
|
// rx: OwnedReadHalf,
|
||||||
|
buffer: Vec<u8>,
|
||||||
|
cursor: usize,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ReadState {
|
||||||
|
fn has_next(&self) -> bool {
|
||||||
|
self.buffer[..self.cursor]
|
||||||
|
.iter()
|
||||||
|
.position(|v| *v == b'\n')
|
||||||
|
.is_some()
|
||||||
|
}
|
||||||
|
fn shift_ramaining(&mut self) {
|
||||||
|
if !self.has_next() {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
let pos = self.cursor;
|
||||||
|
let mut last_old = 0;
|
||||||
|
let mut last_new = 0;
|
||||||
|
for (new, old) in (pos..self.buffer.len()).enumerate() {
|
||||||
|
let v = self.buffer[old];
|
||||||
|
if v == 0 {
|
||||||
|
last_old = old;
|
||||||
|
last_new = new;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
// eprintln!("new {new} old {old} = {c} ({v})", c = v as char);
|
||||||
|
self.buffer[new] = v;
|
||||||
|
self.buffer[old] = 0;
|
||||||
|
}
|
||||||
|
for old in (last_new..pos).chain(last_old..self.buffer.len()) {
|
||||||
|
// eprintln!("erase {old}");
|
||||||
|
self.buffer[old] = 0;
|
||||||
|
}
|
||||||
|
self.cursor = 0;
|
||||||
|
}
|
||||||
|
fn next(&mut self) -> Option<&[u8]> {
|
||||||
|
let Some(end) = self.buffer[self.cursor..self.buffer.len()]
|
||||||
|
.iter()
|
||||||
|
.position(|v| *v == b'\n').map(|n| n + self.cursor) else {
|
||||||
|
return None;
|
||||||
|
};
|
||||||
|
// eprintln!("cursor {} end {end}", self.cursor);
|
||||||
|
let s = &self.buffer[self.cursor..=end];
|
||||||
|
self.cursor = (end + 1).min(self.buffer.len() - 1);
|
||||||
|
Some(s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn shift_empty() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: vec![0, 0, 0, 0, 0, 0, 0, 0],
|
||||||
|
cursor: 0,
|
||||||
|
};
|
||||||
|
state.shift_ramaining();
|
||||||
|
let expected = vec![0, 0, 0, 0, 0, 0, 0, 0];
|
||||||
|
assert_eq!(
|
||||||
|
std::str::from_utf8(&state.buffer),
|
||||||
|
std::str::from_utf8(&expected),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn shift_single_unterminated() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: vec![b'h', b'e', b'l', b'l', b'o', 0, 0, 0],
|
||||||
|
cursor: 0,
|
||||||
|
};
|
||||||
|
state.shift_ramaining();
|
||||||
|
let expected = vec![b'h', b'e', b'l', b'l', b'o', 0, 0, 0];
|
||||||
|
assert_eq!(
|
||||||
|
std::str::from_utf8(&state.buffer),
|
||||||
|
std::str::from_utf8(&expected),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn shift_single_terminated() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: vec![b'h', b'e', b'l', b'l', b'o', b'\r', b'\n', 0],
|
||||||
|
cursor: 0,
|
||||||
|
};
|
||||||
|
state.shift_ramaining();
|
||||||
|
let expected = vec![b'h', b'e', b'l', b'l', b'o', b'\r', b'\n', 0];
|
||||||
|
assert_eq!(
|
||||||
|
std::str::from_utf8(&state.buffer),
|
||||||
|
std::str::from_utf8(&expected),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn shift_single_unterminated_moved() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: b"hello\r\nworld\0\0\0\0".to_vec(),
|
||||||
|
cursor: 7,
|
||||||
|
};
|
||||||
|
state.shift_ramaining();
|
||||||
|
let expected = b"world\0\0\0\0\0\0\0\0\0\0\0".to_vec();
|
||||||
|
assert_eq!(
|
||||||
|
std::str::from_utf8(&state.buffer),
|
||||||
|
std::str::from_utf8(&expected),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn shift_single_terminated_moved() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: b"hello\r\nworld\r\n\0\0\0\0\0\0\0\0".to_vec(),
|
||||||
|
cursor: 7,
|
||||||
|
};
|
||||||
|
state.shift_ramaining();
|
||||||
|
let expected = b"world\r\n\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0".to_vec();
|
||||||
|
assert_eq!(
|
||||||
|
std::str::from_utf8(&state.buffer),
|
||||||
|
std::str::from_utf8(&expected),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn empty_next() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: b"\0\0\0\0\0\0\0\0".to_vec(),
|
||||||
|
cursor: 0,
|
||||||
|
};
|
||||||
|
assert_eq!(state.next(), None);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn depleted_next() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: b"hello\0\0\0\0\0\0\0\0".to_vec(),
|
||||||
|
cursor: 5,
|
||||||
|
};
|
||||||
|
assert_eq!(state.next(), None);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn one_world_next() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: b"hello\r\n\0\0\0\0\0\0\0\0".to_vec(),
|
||||||
|
cursor: 0,
|
||||||
|
};
|
||||||
|
assert_eq!(state.next(), Some(b"hello\r\n".as_slice()));
|
||||||
|
assert_eq!(state.cursor, 7);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn tree_world_next() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: b"hello\r\nworld\r\nnats\r\n\0\0\0\0\0\0\0\0".to_vec(),
|
||||||
|
cursor: 0,
|
||||||
|
};
|
||||||
|
assert_eq!(state.next(), Some(b"hello\r\n".as_slice()));
|
||||||
|
assert_eq!(state.cursor, 7);
|
||||||
|
assert_eq!(state.next(), Some(b"world\r\n".as_slice()));
|
||||||
|
assert_eq!(state.cursor, 14);
|
||||||
|
assert_eq!(state.next(), Some(b"nats\r\n".as_slice()));
|
||||||
|
assert_eq!(state.cursor, 20);
|
||||||
|
assert_eq!(state.next(), None);
|
||||||
|
assert_eq!(state.cursor, 20);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn partial_move_and_shift() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: b"hello\r\nworld\r\nnats\r\n\0\0\0\0\0\0\0\0".to_vec(),
|
||||||
|
cursor: 0,
|
||||||
|
};
|
||||||
|
state.next();
|
||||||
|
state.next();
|
||||||
|
state.shift_ramaining();
|
||||||
|
let expected = b"nats\r\n\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0".to_vec();
|
||||||
|
assert_eq!(
|
||||||
|
std::str::from_utf8(&state.buffer),
|
||||||
|
std::str::from_utf8(&expected),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn move_and_shift_non_terminated() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: b"hello\r\nworld\r\nnats\0\0\0\0\0\0\0\0\0\0".to_vec(),
|
||||||
|
cursor: 0,
|
||||||
|
};
|
||||||
|
state.next();
|
||||||
|
state.next();
|
||||||
|
state.next();
|
||||||
|
state.shift_ramaining();
|
||||||
|
let expected = b"nats\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0".to_vec();
|
||||||
|
assert_eq!(
|
||||||
|
std::str::from_utf8(&state.buffer),
|
||||||
|
std::str::from_utf8(&expected),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn full_move_and_shift() {
|
||||||
|
let mut state = ReadState {
|
||||||
|
buffer: b"hello\r\nworld\r\nnats\r\n\0\0\0\0\0\0\0\0".to_vec(),
|
||||||
|
cursor: 0,
|
||||||
|
};
|
||||||
|
state.next();
|
||||||
|
state.next();
|
||||||
|
state.next();
|
||||||
|
assert_eq!(state.cursor, 20);
|
||||||
|
state.shift_ramaining();
|
||||||
|
let expected = b"\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0".to_vec();
|
||||||
|
assert_eq!(
|
||||||
|
std::str::from_utf8(&state.buffer),
|
||||||
|
std::str::from_utf8(&expected),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user