Companion for new trie cache

pallets/parachain-system/src/relay_state_snapshot.rs

@@ -20,7 +20,7 @@ use cumulus_primitives_core::{
 };
 use scale_info::TypeInfo;
 use sp_runtime::traits::HashFor;
-use sp_state_machine::{Backend, TrieBackend};
+use sp_state_machine::{Backend, TrieBackend, TrieBackendBuilder};
 use sp_std::vec::Vec;
 use sp_trie::{HashDBT, MemoryDB, StorageProof, EMPTY_PREFIX};
 
@@ -156,7 +156,7 @@ impl RelayChainStateProof {
 		if !db.contains(&relay_parent_storage_root, EMPTY_PREFIX) {
 			return Err(Error::RootMismatch)
 		}
-		let trie_backend = TrieBackend::new(db, relay_parent_storage_root);
+		let trie_backend = TrieBackendBuilder::new(db, relay_parent_storage_root).build();
 
 		Ok(Self { para_id, trie_backend })
 	}

pallets/parachain-system/src/validate_block/implementation.rs

@@ -74,7 +74,7 @@ where
 
 	sp_std::mem::drop(storage_proof);
 
-	let backend = sp_state_machine::TrieBackend::new(db, *parent_head.state_root());
+	let backend = sp_state_machine::TrieBackendBuilder::new(db, *parent_head.state_root()).build();
 
 	let _guard = (
 		// Replace storage calls with our own implementations

parachain-template/node/src/command.rs

@@ -399,8 +399,8 @@ impl CliConfiguration<Self> for RelayChainCli {
 		self.base.base.transaction_pool(is_dev)
 	}
 
-	fn state_cache_child_ratio(&self) -> Result<Option<usize>> {
-		self.base.base.state_cache_child_ratio()
+	fn trie_cache_maximum_size(&self) -> Result<Option<usize>> {
+		self.base.base.trie_cache_maximum_size()
 	}
 
 	fn rpc_methods(&self) -> Result<sc_service::config::RpcMethods> {

polkadot-parachain/src/command.rs

@@ -832,8 +832,8 @@ impl CliConfiguration<Self> for RelayChainCli {
 		self.base.base.transaction_pool(is_dev)
 	}
 
-	fn state_cache_child_ratio(&self) -> Result<Option<usize>> {
-		self.base.base.state_cache_child_ratio()
+	fn trie_cache_maximum_size(&self) -> Result<Option<usize>> {
+		self.base.base.trie_cache_maximum_size()
 	}
 
 	fn rpc_methods(&self) -> Result<sc_service::config::RpcMethods> {

test/relay-sproof-builder/src/lib.rs

@@ -102,7 +102,7 @@ impl RelayStateSproofBuilder {
 	) -> (polkadot_primitives::v2::Hash, sp_state_machine::StorageProof) {
 		let (db, root) = MemoryDB::<HashFor<polkadot_primitives::v2::Block>>::default_with_root();
 		let state_version = Default::default(); // for test using default.
-		let mut backend = sp_state_machine::TrieBackend::new(db, root);
+		let mut backend = sp_state_machine::TrieBackendBuilder::new(db, root).build();
 
 		let mut relevant_keys = Vec::new();
 		{

test/service/src/cli.rs

@@ -191,8 +191,8 @@ impl CliConfiguration<Self> for RelayChainCli {
 		self.base.base.transaction_pool(is_dev)
 	}
 
-	fn state_cache_child_ratio(&self) -> CliResult<Option<usize>> {
-		self.base.base.state_cache_child_ratio()
+	fn trie_cache_maximum_size(&self) -> CliResult<Option<usize>> {
+		self.base.base.trie_cache_maximum_size()
 	}
 
 	fn rpc_methods(&self) -> CliResult<sc_service::config::RpcMethods> {

test/service/src/lib.rs

@@ -650,8 +650,7 @@ pub fn node_config(
 		keystore: KeystoreConfig::InMemory,
 		keystore_remote: Default::default(),
 		database: DatabaseSource::RocksDb { path: root.join("db"), cache_size: 128 },
-		state_cache_size: 67108864,
-		state_cache_child_ratio: None,
+		trie_cache_maximum_size: Some(64 * 1024 * 1024),
 		state_pruning: Some(PruningMode::ArchiveAll),
 		blocks_pruning: BlocksPruning::All,
 		chain_spec: spec,