Add store and load methods

Author: tbro

sequencer/api/migrations/sqlite/V302__epoch_drb_and_root.sql

@@ -2,4 +2,5 @@ CREATE TABLE epoch_drb_and_root (
     epoch BIGINT PRIMARY KEY,
     drb_result BLOB,
     block_header BLOB
-);
+    stake BLOB
+);

sequencer/src/persistence/sql.rs

@@ -9,6 +9,7 @@ use derive_more::derive::{From, Into};
 use espresso_types::{
     parse_duration, parse_size, upgrade_commitment_map,
     v0::traits::{EventConsumer, PersistenceOptions, SequencerPersistence, StateCatchup},
+    v0_3::CombinedStakeTable,
     BackoffParams, BlockMerkleTree, FeeMerkleTree, Leaf, Leaf2, NetworkConfig, Payload,
 };
 use futures::stream::StreamExt;
@@ -1018,6 +1019,42 @@ impl SequencerPersistence for Persistence {
         Ok(())
     }
 
+    async fn load_stake(&self, epoch: EpochNumber) -> anyhow::Result<Option<CombinedStakeTable>> {
+        let result = self
+            .db
+            .read()
+            .await?
+            .fetch_optional(
+                query("SELECT stake FROM epoch_drb_and_root WHERE epoch = ?1")
+                    .bind(epoch.u64() as i64),
+            )
+            .await?;
+
+        result
+            .map(|row| {
+                let bytes: Vec<u8> = row.get("stake");
+                anyhow::Result::<_>::Ok(bincode::deserialize(&bytes)?)
+            })
+            .transpose()
+    }
+    async fn store_stake(
+        &self,
+        epoch: EpochNumber,
+        stake: CombinedStakeTable,
+    ) -> anyhow::Result<()> {
+        let stake_table_bytes = bincode::serialize(&stake).context("serializing stake table")?;
+
+        let mut tx = self.db.write().await?;
+        tx.upsert(
+            "epoch_drb_and_root",
+            ["epoch", "drb_result"],
+            ["epoch"],
+            [(epoch.u64() as i64, stake_table_bytes)],
+        )
+        .await?;
+        tx.commit().await
+    }
+
     async fn load_latest_acted_view(&self) -> anyhow::Result<Option<ViewNumber>> {
         Ok(self
             .db

types/src/v0/traits.rs

@@ -31,7 +31,8 @@ use itertools::Itertools;
 use serde::{de::DeserializeOwned, Serialize};
 
 use super::{
-    impls::NodeState, utils::BackoffParams, EpochCommittees, EpochVersion, Leaf, SequencerVersions,
+    impls::NodeState, utils::BackoffParams, v0_3::CombinedStakeTable, EpochCommittees,
+    EpochVersion, Leaf, SequencerVersions,
 };
 use crate::{
     v0::impls::ValidatedState, v0_99::ChainConfig, BlockMerkleTree, Event, FeeAccount,
@@ -539,6 +540,12 @@ pub trait SequencerPersistence: Sized + Send + Sync + Clone + 'static {
         &self,
     ) -> anyhow::Result<Option<UpgradeCertificate<SeqTypes>>>;
     async fn load_start_epoch_info(&self) -> anyhow::Result<Vec<InitializerEpochInfo<SeqTypes>>>;
+    async fn load_stake(&self, epoch: EpochNumber) -> anyhow::Result<Option<CombinedStakeTable>>;
+    async fn store_stake(
+        &self,
+        epoch: EpochNumber,
+        stake: CombinedStakeTable,
+    ) -> anyhow::Result<()>;
 
     /// Load the latest known consensus state.
     ///