fuel_core/

combined_database.rs

#[cfg(feature = "rocksdb")]
use crate::state::{
    historical_rocksdb::StateRewindPolicy,
    rocks_db::DatabaseConfig,
};

use crate::{
    database::{
        database_description::{
            gas_price::GasPriceDatabase,
            off_chain::OffChain,
            on_chain::OnChain,
            relayer::Relayer,
        },
        Database,
        GenesisDatabase,
        Result as DatabaseResult,
    },
    service::DbType,
};
#[cfg(feature = "test-helpers")]
use fuel_core_chain_config::{
    StateConfig,
    StateConfigBuilder,
};
#[cfg(feature = "backup")]
use fuel_core_services::TraceErr;
#[cfg(feature = "test-helpers")]
use fuel_core_storage::tables::{
    Coins,
    ContractsAssets,
    ContractsLatestUtxo,
    ContractsRawCode,
    ContractsState,
    Messages,
};
use fuel_core_storage::Result as StorageResult;
use fuel_core_types::fuel_types::BlockHeight;
use std::path::PathBuf;

#[derive(Clone, Debug, Eq, PartialEq)]
pub struct CombinedDatabaseConfig {
    pub database_path: PathBuf,
    pub database_type: DbType,
    #[cfg(feature = "rocksdb")]
    pub database_config: DatabaseConfig,
    #[cfg(feature = "rocksdb")]
    pub state_rewind_policy: StateRewindPolicy,
}

/// A database that combines the on-chain, off-chain and relayer databases into one entity.
#[derive(Default, Clone)]
pub struct CombinedDatabase {
    on_chain: Database<OnChain>,
    off_chain: Database<OffChain>,
    relayer: Database<Relayer>,
    gas_price: Database<GasPriceDatabase>,
}

impl CombinedDatabase {
    pub fn new(
        on_chain: Database<OnChain>,
        off_chain: Database<OffChain>,
        relayer: Database<Relayer>,
        gas_price: Database<GasPriceDatabase>,
    ) -> Self {
        Self {
            on_chain,
            off_chain,
            relayer,
            gas_price,
        }
    }

    #[cfg(feature = "rocksdb")]
    pub fn prune(path: &std::path::Path) -> crate::database::Result<()> {
        crate::state::rocks_db::RocksDb::<OnChain>::prune(path)?;
        crate::state::rocks_db::RocksDb::<OffChain>::prune(path)?;
        crate::state::rocks_db::RocksDb::<Relayer>::prune(path)?;
        crate::state::rocks_db::RocksDb::<GasPriceDatabase>::prune(path)?;
        Ok(())
    }

    #[cfg(feature = "backup")]
    pub fn backup(
        db_dir: &std::path::Path,
        backup_dir: &std::path::Path,
    ) -> crate::database::Result<()> {
        use tempfile::TempDir;

        let temp_backup_dir = TempDir::new()
            .trace_err("Failed to create temporary backup directory")
            .map_err(|e| anyhow::anyhow!(e))?;

        Self::backup_databases(db_dir, temp_backup_dir.path())?;

        std::fs::rename(temp_backup_dir.path(), backup_dir)
            .trace_err("Failed to move temporary backup directory")
            .map_err(|e| anyhow::anyhow!(e))?;

        Ok(())
    }

    #[cfg(feature = "backup")]
    fn backup_databases(
        db_dir: &std::path::Path,
        temp_dir: &std::path::Path,
    ) -> crate::database::Result<()> {
        crate::state::rocks_db::RocksDb::<OnChain>::backup(db_dir, temp_dir)
            .trace_err("Failed to backup on-chain database")?;

        crate::state::rocks_db::RocksDb::<OffChain>::backup(db_dir, temp_dir)
            .trace_err("Failed to backup off-chain database")?;

        crate::state::rocks_db::RocksDb::<Relayer>::backup(db_dir, temp_dir)
            .trace_err("Failed to backup relayer database")?;

        crate::state::rocks_db::RocksDb::<GasPriceDatabase>::backup(db_dir, temp_dir)
            .trace_err("Failed to backup gas-price database")?;

        Ok(())
    }

    #[cfg(feature = "backup")]
    pub fn restore(
        restore_to: &std::path::Path,
        backup_dir: &std::path::Path,
    ) -> crate::database::Result<()> {
        use tempfile::TempDir;

        let temp_restore_dir = TempDir::new()
            .trace_err("Failed to create temporary restore directory")
            .map_err(|e| anyhow::anyhow!(e))?;

        Self::restore_database(backup_dir, temp_restore_dir.path())?;

        std::fs::rename(temp_restore_dir.path(), restore_to)
            .trace_err("Failed to move temporary restore directory")
            .map_err(|e| anyhow::anyhow!(e))?;

        // we don't return a CombinedDatabase here
        // because the consumer can use any db config while opening it
        Ok(())
    }

    #[cfg(feature = "backup")]
    fn restore_database(
        backup_dir: &std::path::Path,
        temp_restore_dir: &std::path::Path,
    ) -> crate::database::Result<()> {
        crate::state::rocks_db::RocksDb::<OnChain>::restore(temp_restore_dir, backup_dir)
            .trace_err("Failed to backup on-chain database")?;

        crate::state::rocks_db::RocksDb::<OffChain>::restore(
            temp_restore_dir,
            backup_dir,
        )
        .trace_err("Failed to backup off-chain database")?;

        crate::state::rocks_db::RocksDb::<Relayer>::restore(temp_restore_dir, backup_dir)
            .trace_err("Failed to backup relayer database")?;

        crate::state::rocks_db::RocksDb::<GasPriceDatabase>::restore(
            temp_restore_dir,
            backup_dir,
        )
        .trace_err("Failed to backup gas-price database")?;

        Ok(())
    }

    #[cfg(feature = "rocksdb")]
    pub fn open(
        path: &std::path::Path,
        state_rewind_policy: StateRewindPolicy,
        database_config: DatabaseConfig,
    ) -> crate::database::Result<Self> {
        // Split the fds in equitable manner between the databases

        let max_fds = match database_config.max_fds {
            -1 => -1,
            _ => database_config.max_fds.saturating_div(4),
        };

        // TODO: Use different cache sizes for different databases
        let on_chain = Database::open_rocksdb(
            path,
            state_rewind_policy,
            DatabaseConfig {
                max_fds,
                ..database_config
            },
        )?;
        let off_chain = Database::open_rocksdb(
            path,
            state_rewind_policy,
            DatabaseConfig {
                max_fds,
                ..database_config
            },
        )?;
        let relayer = Database::open_rocksdb(
            path,
            StateRewindPolicy::NoRewind,
            DatabaseConfig {
                max_fds,
                ..database_config
            },
        )?;
        let gas_price = Database::open_rocksdb(
            path,
            state_rewind_policy,
            DatabaseConfig {
                max_fds,
                ..database_config
            },
        )?;
        Ok(Self {
            on_chain,
            off_chain,
            relayer,
            gas_price,
        })
    }

    /// A test-only temporary rocksdb database with given rewind policy.
    #[cfg(feature = "rocksdb")]
    pub fn temp_database_with_state_rewind_policy(
        state_rewind_policy: StateRewindPolicy,
        database_config: DatabaseConfig,
    ) -> DatabaseResult<Self> {
        Ok(Self {
            on_chain: Database::rocksdb_temp(state_rewind_policy, database_config)?,
            off_chain: Database::rocksdb_temp(state_rewind_policy, database_config)?,
            relayer: Default::default(),
            gas_price: Default::default(),
        })
    }

    pub fn from_config(config: &CombinedDatabaseConfig) -> DatabaseResult<Self> {
        let combined_database = match config.database_type {
            #[cfg(feature = "rocksdb")]
            DbType::RocksDb => {
                // use a default tmp rocksdb if no path is provided
                if config.database_path.as_os_str().is_empty() {
                    tracing::warn!(
                        "No RocksDB path configured, initializing database with a tmp directory"
                    );
                    CombinedDatabase::temp_database_with_state_rewind_policy(
                        config.state_rewind_policy,
                        config.database_config,
                    )?
                } else {
                    tracing::info!(
                        "Opening database {:?} with cache size \"{:?}\" and state rewind policy \"{:?}\"",
                        config.database_path,
                        config.database_config.cache_capacity,
                        config.state_rewind_policy,
                    );
                    CombinedDatabase::open(
                        &config.database_path,
                        config.state_rewind_policy,
                        config.database_config,
                    )?
                }
            }
            DbType::InMemory => CombinedDatabase::in_memory(),
            #[cfg(not(feature = "rocksdb"))]
            _ => CombinedDatabase::in_memory(),
        };

        Ok(combined_database)
    }

    pub fn in_memory() -> Self {
        Self::new(
            Database::in_memory(),
            Database::in_memory(),
            Database::in_memory(),
            Database::in_memory(),
        )
    }

    pub fn check_version(&self) -> StorageResult<()> {
        self.on_chain.check_version()?;
        self.off_chain.check_version()?;
        self.relayer.check_version()?;
        self.gas_price.check_version()?;
        Ok(())
    }

    pub fn on_chain(&self) -> &Database<OnChain> {
        &self.on_chain
    }

    #[cfg(any(feature = "test-helpers", test))]
    pub fn on_chain_mut(&mut self) -> &mut Database<OnChain> {
        &mut self.on_chain
    }

    pub fn off_chain(&self) -> &Database<OffChain> {
        &self.off_chain
    }

    #[cfg(any(feature = "test-helpers", test))]
    pub fn off_chain_mut(&mut self) -> &mut Database<OffChain> {
        &mut self.off_chain
    }

    pub fn relayer(&self) -> &Database<Relayer> {
        &self.relayer
    }

    #[cfg(any(feature = "test-helpers", test))]
    pub fn relayer_mut(&mut self) -> &mut Database<Relayer> {
        &mut self.relayer
    }

    pub fn gas_price(&self) -> &Database<GasPriceDatabase> {
        &self.gas_price
    }

    #[cfg(any(feature = "test-helpers", test))]
    pub fn gas_price_mut(&mut self) -> &mut Database<GasPriceDatabase> {
        &mut self.gas_price
    }

    #[cfg(feature = "test-helpers")]
    pub fn read_state_config(&self) -> StorageResult<StateConfig> {
        use fuel_core_chain_config::AddTable;
        use fuel_core_producer::ports::BlockProducerDatabase;
        use fuel_core_storage::transactional::AtomicView;
        use fuel_core_types::fuel_vm::BlobData;
        use itertools::Itertools;
        let mut builder = StateConfigBuilder::default();

        macro_rules! add_tables {
            ($($table: ty),*) => {
                $(
                    let table = self
                        .on_chain()
                        .entries::<$table>(None, fuel_core_storage::iter::IterDirection::Forward)
                        .try_collect()?;
                    builder.add(table);
                )*
            };
        }

        add_tables!(
            Coins,
            Messages,
            BlobData,
            ContractsAssets,
            ContractsState,
            ContractsRawCode,
            ContractsLatestUtxo
        );

        let view = self.on_chain().latest_view()?;
        let latest_block = view.latest_block()?;
        let blocks_root =
            view.block_header_merkle_root(latest_block.header().height())?;
        let state_config =
            builder.build(Some(fuel_core_chain_config::LastBlockConfig::from_header(
                latest_block.header(),
                blocks_root,
            )))?;

        Ok(state_config)
    }

    /// Rollbacks the state of the blockchain to a specific block height.
    pub fn rollback_to<S>(
        &self,
        target_block_height: BlockHeight,
        shutdown_listener: &mut S,
    ) -> anyhow::Result<()>
    where
        S: ShutdownListener,
    {
        while !shutdown_listener.is_cancelled() {
            let on_chain_height = self
                .on_chain()
                .latest_height_from_metadata()?
                .ok_or(anyhow::anyhow!("on-chain database doesn't have height"))?;

            let off_chain_height = self
                .off_chain()
                .latest_height_from_metadata()?
                .ok_or(anyhow::anyhow!("off-chain database doesn't have height"))?;

            let gas_price_chain_height =
                self.gas_price().latest_height_from_metadata()?;

            let gas_price_rolled_back = gas_price_chain_height.is_none()
                || gas_price_chain_height.expect("We checked height before")
                    == target_block_height;

            if on_chain_height == target_block_height
                && off_chain_height == target_block_height
                && gas_price_rolled_back
            {
                break;
            }

            if on_chain_height < target_block_height {
                return Err(anyhow::anyhow!(
                    "on-chain database height({on_chain_height}) \
                    is less than target height({target_block_height})"
                ));
            }

            if off_chain_height < target_block_height {
                return Err(anyhow::anyhow!(
                    "off-chain database height({off_chain_height}) \
                    is less than target height({target_block_height})"
                ));
            }

            if let Some(gas_price_chain_height) = gas_price_chain_height {
                if gas_price_chain_height < target_block_height {
                    return Err(anyhow::anyhow!(
                        "gas-price-chain database height({gas_price_chain_height}) \
                        is less than target height({target_block_height})"
                    ));
                }
            }

            if on_chain_height > target_block_height {
                self.on_chain().rollback_last_block()?;
            }

            if off_chain_height > target_block_height {
                self.off_chain().rollback_last_block()?;
            }

            if let Some(gas_price_chain_height) = gas_price_chain_height {
                if gas_price_chain_height > target_block_height {
                    self.gas_price().rollback_last_block()?;
                }
            }
        }

        if shutdown_listener.is_cancelled() {
            return Err(anyhow::anyhow!(
                "Stop the rollback due to shutdown signal received"
            ));
        }

        Ok(())
    }

    /// This function is fundamentally different from `rollback_to` in that it
    /// will rollback the off-chain/gas-price databases if they are ahead of the
    /// on-chain database. If they don't have a height or are behind the on-chain
    /// we leave it to the caller to decide how to bring them up to date.
    /// We don't rollback the on-chain database as it is the source of truth.
    /// The target height of the rollback is the latest height of the on-chain database.
    pub fn sync_aux_db_heights<S>(&self, shutdown_listener: &mut S) -> anyhow::Result<()>
    where
        S: ShutdownListener,
    {
        while !shutdown_listener.is_cancelled() {
            let on_chain_height = match self.on_chain().latest_height_from_metadata()? {
                Some(height) => height,
                None => break, // Exit loop if on-chain height is None
            };

            let off_chain_height = self.off_chain().latest_height_from_metadata()?;
            let gas_price_height = self.gas_price().latest_height_from_metadata()?;

            // Handle off-chain rollback if necessary
            if let Some(off_height) = off_chain_height {
                if off_height > on_chain_height {
                    self.off_chain().rollback_last_block()?;
                }
            }

            // Handle gas price rollback if necessary
            if let Some(gas_height) = gas_price_height {
                if gas_height > on_chain_height {
                    self.gas_price().rollback_last_block()?;
                }
            }

            // If both off-chain and gas price heights are synced, break
            if off_chain_height.map_or(true, |h| h <= on_chain_height)
                && gas_price_height.map_or(true, |h| h <= on_chain_height)
            {
                break;
            }
        }

        Ok(())
    }
}

/// A trait for listening to shutdown signals.
pub trait ShutdownListener {
    /// Returns true if the shutdown signal has been received.
    fn is_cancelled(&self) -> bool;
}

/// A genesis database that combines the on-chain, off-chain and relayer
/// genesis databases into one entity.
#[derive(Default, Clone)]
pub struct CombinedGenesisDatabase {
    pub on_chain: GenesisDatabase<OnChain>,
    pub off_chain: GenesisDatabase<OffChain>,
}

impl CombinedGenesisDatabase {
    pub fn on_chain(&self) -> &GenesisDatabase<OnChain> {
        &self.on_chain
    }

    pub fn off_chain(&self) -> &GenesisDatabase<OffChain> {
        &self.off_chain
    }
}

#[allow(non_snake_case)]
#[cfg(feature = "backup")]
#[cfg(test)]
mod tests {
    use super::*;
    use fuel_core_storage::StorageAsMut;
    use fuel_core_types::{
        entities::coins::coin::CompressedCoin,
        fuel_tx::UtxoId,
    };
    use tempfile::TempDir;

    #[test]
    fn backup_and_restore__works_correctly__happy_path() {
        // given
        let db_dir = TempDir::new().unwrap();
        let mut combined_db = CombinedDatabase::open(
            db_dir.path(),
            StateRewindPolicy::NoRewind,
            DatabaseConfig::config_for_tests(),
        )
        .unwrap();
        let key = UtxoId::new(Default::default(), Default::default());
        let expected_value = CompressedCoin::default();

        let on_chain_db = combined_db.on_chain_mut();
        on_chain_db
            .storage_as_mut::<Coins>()
            .insert(&key, &expected_value)
            .unwrap();
        drop(combined_db);

        // when
        let backup_dir = TempDir::new().unwrap();
        CombinedDatabase::backup(db_dir.path(), backup_dir.path()).unwrap();

        // then
        let restore_dir = TempDir::new().unwrap();
        CombinedDatabase::restore(restore_dir.path(), backup_dir.path()).unwrap();
        let restored_db = CombinedDatabase::open(
            restore_dir.path(),
            StateRewindPolicy::NoRewind,
            DatabaseConfig::config_for_tests(),
        )
        .unwrap();

        let mut restored_on_chain_db = restored_db.on_chain();
        let restored_value = restored_on_chain_db
            .storage::<Coins>()
            .get(&key)
            .unwrap()
            .unwrap()
            .into_owned();
        assert_eq!(expected_value, restored_value);

        // cleanup
        std::fs::remove_dir_all(db_dir.path()).unwrap();
        std::fs::remove_dir_all(backup_dir.path()).unwrap();
        std::fs::remove_dir_all(restore_dir.path()).unwrap();
    }

    #[test]
    fn backup__when_backup_fails_it_should_not_leave_any_residue() {
        use std::os::unix::fs::PermissionsExt;

        // given
        let db_dir = TempDir::new().unwrap();
        let mut combined_db = CombinedDatabase::open(
            db_dir.path(),
            StateRewindPolicy::NoRewind,
            DatabaseConfig::config_for_tests(),
        )
        .unwrap();
        let key = UtxoId::new(Default::default(), Default::default());
        let expected_value = CompressedCoin::default();

        let on_chain_db = combined_db.on_chain_mut();
        on_chain_db
            .storage_as_mut::<Coins>()
            .insert(&key, &expected_value)
            .unwrap();
        drop(combined_db);

        // when
        // we set the permissions of db_dir to not allow reading
        std::fs::set_permissions(db_dir.path(), std::fs::Permissions::from_mode(0o030))
            .unwrap();
        let backup_dir = TempDir::new().unwrap();

        // then
        CombinedDatabase::backup(db_dir.path(), backup_dir.path())
            .expect_err("Backup should fail");
        let backup_dir_contents = std::fs::read_dir(backup_dir.path()).unwrap();
        assert_eq!(backup_dir_contents.count(), 0);

        // cleanup
        std::fs::set_permissions(db_dir.path(), std::fs::Permissions::from_mode(0o770))
            .unwrap();
        std::fs::remove_dir_all(db_dir.path()).unwrap();
        std::fs::remove_dir_all(backup_dir.path()).unwrap();
    }

    #[test]
    fn restore__when_restore_fails_it_should_not_leave_any_residue() {
        use std::os::unix::fs::PermissionsExt;

        // given
        let db_dir = TempDir::new().unwrap();
        let mut combined_db = CombinedDatabase::open(
            db_dir.path(),
            StateRewindPolicy::NoRewind,
            DatabaseConfig::config_for_tests(),
        )
        .unwrap();
        let key = UtxoId::new(Default::default(), Default::default());
        let expected_value = CompressedCoin::default();

        let on_chain_db = combined_db.on_chain_mut();
        on_chain_db
            .storage_as_mut::<Coins>()
            .insert(&key, &expected_value)
            .unwrap();
        drop(combined_db);

        let backup_dir = TempDir::new().unwrap();
        CombinedDatabase::backup(db_dir.path(), backup_dir.path()).unwrap();

        // when
        // we set the permissions of backup_dir to not allow reading
        std::fs::set_permissions(
            backup_dir.path(),
            std::fs::Permissions::from_mode(0o030),
        )
        .unwrap();
        let restore_dir = TempDir::new().unwrap();

        // then
        CombinedDatabase::restore(restore_dir.path(), backup_dir.path())
            .expect_err("Restore should fail");
        let restore_dir_contents = std::fs::read_dir(restore_dir.path()).unwrap();
        assert_eq!(restore_dir_contents.count(), 0);

        // cleanup
        std::fs::set_permissions(
            backup_dir.path(),
            std::fs::Permissions::from_mode(0o770),
        )
        .unwrap();
        std::fs::remove_dir_all(db_dir.path()).unwrap();
        std::fs::remove_dir_all(backup_dir.path()).unwrap();
        std::fs::remove_dir_all(restore_dir.path()).unwrap();
    }

    #[test]
    fn backup__cannot_backup_while_db_is_opened() {
        // given
        let db_dir = TempDir::new().unwrap();
        let mut combined_db = CombinedDatabase::open(
            db_dir.path(),
            StateRewindPolicy::NoRewind,
            DatabaseConfig::config_for_tests(),
        )
        .unwrap();
        let key = UtxoId::new(Default::default(), Default::default());
        let expected_value = CompressedCoin::default();

        let on_chain_db = combined_db.on_chain_mut();
        on_chain_db
            .storage_as_mut::<Coins>()
            .insert(&key, &expected_value)
            .unwrap();

        // when
        let backup_dir = TempDir::new().unwrap();
        // no drop for combined_db

        // then
        CombinedDatabase::backup(db_dir.path(), backup_dir.path())
            .expect_err("Backup should fail");

        // cleanup
        std::fs::remove_dir_all(db_dir.path()).unwrap();
        std::fs::remove_dir_all(backup_dir.path()).unwrap();
    }
}