Use network parameter in archive filename

Author: Alenar

mithril-aggregator/src/http_server.rs

@@ -413,7 +413,7 @@ mod handlers {
         let snapshot_store = snapshot_store.read().await;
         match snapshot_store.get_snapshot_details(digest).await {
             Ok(Some(snapshot)) => {
-                let filename = format!("testnet.{}.tar.gz", snapshot.digest);
+                let filename = format!("{}.{}.tar.gz", config.network, snapshot.digest);
                 let snapshot_uri = format!(
                     "{}/{}/snapshot_download/{}",
                     config.server_url, SERVER_BASE_PATH, filename

mithril-aggregator/src/main.rs

@@ -141,6 +141,7 @@ async fn main() -> Result<(), String> {
     let handle = tokio::spawn(async move {
         let runtime = AggregatorRuntime::new(
             args.snapshot_interval * 1000,
+            config.network.clone(),
             config.db_directory.clone(),
             snapshot_directory,
             beacon_store.clone(),

mithril-aggregator/src/runtime.rs

@@ -48,6 +48,9 @@ pub struct AggregatorRuntime {
     /// Interval between each snapshot, in seconds
     interval: u32,
 
+    /// Cardano network
+    network: String,
+
     /// DB directory to snapshot
     db_directory: PathBuf,
 
@@ -71,6 +74,7 @@ impl AggregatorRuntime {
     /// AggregatorRuntime factory
     pub fn new(
         interval: u32,
+        network: String,
         db_directory: PathBuf,
         snapshot_directory: PathBuf,
         beacon_store: BeaconStoreWrapper,
@@ -80,6 +84,7 @@ impl AggregatorRuntime {
     ) -> Self {
         Self {
             interval,
+            network,
             db_directory,
             snapshot_directory,
             beacon_store,
@@ -119,7 +124,8 @@ impl AggregatorRuntime {
 
                 match self.manage_trigger_snapshot(&message, &beacon).await {
                     Ok(true) => {
-                        let snapshot_name = format!("testnet.{}.tar.gz", &digest_result.digest);
+                        let snapshot_name =
+                            format!("{}.{}.tar.gz", self.network, &digest_result.digest);
                         let snapshot_path = snapshotter.snapshot(&snapshot_name)?;
 
                         let uploaded_snapshot_location = self