Merge pull request #805 from input-output-hk/jpraynaud/804-fix-stake-distribution-compute-optimization

Avoid compute stake distribution multiple times per epoch

Author: jpraynaud

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "mithril-aggregator"
-version = "0.2.29"
+version = "0.2.30"
 description = "A Mithril Aggregator server"
 authors = { workspace = true }
 edition = { workspace = true }

mithril-aggregator/Cargo.toml

@@ -21,6 +21,7 @@ use crate::snapshotter::OngoingSnapshot;
 use crate::store::SingleSignatureStorer;
 use crate::CertificateCreator;
 use crate::MithrilCertificateCreator;
+use crate::RuntimeError;
 use crate::{DependencyManager, ProtocolError, SnapshotError};
 
 #[cfg(test)]
@@ -307,6 +308,29 @@ impl AggregatorRunnerTrait for AggregatorRunner {
         new_beacon: &Beacon,
     ) -> Result<(), Box<dyn StdError + Sync + Send>> {
         debug!("RUNNER: update stake distribution"; "beacon" => #?new_beacon);
+        let exists_stake_distribution = !self
+            .dependencies
+            .stake_store
+            .get_stakes(
+                self.dependencies
+                    .multi_signer
+                    .read()
+                    .await
+                    .get_current_beacon()
+                    .await
+                    .ok_or_else(|| {
+                        RuntimeError::keep_state("Current beacon is not available", None)
+                    })?
+                    .epoch
+                    .offset_to_recording_epoch(),
+            )
+            .await?
+            .unwrap_or_default()
+            .is_empty();
+        if exists_stake_distribution {
+            return Ok(());
+        }
+
         let stake_distribution = self
             .dependencies
             .chain_observer

mithril-aggregator/src/runtime/runner.rs

@@ -1,6 +1,6 @@
 [package]
 name = "mithril-signer"
-version = "0.2.20"
+version = "0.2.21"
 description = "A Mithril Signer"
 authors = { workspace = true }
 edition = { workspace = true }

mithril-signer/Cargo.toml

@@ -240,6 +240,17 @@ impl Runner for SignerRunner {
     ) -> Result<(), Box<dyn StdError + Sync + Send>> {
         debug!("RUNNER: update_stake_distribution");
 
+        let exists_stake_distribution = !self
+            .services
+            .stake_store
+            .get_stakes(epoch.offset_to_recording_epoch())
+            .await?
+            .unwrap_or_default()
+            .is_empty();
+        if exists_stake_distribution {
+            return Ok(());
+        }
+
         let stake_distribution = self
             .services
             .chain_observer