Merge pull request #69 from nazar-pc/fix-balance-after-reorg

Support chain reorgs gracefully where balance can decrease
autonomys · Dec 23, 2023 · da12e04 · da12e04
2 parents 0876c50 + 273ca3f
commit da12e04
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/frontend/running.rs b/src/frontend/running.rs
@@ -392,6 +392,15 @@ impl RunningView {
                         self.farmer_state.initial_reward_address_balance =
                             imported_block.reward_address_balance - previous_diff;
                     }
+                    // In case balance decreased, subtract it from initial balance to ignore, this
+                    // typically happens due to chain reorg when reward is "disappears"
+                    if let Some(decreased_by) = self
+                        .farmer_state
+                        .reward_address_balance
+                        .checked_sub(imported_block.reward_address_balance)
+                    {
+                        self.farmer_state.initial_reward_address_balance -= decreased_by;
+                    }
                     self.farmer_state.reward_address_balance =
                         imported_block.reward_address_balance;