feature(aggregator): implement local network configuration provider

Author: turmelclem

mithril-aggregator/src/dependency_injection/builder/enablers/epoch.rs

@@ -1,25 +1,28 @@
+use mithril_protocol_config::http_client::http_impl::HttpMithrilNetworkConfigurationProvider;
 use std::sync::Arc;
 use tokio::sync::RwLock;
 
+use mithril_protocol_config::interface::MithrilNetworkConfigurationProvider;
+
 use crate::dependency_injection::{DependenciesBuilder, EpochServiceWrapper, Result};
 use crate::get_dependency;
-use crate::services::{EpochServiceDependencies, MithrilEpochService};
+use crate::services::{
+    EpochServiceDependencies, LocalMithrilNetworkConfigurationProvider, MithrilEpochService,
+};
+
 impl DependenciesBuilder {
     async fn build_epoch_service(&mut self) -> Result<EpochServiceWrapper> {
         let verification_key_store = self.get_verification_key_store().await?;
         let epoch_settings_storer = self.get_epoch_settings_store().await?;
         let chain_observer = self.get_chain_observer().await?;
         let era_checker = self.get_era_checker().await?;
         let stake_store = self.get_stake_store().await?;
-        //remove conf usage for epoch_settings and allowed_discriminants and send mithril_network_configuration instread
-        let epoch_settings = self.configuration.get_epoch_settings_configuration();
         let allowed_discriminants = self
             .configuration
             .compute_allowed_signed_entity_types_discriminants()?;
 
-        let network_configuration_provider = self.get_mithril_network_configuration_provider();
         let epoch_service = Arc::new(RwLock::new(MithrilEpochService::new(
-            network_configuration_provider,
+            self.get_mithril_network_configuration_provider().await?,
             EpochServiceDependencies::new(
                 epoch_settings_storer,
                 verification_key_store,
@@ -38,4 +41,45 @@ impl DependenciesBuilder {
     pub async fn get_epoch_service(&mut self) -> Result<EpochServiceWrapper> {
         get_dependency!(self.epoch_service)
     }
+
+    async fn build_mithril_network_configuration_provider(
+        &mut self,
+    ) -> Result<Arc<dyn MithrilNetworkConfigurationProvider>> {
+        let network_configuration_provider: Arc<dyn MithrilNetworkConfigurationProvider> =
+            if self.configuration.is_follower_aggregator() {
+                Arc::new(HttpMithrilNetworkConfigurationProvider::new(
+                    self.get_leader_aggregator_client().await?,
+                ))
+            } else {
+                Arc::new(LocalMithrilNetworkConfigurationProvider::new(
+                    self.configuration.get_epoch_settings_configuration(),
+                    self.configuration
+                        .compute_allowed_signed_entity_types_discriminants()?,
+                    self.get_epoch_settings_store().await?,
+                ))
+            };
+
+        Ok(network_configuration_provider)
+    }
+
+    /// [MithrilNetworkConfigurationProvider][mithril_protocol_config::interface::MithrilNetworkConfigurationProvider] service
+    pub async fn get_mithril_network_configuration_provider(
+        &mut self,
+    ) -> Result<Arc<dyn MithrilNetworkConfigurationProvider>> {
+        get_dependency!(self.mithril_network_configuration_provider)
+    }
+}
+
+// Todo remove after rebase with shared aggregator client:
+#[async_trait::async_trait]
+impl mithril_protocol_config::http_client::http_impl::ProtocolConfigurationRetrieverFromAggregator
+    for crate::services::AggregatorHTTPClient
+{
+    /// Retrieves protocol configuration for a given epoch from the aggregator
+    async fn retrieve_protocol_configuration(
+        &self,
+        _epoch: mithril_common::entities::Epoch,
+    ) -> mithril_common::StdResult<mithril_common::messages::ProtocolConfigurationMessage> {
+        unimplemented!()
+    }
 }

mithril-aggregator/src/dependency_injection/builder/enablers/mithril_network_configuration.rs

@@ -3,5 +3,4 @@
 mod cardano_node;
 mod epoch;
 mod misc;
-mod mithril_network_configuration;
 mod ticker;

mithril-aggregator/src/dependency_injection/builder/enablers/mod.rs

@@ -15,6 +15,7 @@ mod certificate_chain_synchronizer;
 mod certifier;
 mod epoch_service;
 mod message;
+mod network_configuration_provider;
 mod prover;
 mod signable_builder;
 mod signature_consumer;
@@ -32,6 +33,7 @@ pub use certificate_chain_synchronizer::*;
 pub use certifier::*;
 pub use epoch_service::*;
 pub use message::*;
+pub use network_configuration_provider::*;
 pub use prover::*;
 pub use signable_builder::*;
 pub use signature_consumer::*;