Write extraction marker for mongo

Author: palfrey

nativelink-store/tests/mongo_runner/mod.rs

@@ -5,6 +5,7 @@ pub(crate) mod extractor;
 pub(crate) mod process;
 
 use core::time::Duration;
+use std::fs;
 use std::path::PathBuf;
 use std::sync::{Condvar, LazyLock, Mutex};
 
@@ -110,9 +111,10 @@ impl MongoEmbedded {
         callback(InitStatus::ValidatingInstallation);
 
         let extract_target = self.extract_path.join(self.version.as_str());
+        let extract_marker = extract_target.join("extracted.marker");
 
-        if extract_target.exists() {
-            info!(?extract_target, "Already have Mongo downloaded");
+        if extract_marker.exists() {
+            info!(?extract_marker, "Already have Mongo downloaded");
         } else {
             let existing_downloader = {
                 let mut lock = CURRENTLY_DOWNLOADING.0.lock().unwrap();
@@ -128,7 +130,7 @@ impl MongoEmbedded {
             };
             if !existing_downloader {
                 if !self.download_path.exists() {
-                    std::fs::create_dir_all(&self.download_path)
+                    fs::create_dir_all(&self.download_path)
                         .err_tip(|| format!("Making {}", self.download_path.display()))?;
                 }
                 callback(InitStatus::Downloading);
@@ -153,6 +155,7 @@ impl MongoEmbedded {
                     )
                 })?;
                 info!("Downloaded, notifying");
+                fs::write(extract_marker, "")?;
                 let lock = CURRENTLY_DOWNLOADING.0.lock().unwrap();
                 CURRENTLY_DOWNLOADING.1.notify_all();
                 drop(lock);