Skip to content

Commit

Permalink
Use sqlite buffer store everywhere and remove in memory version
Browse files Browse the repository at this point in the history
As it has no more use.
  • Loading branch information
Alenar committed Sep 13, 2024
1 parent f9bf269 commit d7e9138
Show file tree
Hide file tree
Showing 4 changed files with 11 additions and 182 deletions.
21 changes: 10 additions & 11 deletions mithril-aggregator/src/dependency_injection/builder.rs
Original file line number Diff line number Diff line change
Expand Up @@ -59,9 +59,9 @@ use crate::{
},
configuration::ExecutionEnvironment,
database::repository::{
CertificateRepository, EpochSettingStore, OpenMessageRepository, SignedEntityStore,
SignedEntityStorer, SignerRegistrationStore, SignerStore, SingleSignatureRepository,
StakePoolStore,
BufferedSingleSignatureRepository, CertificateRepository, EpochSettingStore,
OpenMessageRepository, SignedEntityStore, SignedEntityStorer, SignerRegistrationStore,
SignerStore, SingleSignatureRepository, StakePoolStore,
},
event_store::{EventMessage, EventStore, TransmitterService},
http_server::routes::router,
Expand All @@ -75,10 +75,9 @@ use crate::{
tools::{CExplorerSignerRetriever, GcpFileUploader, GenesisToolsDependency, SignersImporter},
AggregatorConfig, AggregatorRunner, AggregatorRuntime, CertificatePendingStore,
CompressedArchiveSnapshotter, Configuration, DependencyContainer, DumbSnapshotUploader,
DumbSnapshotter, InMemoryBufferedSingleSignatureStore, LocalSnapshotUploader,
MithrilSignerRegisterer, MultiSigner, MultiSignerImpl, ProtocolParametersStorer,
RemoteSnapshotUploader, SnapshotUploader, SnapshotUploaderType, Snapshotter,
SnapshotterCompressionAlgorithm, VerificationKeyStorer,
DumbSnapshotter, LocalSnapshotUploader, MithrilSignerRegisterer, MultiSigner, MultiSignerImpl,
ProtocolParametersStorer, RemoteSnapshotUploader, SnapshotUploader, SnapshotUploaderType,
Snapshotter, SnapshotterCompressionAlgorithm, VerificationKeyStorer,
};

const SQLITE_FILE: &str = "aggregator.sqlite3";
Expand Down Expand Up @@ -1428,10 +1427,10 @@ impl DependenciesBuilder {
let cardano_network = self.configuration.get_network().with_context(|| {
"Dependencies Builder can not get Cardano network while building the chain observer"
})?;
let sqlite_connection = self.get_sqlite_connection().await?;
let open_message_repository = self.get_open_message_repository().await?;
let single_signature_repository = Arc::new(SingleSignatureRepository::new(
self.get_sqlite_connection().await?,
));
let single_signature_repository =
Arc::new(SingleSignatureRepository::new(sqlite_connection.clone()));
let certificate_repository = self.get_certificate_repository().await?;
let certificate_verifier = self.get_certificate_verifier().await?;
let genesis_verifier = self.get_genesis_verifier().await?;
Expand All @@ -1456,7 +1455,7 @@ impl DependenciesBuilder {
Ok(Arc::new(BufferedCertifierService::new(
certifier,
multi_signer,
Arc::new(InMemoryBufferedSingleSignatureStore::default()),
Arc::new(BufferedSingleSignatureRepository::new(sqlite_connection)),
self.get_logger()?,
)))
}
Expand Down
3 changes: 1 addition & 2 deletions mithril-aggregator/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -53,8 +53,7 @@ pub use snapshotter::{
SnapshotterCompressionAlgorithm,
};
pub use store::{
CertificatePendingStore, InMemoryBufferedSingleSignatureStore, ProtocolParametersStorer,
VerificationKeyStore, VerificationKeyStorer,
CertificatePendingStore, ProtocolParametersStorer, VerificationKeyStore, VerificationKeyStorer,
};
pub use tools::{
CExplorerSignerRetriever, SignersImporter, SignersImporterPersister, SignersImporterRetriever,
Expand Down
167 changes: 0 additions & 167 deletions mithril-aggregator/src/store/buffered_single_signatures_store.rs

This file was deleted.

2 changes: 0 additions & 2 deletions mithril-aggregator/src/store/mod.rs
Original file line number Diff line number Diff line change
@@ -1,9 +1,7 @@
mod buffered_single_signatures_store;
mod pending_certificate_store;
mod protocol_parameters_store;
mod verification_key_store;

pub use buffered_single_signatures_store::InMemoryBufferedSingleSignatureStore;
pub use pending_certificate_store::CertificatePendingStore;
pub use protocol_parameters_store::ProtocolParametersStorer;
pub use verification_key_store::{VerificationKeyStore, VerificationKeyStorer};
Expand Down

0 comments on commit d7e9138

Please sign in to comment.