Add API version provider dependency in aggregator

input-output-hk · Mar 17, 2023 · 053f46e · 053f46e
1 parent 01f780f
commit 053f46e
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/mithril-aggregator/src/command_args.rs b/mithril-aggregator/src/command_args.rs
@@ -18,6 +18,7 @@ use tokio::{
 };
 
 use mithril_common::{
+    api::APIVersionProvider,
     certificate_chain::MithrilCertificateVerifier,
     chain_observer::{CardanoCliRunner, ChainObserver},
     crypto_helper::{
@@ -446,6 +447,8 @@ impl ServeCommand {
         let event_transmitter = Arc::new(TransmitterService::new(tx));
         let mut event_store = event_store::EventStore::new(rx);
 
+        let api_version_provider = Arc::new(APIVersionProvider::new(era_checker.clone()));
+
         // Init dependency manager
         let dependency_manager = DependencyManager {
             config: config.clone(),
@@ -470,6 +473,7 @@ impl ServeCommand {
             era_checker: era_checker.clone(),
             era_reader: era_reader.clone(),
             event_transmitter,
+            api_version_provider,
         };
         let dependency_manager = Arc::new(dependency_manager);
 

diff --git a/mithril-aggregator/src/dependency.rs b/mithril-aggregator/src/dependency.rs
@@ -1,4 +1,5 @@
 use mithril_common::{
+    api::APIVersionProvider,
     certificate_chain::CertificateVerifier,
     chain_observer::ChainObserver,
     crypto_helper::ProtocolGenesisVerifier,
@@ -93,6 +94,9 @@ pub struct DependencyManager {
 
     /// Event Transmitter Service
     pub event_transmitter: Arc<TransmitterService<EventMessage>>,
+
+    /// API Version provider
+    pub api_version_provider: Arc<APIVersionProvider>,
 }
 
 #[doc(hidden)]
@@ -271,6 +275,7 @@ pub mod tests {
         SingleSignatureStore, SnapshotUploaderType, VerificationKeyStore,
     };
     use mithril_common::{
+        api::APIVersionProvider,
         certificate_chain::MithrilCertificateVerifier,
         chain_observer::FakeObserver,
         crypto_helper::{key_encode_hex, ProtocolGenesisSigner},
@@ -374,6 +379,8 @@ pub mod tests {
             Arc::new(TransmitterService::new(tx))
         };
 
+        let api_version_provider = Arc::new(APIVersionProvider::new(era_checker.clone()));
+
         let dependency_manager = DependencyManager {
             config,
             snapshot_store,
@@ -397,6 +404,7 @@ pub mod tests {
             era_checker,
             era_reader,
             event_transmitter,
+            api_version_provider,
         };
 
         let config = AggregatorConfig::new(

diff --git a/mithril-aggregator/tests/test_extensions/dependency.rs b/mithril-aggregator/tests/test_extensions/dependency.rs
@@ -5,6 +5,7 @@ use mithril_aggregator::{
     MultiSignerImpl, ProtocolParametersStore, SingleSignatureStore, SnapshotUploaderType,
     VerificationKeyStore,
 };
+use mithril_common::api::APIVersionProvider;
 use mithril_common::certificate_chain::MithrilCertificateVerifier;
 use mithril_common::chain_observer::FakeObserver;
 use mithril_common::crypto_helper::{key_encode_hex, ProtocolGenesisSigner};
@@ -118,6 +119,8 @@ pub async fn initialize_dependencies(
     let (tx, rx) = tokio::sync::mpsc::unbounded_channel();
     let event_transmitter = Arc::new(TransmitterService::new(tx));
 
+    let api_version_provider = Arc::new(APIVersionProvider::new(era_checker.clone()));
+
     let dependency_manager = DependencyManager {
         config,
         snapshot_store,
@@ -141,6 +144,7 @@ pub async fn initialize_dependencies(
         era_checker,
         era_reader,
         event_transmitter,
+        api_version_provider,
     };
 
     let config = AggregatorConfig::new(