Keep track of chain PoS power

ziesha-network · May 23, 2023 · 358cba4 · 358cba4
1 parent cc3a592
commit 358cba4
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 0 deletions.
diff --git a/src/blockchain/mod.rs b/src/blockchain/mod.rs
@@ -61,6 +61,7 @@ pub trait Blockchain<K: KvStore> {
     fn database(&self) -> &K;
     fn database_mut(&mut self) -> &mut K;
     fn epoch_slot(&self, timestamp: u32) -> (u32, u32);
+    fn get_power(&self) -> Result<f64, BlockchainError>;
     fn get_mpn_account_count(&self) -> Result<u64, BlockchainError>;
     fn get_mpn_account_indices(&self, addr: MpnAddress) -> Result<Vec<u64>, BlockchainError>;
     fn get_stake(&self, addr: Address) -> Result<Amount, BlockchainError>;
@@ -609,6 +610,13 @@ impl<K: KvStore> Blockchain<K> for KvStoreChain<K> {
         Ok(None)
     }
 
+    fn get_power(&self) -> Result<f64, BlockchainError> {
+        Ok(match self.database.get(keys::power())? {
+            Some(b) => b.try_into()?,
+            None => 0.into(),
+        })
+    }
+
     fn get_stake(&self, addr: Address) -> Result<Amount, BlockchainError> {
         Ok(match self.database.get(keys::stake(&addr))? {
             Some(b) => b.try_into()?,

diff --git a/src/blockchain/ops/apply_block.rs b/src/blockchain/ops/apply_block.rs
@@ -6,6 +6,7 @@ pub fn apply_block<K: KvStore>(
 ) -> Result<(), BlockchainError> {
     let (ops, _) = chain.isolated(|chain| {
         let curr_height = chain.get_height()?;
+        let mut curr_pow = chain.get_power()?;
 
         if let Some(height_limit) = chain.config.testnet_height_limit {
             if block.header.number >= height_limit {
@@ -26,6 +27,7 @@ pub fn apply_block<K: KvStore>(
         if !is_genesis {
             if chain.config.check_validator {
                 if let Some(proof) = block.header.proof_of_stake.proof.clone() {
+                    curr_pow += 1f64 / ((proof.attempt + 1) as f64);
                     if !chain.is_validator(
                         block.header.proof_of_stake.timestamp,
                         block.header.proof_of_stake.validator.clone(),
@@ -134,6 +136,7 @@ pub fn apply_block<K: KvStore>(
         }
 
         chain.database.update(&[
+            WriteOp::Put(keys::power(), curr_pow.into()),
             WriteOp::Put(keys::height(), (curr_height + 1).into()),
             WriteOp::Put(
                 keys::header(block.header.number),

diff --git a/src/db/keys.rs b/src/db/keys.rs
@@ -11,6 +11,10 @@ pub fn randomness() -> StringKey {
     "RND".into()
 }
 
+pub fn power() -> StringKey {
+    "PWR".into()
+}
+
 pub fn block(index: u64) -> StringKey {
     format!("BLK-{:010}", index).into()
 }

diff --git a/src/db/mod.rs b/src/db/mod.rs
@@ -93,6 +93,7 @@ gen_try_into!(
     u32,
     u64,
     u128,
+    f64,
     usize,
     <Hasher as Hash>::Output,
     Ratio,
@@ -122,6 +123,7 @@ gen_from!(
     u32,
     u64,
     u128,
+    f64,
     usize,
     <Hasher as Hash>::Output,
     Ratio,