add extra logs on diskspace when downloading

Author: turmelclem

Cargo.lock

@@ -236,6 +236,17 @@ impl PreparedCardanoDbV2Download {
     ) -> MithrilResult<()> {
         progress_printer.report_step(step_number, "Checking local disk info…")?;
 
+        let free_space = fs2::available_space(&restoration_options.db_dir)?;
+        let total_size_uncompressed =
+            Self::compute_required_disk_space_for_snapshot(cardano_db, restoration_options);
+
+        let message = format!(
+            "free space is {} bytes, required space for the snapshot is {} bytes",
+            free_space, total_size_uncompressed
+        );
+
+        progress_printer.report_step(step_number, &message)?;
+
         CardanoDbDownloadChecker::ensure_dir_exist(&restoration_options.db_dir)?;
         if let Err(e) = CardanoDbDownloadChecker::check_prerequisites_for_uncompressed_data(
             &restoration_options.db_dir,

mithril-client-cli/src/commands/cardano_db/download/v2.rs

@@ -80,6 +80,7 @@ thiserror = { workspace = true }
 tokio = { workspace = true }
 uuid = { version = "1.21.0", features = ["v4"] }
 zstd = { version = "0.13.3", optional = true }
+fs2 = "0.4.3"
 
 [target.'cfg(not(target_family = "wasm"))'.dependencies]
 mithril-aggregator-discovery = { path = "../internal/mithril-aggregator-discovery", version = "0.1" }

mithril-client/Cargo.toml

@@ -48,6 +48,16 @@ impl DownloadTask {
         let mut download_succeeded = false;
 
         for location_to_try in &self.locations_to_try {
+            //print a log with space disk of target dir before downloading
+            let free_space = fs2::available_space(&target_dir)?;
+            slog::warn!(
+                logger,
+                "Attempting to download {} from location {:?} (available disk space: {} bytes)",
+                self.name(),
+                location_to_try.file_downloader_uri,
+                free_space
+            );
+
             let downloaded = location_to_try
                 .file_downloader
                 .download_unpack(