mithril_aggregator/dependency_injection/builder/support/
upkeep.rs

1use std::sync::Arc;
2
3use crate::database::repository::SignerRegistrationStore;
4use crate::dependency_injection::{DependenciesBuilder, Result};
5use crate::get_dependency;
6use crate::services::{AggregatorUpkeepService, EpochPruningTask, UpkeepService};
7
8impl DependenciesBuilder {
9    /// Return a [EpochPruningTask] instance
10    pub async fn get_signer_registration_pruning_task(
11        &mut self,
12    ) -> Result<Arc<dyn EpochPruningTask>> {
13        Ok(Arc::new(SignerRegistrationStore::new(
14            self.get_sqlite_connection().await?,
15            self.configuration.safe_epoch_retention_limit(),
16        )))
17    }
18
19    async fn build_upkeep_service(&mut self) -> Result<Arc<dyn UpkeepService>> {
20        let stake_pool_pruning_task = self.get_stake_store().await?;
21        let epoch_settings_pruning_task = self.get_epoch_settings_store().await?;
22        let signer_registration_pruning_task = self.get_signer_registration_pruning_task().await?;
23
24        let upkeep_service = Arc::new(AggregatorUpkeepService::new(
25            self.get_sqlite_connection().await?,
26            self.get_sqlite_connection_cardano_transaction_pool().await?,
27            self.get_event_store_sqlite_connection().await?,
28            self.get_signed_entity_type_lock().await?,
29            vec![
30                stake_pool_pruning_task,
31                epoch_settings_pruning_task,
32                signer_registration_pruning_task,
33            ],
34            self.root_logger(),
35        ));
36
37        Ok(upkeep_service)
38    }
39
40    /// Get the [UpkeepService] instance
41    pub async fn get_upkeep_service(&mut self) -> Result<Arc<dyn UpkeepService>> {
42        get_dependency!(self.upkeep_service)
43    }
44}

mithril_aggregator/dependency_injection/builder/support/upkeep.rs

mithril_aggregator/dependency_injection/builder/support/
upkeep.rs