Merge pull request #703 from input-output-hk/jpraynaud/698-add-snapshot-list-message

Add snapshot list message and adapter

Author: jpraynaud

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "mithril-aggregator"
-version = "0.2.7"
+version = "0.2.8"
 description = "A Mithril Aggregator server"
 authors = { workspace = true }
 edition = { workspace = true }

mithril-aggregator/Cargo.toml

@@ -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/http_server/routes/snapshot_routes.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/mod.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);
+    }
+}

mithril-aggregator/src/message_adapters/to_snapshot_list_message.rs

@@ -1,6 +1,6 @@
 [package]
 name = "mithril-client"
-version = "0.2.4"
+version = "0.2.5"
 description = "A Mithril Client"
 authors = { workspace = true }
 edition = { workspace = true }

mithril-client/Cargo.toml

@@ -16,11 +16,13 @@ use thiserror::Error;
 use mithril_common::{
     certificate_chain::{CertificateRetriever, CertificateRetrieverError},
     entities::{Certificate, Snapshot},
-    messages::{CertificateMessage, SnapshotMessage},
+    messages::{CertificateMessage, SnapshotListMessage, SnapshotMessage},
     MITHRIL_API_VERSION,
 };
 
-use crate::{FromCertificateMessageAdapter, FromSnapshotMessageAdapter};
+use crate::{
+    FromCertificateMessageAdapter, FromSnapshotListMessageAdapter, FromSnapshotMessageAdapter,
+};
 
 /// [AggregatorHandler] related errors.
 #[derive(Error, Debug)]
@@ -173,8 +175,8 @@ impl AggregatorHandler for AggregatorHTTPClient {
 
         match response {
             Ok(response) => match response.status() {
-                StatusCode::OK => match response.json::<Vec<Snapshot>>().await {
-                    Ok(snapshots) => Ok(snapshots),
+                StatusCode::OK => match response.json::<SnapshotListMessage>().await {
+                    Ok(snapshots) => Ok(FromSnapshotListMessageAdapter::adapt(snapshots)),
                     Err(err) => Err(AggregatorHandlerError::JsonParseFailed(err.to_string())),
                 },
                 StatusCode::PRECONDITION_FAILED => Err(self.handle_api_error(&response)),

mithril-client/src/aggregator.rs

@@ -19,7 +19,9 @@ mod runtime;
 
 pub use aggregator::{AggregatorHTTPClient, AggregatorHandler, AggregatorHandlerError};
 pub use entities::Config;
-pub use message_adapters::{FromCertificateMessageAdapter, FromSnapshotMessageAdapter};
+pub use message_adapters::{
+    FromCertificateMessageAdapter, FromSnapshotListMessageAdapter, FromSnapshotMessageAdapter,
+};
 pub use runtime::{Runtime, RuntimeError};
 
 pub use runtime::convert_to_field_items;

mithril-client/src/lib.rs

@@ -0,0 +1,37 @@
+use mithril_common::entities::Snapshot;
+use mithril_common::messages::SnapshotListMessage;
+
+/// Adapter to convert [SnapshotListMessage] to [SnapshotList] instances
+pub struct FromSnapshotListMessageAdapter;
+
+impl FromSnapshotListMessageAdapter {
+    /// Method to trigger the conversion
+    pub fn adapt(snapshot_list_message: SnapshotListMessage) -> Vec<Snapshot> {
+        snapshot_list_message
+            .into_iter()
+            .map(|snapshot_list_item_message| Snapshot {
+                digest: snapshot_list_item_message.digest,
+                beacon: snapshot_list_item_message.beacon,
+                certificate_hash: snapshot_list_item_message.certificate_hash,
+                size: snapshot_list_item_message.size,
+                created_at: snapshot_list_item_message.created_at,
+                locations: snapshot_list_item_message.locations,
+            })
+            .collect()
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use mithril_common::messages::SnapshotListItemMessage;
+
+    use super::*;
+
+    #[test]
+    fn adapt_ok() {
+        let snapshot_list_message: SnapshotListMessage = vec![SnapshotListItemMessage::dummy()];
+        let snapshot_list = FromSnapshotListMessageAdapter::adapt(snapshot_list_message.clone());
+
+        assert_eq!(snapshot_list_message[0].digest, snapshot_list[0].digest);
+    }
+}

mithril-client/src/message_adapters/from_snapshot_list_message.rs

@@ -1,5 +1,7 @@
 mod from_certificate_message_adapter;
+mod from_snapshot_list_message;
 mod from_snapshot_message;
 
 pub use from_certificate_message_adapter::FromCertificateMessageAdapter;
+pub use from_snapshot_list_message::FromSnapshotListMessageAdapter;
 pub use from_snapshot_message::FromSnapshotMessageAdapter;