add destroy method to segment

Author: MarinPostma

libsql-wal/src/segment/mod.rs

@@ -21,6 +21,7 @@ use zerocopy::AsBytes;
 use crate::error::{Error, Result};
 use crate::io::buf::IoBufMut;
 use crate::io::FileExt;
+use crate::io::Io;
 use crate::LIBSQL_MAGIC;
 use crate::LIBSQL_PAGE_SIZE;
 
@@ -167,6 +168,8 @@ pub trait Segment: Send + Sync + 'static {
     async fn read_frame_offset_async<B>(&self, offset: u32, buf: B) -> (B, Result<()>)
     where
         B: IoBufMut + Send + 'static;
+
+    fn destroy<IO: Io>(&self, io: &IO) -> impl Future<Output = ()>;
 }
 
 impl<T: Segment> Segment for Arc<T> {
@@ -208,6 +211,10 @@ impl<T: Segment> Segment for Arc<T> {
     fn size_after(&self) -> u32 {
         self.as_ref().size_after()
     }
+
+    fn destroy<IO: Io>(&self, io: &IO) -> impl Future<Output = ()> {
+        self.as_ref().destroy(io)
+    }
 }
 
 #[repr(C)]

libsql-wal/src/segment/sealed.rs

@@ -183,6 +183,14 @@ where
     fn size_after(&self) -> u32 {
         self.header().size_after()
     }
+
+    fn destroy<IO: crate::io::Io>(&self, io: &IO) -> impl std::future::Future<Output = ()> {
+        async move {
+            if let Err(e) = io.remove_file_async(&self.path).await {
+                tracing::error!("failed to remove segment file {:?}: {e}", self.path);
+            }
+        }
+    }
 }
 
 impl<F: FileExt> SealedSegment<F> {

libsql-wal/src/storage/job.rs

@@ -421,6 +421,10 @@ mod test {
             {
                 todo!()
             }
+
+            fn destroy<IO: Io>(&self, _io: &IO) -> impl std::future::Future<Output = ()> {
+                async move { todo!() }
+            }
         }
 
         struct TestBackend;