Skip to content

Commit c264210

Browse files
committed
Wire EpochSettingStore in dependencies
1 parent fb61a27 commit c264210

File tree

1 file changed

+17
-25
lines changed
  • mithril-aggregator/src/dependency_injection

1 file changed

+17
-25
lines changed

mithril-aggregator/src/dependency_injection/builder.rs

Lines changed: 17 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ use warp::Filter;
3636

3737
use crate::{
3838
configuration::{ExecutionEnvironment, LIST_SNAPSHOTS_MAX_ITEMS},
39-
database::provider::StakePoolStore,
39+
database::provider::{EpochSettingStore, StakePoolStore},
4040
event_store::{EventMessage, EventStore, TransmitterService},
4141
http_server::routes::router,
4242
stake_distribution_service::{MithrilStakeDistributionService, StakeDistributionService},
@@ -495,32 +495,24 @@ impl DependenciesBuilder {
495495
}
496496

497497
async fn build_protocol_parameters_store(&mut self) -> Result<Arc<ProtocolParametersStore>> {
498-
let adapter: Box<dyn StoreAdapter<Key = Epoch, Record = ProtocolParameters>> = match self
499-
.configuration
500-
.environment
501-
{
502-
ExecutionEnvironment::Production => {
503-
let adapter =
504-
SQLiteAdapter::new("protocol_parameters", self.get_sqlite_connection().await?)
505-
.map_err(|e| DependenciesBuilderError::Initialization {
506-
message: "Cannot create SQLite adapter for ProtocolParametersStore."
498+
let adapter: Box<dyn StoreAdapter<Key = Epoch, Record = ProtocolParameters>> =
499+
match self.configuration.environment {
500+
ExecutionEnvironment::Production => {
501+
let adapter = EpochSettingStore::new(self.get_sqlite_connection().await?);
502+
503+
Box::new(adapter)
504+
}
505+
_ => {
506+
let adapter = MemoryAdapter::new(None).map_err(|e| {
507+
DependenciesBuilderError::Initialization {
508+
message: "Cannot create Memory adapter for ProtocolParametersStore."
507509
.to_string(),
508510
error: Some(e.into()),
509-
})?;
510-
511-
Box::new(adapter)
512-
}
513-
_ => {
514-
let adapter = MemoryAdapter::new(None).map_err(|e| {
515-
DependenciesBuilderError::Initialization {
516-
message: "Cannot create Memory adapter for ProtocolParametersStore."
517-
.to_string(),
518-
error: Some(e.into()),
519-
}
520-
})?;
521-
Box::new(adapter)
522-
}
523-
};
511+
}
512+
})?;
513+
Box::new(adapter)
514+
}
515+
};
524516

525517
Ok(Arc::new(ProtocolParametersStore::new(
526518
adapter,

0 commit comments

Comments
 (0)