Make Consensus version-able

CHANGELOG.md

@@ -16,6 +16,7 @@ Description of the upcoming release here.
 - [#1577](https://github.com/FuelLabs/fuel-core/pull/1577): Moved insertion of sealed blocks into the `BlockImporter` instead of the executor.
 
 #### Breaking
+- [#1596](https://github.com/FuelLabs/fuel-core/pull/1596) Make `Consensus` type a version-able enum
 - [#1593](https://github.com/FuelLabs/fuel-core/pull/1593) Make `Block` type a version-able enum
 - [#1573](https://github.com/FuelLabs/fuel-core/pull/1573): Remove nested p2p request/response encoding. Only breaks p2p networking compatibility with older fuel-core versions, but is otherwise fully internal.
 

crates/fuel-core/src/schema/block.rs

@@ -53,12 +53,14 @@ use fuel_core_types::{
     fuel_types,
     fuel_types::BlockHeight,
 };
+use thiserror::Error as ThisError;
 
 pub struct Block(pub(crate) CompressedBlock);
 
 pub struct Header(pub(crate) BlockHeader);
 
 #[derive(Union)]
+#[non_exhaustive]
 pub enum Consensus {
     Genesis(Genesis),
     PoA(PoAConsensus),
@@ -84,6 +86,16 @@ pub struct PoAConsensus {
     signature: Signature,
 }
 
+#[derive(ThisError, Debug)]
+pub enum BlockError {
+    #[error("Error from `async_graphql`")]
+    AsyncGraphQl(async_graphql::Error),
+    #[error("Error from `fuel_core_storage`")]
+    FuelCoreStorage(fuel_core_storage::Error),
+    #[error("Found unsupported consensus variant: {0}")]
+    UnsupportedConsensusVariant(String),
+}
+
 #[Object]
 impl Block {
     async fn id(&self) -> BlockId {
@@ -95,12 +107,15 @@ impl Block {
         self.0.header().clone().into()
     }
 
-    async fn consensus(&self, ctx: &Context<'_>) -> async_graphql::Result<Consensus> {
+    async fn consensus(&self, ctx: &Context<'_>) -> Result<Consensus, BlockError> {
         let query: &Database = ctx.data_unchecked();
         let id = self.0.header().id();
-        let consensus = query.consensus(&id)?;
+        let core_consensus = query.consensus(&id).map_err(BlockError::FuelCoreStorage)?;
 
-        Ok(consensus.into())
+        let my_consensus = core_consensus
+            .try_into()
+            .map_err(BlockError::UnsupportedConsensusVariant)?;
+        Ok(my_consensus)
     }
 
     async fn transactions(
@@ -342,13 +357,16 @@ impl From<CoreGenesis> for Genesis {
     }
 }
 
-impl From<CoreConsensus> for Consensus {
-    fn from(consensus: CoreConsensus) -> Self {
+impl TryFrom<CoreConsensus> for Consensus {
+    type Error = String;
+
+    fn try_from(consensus: CoreConsensus) -> Result<Self, Self::Error> {
         match consensus {
-            CoreConsensus::Genesis(genesis) => Consensus::Genesis(genesis.into()),
-            CoreConsensus::PoA(poa) => Consensus::PoA(PoAConsensus {
+            CoreConsensus::Genesis(genesis) => Ok(Consensus::Genesis(genesis.into())),
+            CoreConsensus::PoA(poa) => Ok(Consensus::PoA(PoAConsensus {
                 signature: poa.signature.into(),
-            }),
+            })),
+            _ => Err(format!("Unknown consensus type: {:?}", consensus)),
         }
     }
 }

crates/services/consensus_module/src/block_verifier.rs

@@ -72,6 +72,7 @@ where
             Consensus::PoA(_) => {
                 fuel_core_poa::verifier::verify_block_fields(&self.database, block)
             }
+            _ => Err(anyhow::anyhow!("Unsupported consensus: {:?}", consensus)),
         }
     }
 
@@ -88,6 +89,7 @@ where
                 header,
                 consensus,
             ),
+            _ => false,
         }
     }
 

crates/services/importer/src/importer.rs

@@ -88,6 +88,7 @@ pub enum Error {
     NotUnique(BlockHeight),
     #[from]
     StorageError(StorageError),
+    UnsupportedConsensusVariant(String),
 }
 
 impl From<Error> for anyhow::Error {
@@ -223,6 +224,12 @@ where
                     .ok_or(Error::Overflow)?
                     .into()
             }
+            _ => {
+                return Err(Error::UnsupportedConsensusVariant(format!(
+                    "{:?}",
+                    consensus
+                )))
+            }
         };
 
         if expected_next_height != actual_next_height {

crates/types/src/blockchain/consensus.rs

@@ -16,6 +16,7 @@ use poa::PoAConsensus;
 
 #[derive(Clone, Debug, PartialEq, Eq)]
 #[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
+#[non_exhaustive]
 /// The consensus related data that doesn't live on the
 /// header.
 pub enum Consensus {