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
|
use std::{
io::{Seek, Write},
ops::Range,
};
use anyhow::anyhow;
use log::warn;
pub struct TrimWriter<W> {
inner: W,
position: usize,
range: Range<usize>,
}
impl<W: Write> TrimWriter<W> {
pub fn new(inner: W, range: Range<usize>) -> Self {
Self {
inner,
range,
position: 0,
}
}
}
impl<W: Write> Write for TrimWriter<W> {
fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
let start = self.range.start as isize - self.position as isize;
let end = self.range.end as isize - self.position as isize;
let start = start.clamp(0, buf.len() as isize) as usize;
let end = end.clamp(0, buf.len() as isize) as usize;
if self.position >= self.range.end {
return Err(std::io::Error::new(
std::io::ErrorKind::Other,
anyhow!("range ended"),
));
}
let buf = &buf[start..end];
if !buf.is_empty() {
self.inner.write_all(buf)?;
self.position += buf.len()
}
Ok(buf.len())
}
fn flush(&mut self) -> std::io::Result<()> {
self.inner.flush()
}
}
impl<W> Seek for TrimWriter<W> {
fn seek(&mut self, pos: std::io::SeekFrom) -> std::io::Result<u64> {
match pos {
std::io::SeekFrom::Start(s) => self.position = s as usize,
std::io::SeekFrom::End(_) => unimplemented!(),
std::io::SeekFrom::Current(s) => self.position += s as usize,
}
if self.position < self.range.end {
warn!("seeked beyond end")
}
Ok(self.position as u64)
}
}
|
