add IO::remove_file_async

Author: MarinPostma

libsql-wal/src/io/mod.rs

@@ -1,6 +1,6 @@
-use std::io;
 use std::path::Path;
 use std::sync::Arc;
+use std::{future::Future, io};
 
 use chrono::{DateTime, Utc};
 use rand::{rngs::ThreadRng, thread_rng, Rng};
@@ -40,6 +40,8 @@ pub trait Io: Send + Sync + 'static {
             Uuid::from_u128(n)
         })
     }
+
+    fn remove_file_async(&self, path: &Path) -> impl Future<Output = io::Result<()>> + Send;
 }
 
 #[derive(Default, Debug, Clone, Copy)]
@@ -90,6 +92,10 @@ impl Io for StdIO {
     {
         f(&mut thread_rng())
     }
+
+    async fn remove_file_async(&self, path: &Path) -> io::Result<()> {
+        tokio::fs::remove_file(path).await
+    }
 }
 
 impl<T: Io> Io for Arc<T> {
@@ -133,6 +139,10 @@ impl<T: Io> Io for Arc<T> {
     {
         self.as_ref().with_rng(f)
     }
+
+    async fn remove_file_async(&self, path: &Path) -> io::Result<()> {
+        self.as_ref().remove_file_async(path).await
+    }
 }
 
 pub struct Inspect<W, F> {

libsql-wal/src/segment/current.rs

@@ -1015,6 +1015,13 @@ mod test {
             {
                 f(&mut rand::thread_rng())
             }
+
+            fn remove_file_async(
+                &self,
+                path: &std::path::Path,
+            ) -> impl std::future::Future<Output = io::Result<()>> + Send {
+                async move { std::fs::remove_file(path) }
+            }
         }
 
         let tmp = Arc::new(tempdir().unwrap());

libsql-wal/tests/flaky_fs.rs

@@ -144,10 +144,6 @@ impl Io for FlakyIo {
         todo!()
     }
 
-    fn uuid(&self) -> uuid::Uuid {
-        todo!()
-    }
-
     fn hard_link(&self, _src: &Path, _dst: &Path) -> std::io::Result<()> {
         todo!()
     }
@@ -158,6 +154,13 @@ impl Io for FlakyIo {
     {
         f(&mut self.rng.lock())
     }
+
+    fn remove_file_async(
+        &self,
+        path: &Path,
+    ) -> impl std::future::Future<Output = std::io::Result<()>> + Send {
+        async move { self.with_random_failure(|| std::fs::remove_file(path)) }
+    }
 }
 
 macro_rules! assert_not_corrupt {