lighthouse/beacon_node/beacon_chain/src/persisted_fork_choice.rs

use crate::beacon_fork_choice_store::{PersistedForkChoiceStoreV11, PersistedForkChoiceStoreV17};
use ssz::{Decode, Encode};
use ssz_derive::{Decode, Encode};
use store::{DBColumn, Error, StoreItem};
use superstruct::superstruct;

// If adding a new version you should update this type alias and fix the breakages.
pub type PersistedForkChoice = PersistedForkChoiceV17;

#[superstruct(
    variants(V11, V17),
    variant_attributes(derive(Encode, Decode)),
    no_enum
)]
pub struct PersistedForkChoice {
    pub fork_choice: fork_choice::PersistedForkChoice,
    #[superstruct(only(V11))]
    pub fork_choice_store: PersistedForkChoiceStoreV11,
    #[superstruct(only(V17))]
    pub fork_choice_store: PersistedForkChoiceStoreV17,
}

impl Into<PersistedForkChoice> for PersistedForkChoiceV11 {
    fn into(self) -> PersistedForkChoice {
        PersistedForkChoice {
            fork_choice: self.fork_choice,
            fork_choice_store: self.fork_choice_store.into(),
        }
    }
}

impl Into<PersistedForkChoiceV11> for PersistedForkChoice {
    fn into(self) -> PersistedForkChoiceV11 {
        PersistedForkChoiceV11 {
            fork_choice: self.fork_choice,
            fork_choice_store: self.fork_choice_store.into(),
        }
    }
}

macro_rules! impl_store_item {
    ($type:ty) => {
        impl StoreItem for $type {
            fn db_column() -> DBColumn {
                DBColumn::ForkChoice
            }

            fn as_store_bytes(&self) -> Result<Vec<u8>, Error> {
                Ok(self.as_ssz_bytes())
            }

            fn from_store_bytes(bytes: &[u8]) -> Result<Self, Error> {
                Self::from_ssz_bytes(bytes).map_err(Into::into)
            }
        }
    };
}

impl_store_item!(PersistedForkChoiceV11);
impl_store_item!(PersistedForkChoiceV17);