Add snapshot list message adapter to aggregator

Author: jpraynaud

mithril-aggregator/src/http_server/routes/snapshot_routes.rs

@@ -57,7 +57,7 @@ fn snapshot_digest(
 mod handlers {
     use crate::http_server::routes::reply;
     use crate::http_server::SERVER_BASE_PATH;
-    use crate::message_adapters::ToSnapshotMessageAdapter;
+    use crate::message_adapters::{ToSnapshotListMessageAdapter, ToSnapshotMessageAdapter};
     use crate::{Configuration, SnapshotStore};
     use slog_scope::{debug, warn};
     use std::convert::Infallible;
@@ -72,7 +72,10 @@ mod handlers {
         debug!("⇄ HTTP SERVER: snapshots");
 
         match snapshot_store.list_snapshots().await {
-            Ok(snapshots) => Ok(reply::json(&snapshots, StatusCode::OK)),
+            Ok(snapshots) => Ok(reply::json(
+                &ToSnapshotListMessageAdapter::adapt(snapshots),
+                StatusCode::OK,
+            )),
             Err(err) => {
                 warn!("snapshots::error"; "error" => ?err);
                 Ok(reply::internal_server_error(err.to_string()))

mithril-aggregator/src/message_adapters/mod.rs

@@ -3,11 +3,13 @@ mod from_register_signer;
 mod to_certificate_message;
 mod to_certificate_pending_message;
 mod to_epoch_settings_message;
+mod to_snapshot_list_message;
 mod to_snasphot_message;
 
 pub use from_register_signature::FromRegisterSingleSignatureAdapter;
 pub use from_register_signer::FromRegisterSignerAdapter;
 pub use to_certificate_message::ToCertificateMessageAdapter;
 pub use to_certificate_pending_message::ToCertificatePendingMessageAdapter;
 pub use to_epoch_settings_message::ToEpochSettingsMessageAdapter;
+pub use to_snapshot_list_message::ToSnapshotListMessageAdapter;
 pub use to_snasphot_message::ToSnapshotMessageAdapter;

mithril-aggregator/src/message_adapters/to_snapshot_list_message.rs

@@ -0,0 +1,38 @@
+use mithril_common::entities::Snapshot;
+use mithril_common::messages::{SnapshotListItemMessage, SnapshotListMessage};
+
+/// Adapter to convert a list of [Snapshot] to [SnapshotListMessage] instances
+pub struct ToSnapshotListMessageAdapter;
+
+impl ToSnapshotListMessageAdapter {
+    /// Method to trigger the conversion
+    pub fn adapt(snapshots: Vec<Snapshot>) -> SnapshotListMessage {
+        snapshots
+            .into_iter()
+            .map(|snapshot| SnapshotListItemMessage {
+                digest: snapshot.digest,
+                beacon: snapshot.beacon,
+                certificate_hash: snapshot.certificate_hash,
+                size: snapshot.size,
+                created_at: snapshot.created_at,
+                locations: snapshot.locations,
+            })
+            .collect()
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use mithril_common::test_utils::fake_data;
+
+    use super::*;
+
+    #[test]
+    fn adapt_ok() {
+        let mut snapshot = fake_data::snapshots(1)[0].to_owned();
+        snapshot.digest = "digest123".to_string();
+        let snapshot_list_message = ToSnapshotListMessageAdapter::adapt(vec![snapshot]);
+
+        assert_eq!("digest123".to_string(), snapshot_list_message[0].digest);
+    }
+}