Implement restore_backup_to_database for BackupJob with integrity checks and streaming support

src/datastores/filesystem.rs

@@ -7,7 +7,7 @@ use std::{
 
 use regex::Regex;
 use sha2::{Digest, Sha256};
-use tokio::{fs::File as TokioFile, io::AsyncWrite};
+use tokio::{fs::File as TokioFile, io::{AsyncRead, AsyncWrite}};
 
 use crate::{datastores::DatastoreTrait, utils::backup_manager::DatabaseMetadata};
 
@@ -164,6 +164,19 @@ impl DatastoreTrait for FilesystemDatastore {
 
     Ok(Box::new(file))
   }
+
+  async fn open_read_stream(
+    &self,
+    object_name: &str,
+  ) -> Result<Box<dyn AsyncRead + Unpin + Send>, String> {
+    let full_path = self.base_path.join(object_name);
+
+    let file = TokioFile::open(full_path)
+      .await
+      .map_err(|e| format!("Failed to open file: {}", e))?;
+
+    Ok(Box::new(file))
+  }
 }
 
 #[cfg(test)]

src/datastores/mod.rs

@@ -1,6 +1,6 @@
 use std::{io::Error, path::Path};
 
-use tokio::io::AsyncWrite;
+use tokio::io::{AsyncRead, AsyncWrite};
 
 mod filesystem;
 mod s3;
@@ -27,6 +27,10 @@ pub trait DatastoreTrait {
     &self,
     object_name: &str,
   ) -> Result<Box<dyn AsyncWrite + Unpin + Send>, String>;
+  async fn open_read_stream(
+    &self,
+    object_name: &str,
+  ) -> Result<Box<dyn AsyncRead + Unpin + Send>, String>;
 }
 
 #[derive(Debug, PartialEq, Clone)]
@@ -79,4 +83,14 @@ impl Datastore {
       Datastore::S3(store) => store.open_write_stream(object_name).await,
     }
   }
+
+  pub async fn open_read_stream(
+    &self,
+    object_name: &str,
+  ) -> Result<Box<dyn AsyncRead + Unpin + Send>, String> {
+    match self {
+      Datastore::FileSystem(store) => store.open_read_stream(object_name).await,
+      Datastore::S3(store) => store.open_read_stream(object_name).await,
+    }
+  }
 }

src/datastores/s3.rs

@@ -3,7 +3,7 @@ use std::{
   path::{Path, PathBuf},
 };
 
-use tokio::io::AsyncWrite;
+use tokio::io::{AsyncRead, AsyncWrite};
 
 use crate::datastores::DatastoreTrait;
 
@@ -52,4 +52,11 @@ impl DatastoreTrait for S3Datastore {
   ) -> Result<Box<dyn AsyncWrite + Unpin + Send>, String> {
     todo!()
   }
+
+  async fn open_read_stream(
+    &self,
+    object_name: &str,
+  ) -> Result<Box<dyn AsyncRead + Unpin + Send>, String> {
+    todo!()
+  }
 }