only set written_to if io succeeds

This commit is contained in:
aiden 2023-05-25 20:22:08 +01:00
parent ff2ede75de
commit 08f8772360
Signed by: aiden
GPG Key ID: EFA9C74AEBF806E0
2 changed files with 13 additions and 7 deletions

@ -1,4 +1,4 @@
[package]
name = "dropfile"
version = "1.0.1"
version = "1.0.2"
edition = "2021"

@ -94,18 +94,24 @@ impl Read for DropFile {
}
impl Write for DropFile {
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
self.written_to = true;
return self.deref_mut().write(buf);
return self.deref_mut().write(buf).map(|w| {
self.written_to = true;
return w;
});
}
fn write_vectored(&mut self, bufs: &[IoSlice<'_>]) -> io::Result<usize> {
self.written_to = true;
return self.deref_mut().write_vectored(bufs);
return self.deref_mut().write_vectored(bufs).map(|w| {
self.written_to = true;
return w;
});
}
fn flush(&mut self) -> io::Result<()> {
self.written_to = true;
return self.deref_mut().flush();
return self.deref_mut().flush().map(|_| {
self.written_to = true;
return ();
});
}
}