Wire SignableBuilder & ArtifactBuilder services in aggregator

Author: jpraynaud

mithril-aggregator/src/dependency.rs

@@ -6,6 +6,7 @@ use mithril_common::{
     digesters::{ImmutableDigester, ImmutableFileObserver},
     entities::{Certificate, Epoch, ProtocolParameters, SignerWithStake, StakeDistribution},
     era::{EraChecker, EraReader},
+    signable_builder::SignableBuilderService,
     store::StakeStorer,
     BeaconProvider,
 };
@@ -15,10 +16,11 @@ use std::{collections::HashMap, sync::Arc};
 use tokio::sync::{Mutex, RwLock};
 
 use crate::{
-    configuration::*, database::provider::StakePoolStore, signer_registerer::SignerRecorder,
-    CertificatePendingStore, CertificateStore, ProtocolParametersStore, ProtocolParametersStorer,
-    SignerRegisterer, SignerRegistrationRoundOpener, SingleSignatureStore, Snapshotter,
-    VerificationKeyStore, VerificationKeyStorer,
+    artifact_builder::ArtifactBuilderService, configuration::*, database::provider::StakePoolStore,
+    signer_registerer::SignerRecorder, CertificatePendingStore, CertificateStore,
+    ProtocolParametersStore, ProtocolParametersStorer, SignerRegisterer,
+    SignerRegistrationRoundOpener, SingleSignatureStore, Snapshotter, VerificationKeyStore,
+    VerificationKeyStorer,
 };
 use crate::{event_store::EventMessage, snapshot_stores::SnapshotStore};
 use crate::{event_store::TransmitterService, multi_signer::MultiSigner};
@@ -111,6 +113,12 @@ pub struct DependencyManager {
 
     /// Signer Recorder
     pub signer_recorder: Arc<dyn SignerRecorder>,
+
+    /// Signable Builder Service
+    pub signable_builder_service: Arc<SignableBuilderService>,
+
+    /// Artifact Builder Service
+    pub artifact_builder_service: Arc<ArtifactBuilderService>,
 }
 
 #[doc(hidden)]

mithril-aggregator/src/dependency_injection/builder.rs

@@ -19,6 +19,7 @@ use mithril_common::{
         adapters::{EraReaderAdapterBuilder, EraReaderDummyAdapter},
         EraChecker, EraMarker, EraReader, EraReaderAdapter, SupportedEra,
     },
+    signable_builder::{DummySignableBuilder, SignableBuilderService},
     store::adapter::{MemoryAdapter, SQLiteAdapter, StoreAdapter},
     BeaconProvider, BeaconProviderImpl,
 };
@@ -35,6 +36,7 @@ use tokio::{
 use warp::Filter;
 
 use crate::{
+    artifact_builder::{ArtifactBuilderService, DummyArtifactBuilder},
     configuration::{ExecutionEnvironment, LIST_SNAPSHOTS_MAX_ITEMS},
     database::provider::{
         CertificateStoreAdapter, EpochSettingStore, SignedEntityStoreAdapter,
@@ -156,6 +158,12 @@ pub struct DependenciesBuilder {
 
     /// Signer Recorder
     pub signer_recorder: Option<Arc<dyn SignerRecorder>>,
+
+    /// Signable Builder Service
+    pub signable_builder_service: Option<Arc<SignableBuilderService>>,
+
+    /// Artifact Builder Service
+    pub artifact_builder_service: Option<Arc<ArtifactBuilderService>>,
 }
 
 impl DependenciesBuilder {
@@ -193,6 +201,8 @@ impl DependenciesBuilder {
             stake_distribution_service: None,
             ticker_service: None,
             signer_recorder: None,
+            signable_builder_service: None,
+            artifact_builder_service: None,
         }
     }
 
@@ -968,6 +978,40 @@ impl DependenciesBuilder {
         Ok(self.signer_recorder.as_ref().cloned().unwrap())
     }
 
+    async fn build_signable_builder_service(&mut self) -> Result<Arc<SignableBuilderService>> {
+        let dummy_signable_builder = DummySignableBuilder::new();
+        let signable_builder_service =
+            Arc::new(SignableBuilderService::new(dummy_signable_builder));
+
+        Ok(signable_builder_service)
+    }
+
+    /// [SignableBuilderService] service
+    pub async fn get_signable_builder_service(&mut self) -> Result<Arc<SignableBuilderService>> {
+        if self.signable_builder_service.is_none() {
+            self.signable_builder_service = Some(self.build_signable_builder_service().await?);
+        }
+
+        Ok(self.signable_builder_service.as_ref().cloned().unwrap())
+    }
+
+    async fn build_artifact_builder_service(&mut self) -> Result<Arc<ArtifactBuilderService>> {
+        let dummy_artifact_builder = DummyArtifactBuilder::new();
+        let artifact_builder_service =
+            Arc::new(ArtifactBuilderService::new(dummy_artifact_builder));
+
+        Ok(artifact_builder_service)
+    }
+
+    /// [ArtifactBuilderService] service
+    pub async fn get_artifact_builder_service(&mut self) -> Result<Arc<ArtifactBuilderService>> {
+        if self.artifact_builder_service.is_none() {
+            self.artifact_builder_service = Some(self.build_artifact_builder_service().await?);
+        }
+
+        Ok(self.artifact_builder_service.as_ref().cloned().unwrap())
+    }
+
     /// Return an unconfigured [DependencyManager]
     pub async fn build_dependency_container(&mut self) -> Result<DependencyManager> {
         let dependency_manager = DependencyManager {
@@ -997,6 +1041,8 @@ impl DependenciesBuilder {
             api_version_provider: self.get_api_version_provider().await?,
             stake_distribution_service: self.get_stake_distribution_service().await?,
             signer_recorder: self.get_signer_recorder().await?,
+            signable_builder_service: self.get_signable_builder_service().await?,
+            artifact_builder_service: self.get_artifact_builder_service().await?,
         };
 
         Ok(dependency_manager)