From 3590dbfa5eeebdb0b511862770a3715f35deb527 Mon Sep 17 00:00:00 2001 From: Steven Allen Date: Sat, 8 Apr 2023 13:30:50 -0700 Subject: [PATCH] chore: modern rust (#231) --- src/spooled.rs | 34 +++++++++++++++++----------------- src/util.rs | 9 +++------ 2 files changed, 20 insertions(+), 23 deletions(-) diff --git a/src/spooled.rs b/src/spooled.rs index db14967ca..a736cb718 100644 --- a/src/spooled.rs +++ b/src/spooled.rs @@ -86,7 +86,7 @@ impl SpooledTempFile { pub fn roll(&mut self) -> io::Result<()> { if !self.is_rolled() { let mut file = tempfile()?; - if let SpooledData::InMemory(ref mut cursor) = self.inner { + if let SpooledData::InMemory(cursor) = &mut self.inner { file.write_all(cursor.get_ref())?; file.seek(SeekFrom::Start(cursor.position()))?; } @@ -99,12 +99,12 @@ impl SpooledTempFile { if size as usize > self.max_size { self.roll()?; // does nothing if already rolled over } - match self.inner { - SpooledData::InMemory(ref mut cursor) => { + match &mut self.inner { + SpooledData::InMemory(cursor) => { cursor.get_mut().resize(size as usize, 0); Ok(()) } - SpooledData::OnDisk(ref mut file) => file.set_len(size), + SpooledData::OnDisk(file) => file.set_len(size), } } @@ -117,9 +117,9 @@ impl SpooledTempFile { impl Read for SpooledTempFile { fn read(&mut self, buf: &mut [u8]) -> io::Result { - match self.inner { - SpooledData::InMemory(ref mut cursor) => cursor.read(buf), - SpooledData::OnDisk(ref mut file) => file.read(buf), + match &mut self.inner { + SpooledData::InMemory(cursor) => cursor.read(buf), + SpooledData::OnDisk(file) => file.read(buf), } } } @@ -128,7 +128,7 @@ impl Write for SpooledTempFile { fn write(&mut self, buf: &[u8]) -> io::Result { // roll over to file if necessary let mut rolling = false; - if let SpooledData::InMemory(ref mut cursor) = self.inner { + if let SpooledData::InMemory(cursor) = &mut self.inner { rolling = cursor.position() as usize + buf.len() > self.max_size; } if rolling { @@ -136,26 +136,26 @@ impl Write for SpooledTempFile { } // write the bytes - match self.inner { - SpooledData::InMemory(ref mut cursor) => cursor.write(buf), - SpooledData::OnDisk(ref mut file) => file.write(buf), + match &mut self.inner { + SpooledData::InMemory(cursor) => cursor.write(buf), + SpooledData::OnDisk(file) => file.write(buf), } } #[inline] fn flush(&mut self) -> io::Result<()> { - match self.inner { - SpooledData::InMemory(ref mut cursor) => cursor.flush(), - SpooledData::OnDisk(ref mut file) => file.flush(), + match &mut self.inner { + SpooledData::InMemory(cursor) => cursor.flush(), + SpooledData::OnDisk(file) => file.flush(), } } } impl Seek for SpooledTempFile { fn seek(&mut self, pos: SeekFrom) -> io::Result { - match self.inner { - SpooledData::InMemory(ref mut cursor) => cursor.seek(pos), - SpooledData::OnDisk(ref mut file) => file.seek(pos), + match &mut self.inner { + SpooledData::InMemory(cursor) => cursor.seek(pos), + SpooledData::OnDisk(file) => file.seek(pos), } } } diff --git a/src/util.rs b/src/util.rs index c61082d50..d426ba3d7 100644 --- a/src/util.rs +++ b/src/util.rs @@ -15,16 +15,13 @@ fn tmpname(prefix: &OsStr, suffix: &OsStr, rand_len: usize) -> OsString { buf } -pub fn create_helper( +pub fn create_helper( base: &Path, prefix: &OsStr, suffix: &OsStr, random_len: usize, - mut f: F, -) -> io::Result -where - F: FnMut(PathBuf) -> io::Result, -{ + mut f: impl FnMut(PathBuf) -> io::Result, +) -> io::Result { let num_retries = if random_len != 0 { crate::NUM_RETRIES } else {