feat(node): update global protocol version state when necessary

witnet · Nov 13, 2024 · 27a580b · 27a580b
1 parent 28b630d
commit 27a580b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/node/src/actors/chain_manager/handlers.rs b/node/src/actors/chain_manager/handlers.rs
@@ -19,6 +19,7 @@ use witnet_data_structures::{
     error::{ChainInfoError, TransactionError::DataRequestNotFound},
     get_protocol_version,
     proto::versioning::ProtocolVersion,
+    refresh_protocol_version,
     staking::{errors::StakesError, prelude::StakeKey},
     transaction::{
         DRTransaction, StakeTransaction, Transaction, UnstakeTransaction, VTTransaction,
@@ -138,6 +139,11 @@ impl Handler<EpochNotification<EveryEpochPayload>> for ChainManager {
             protocol.current_version = get_protocol_version(self.current_epoch);
         }
 
+        // Update the global protocol version state if necessary
+        if get_protocol_version(self.current_epoch) != get_protocol_version(None) {
+            refresh_protocol_version(current_epoch);
+        }
+
         match self.sm_state {
             StateMachine::WaitingConsensus => {
                 if let Some(chain_info) = &self.chain_state.chain_info {