starknet_committer: add index db

crates/starknet_committer/src/db/index_db/db.rs

@@ -0,0 +1,118 @@
+use std::collections::HashMap;
+
+use starknet_api::core::ContractAddress;
+use starknet_api::hash::HashOutput;
+use starknet_patricia::patricia_merkle_tree::filled_tree::node::FilledNode;
+use starknet_patricia::patricia_merkle_tree::node_data::leaf::{Leaf, LeafModifications};
+use starknet_patricia::patricia_merkle_tree::types::{NodeIndex, SortedLeafIndices};
+use starknet_patricia::patricia_merkle_tree::updated_skeleton_tree::hash_function::TreeHashFunction;
+use starknet_patricia_storage::db_object::{DBObject, HasStaticPrefix};
+use starknet_patricia_storage::errors::DeserializationError;
+use starknet_patricia_storage::storage_trait::{DbValue, Storage};
+
+use crate::block_committer::input::{ConfigImpl, StarknetStorageValue};
+use crate::db::db_layout::NodeLayout;
+use crate::db::forest_trait::{ForestReader, ForestWriter};
+use crate::db::index_db::leaves::{
+    IndexLayoutCompiledClassHash,
+    IndexLayoutContractState,
+    IndexLayoutStarknetStorageValue,
+    TrieType,
+};
+use crate::db::index_db::types::{IndexFilledNode, IndexLayoutSubTree, IndexNodeContext};
+use crate::db::trie_traversal::{create_classes_trie, create_contracts_trie, create_storage_tries};
+use crate::forest::filled_forest::FilledForest;
+use crate::forest::forest_errors::ForestResult;
+use crate::forest::original_skeleton_forest::{ForestSortedIndices, OriginalSkeletonForest};
+use crate::hash_function::hash::TreeHashFunctionImpl;
+use crate::patricia_merkle_tree::leaf::leaf_impl::ContractState;
+use crate::patricia_merkle_tree::types::CompiledClassHash;
+pub struct IndexDb<S: Storage> {
+    pub storage: S,
+}
+
+impl<S: Storage> IndexDb<S> {
+    pub fn new(storage: S) -> Self {
+        Self { storage }
+    }
+}
+
+pub struct IndexNodeLayout {}
+
+impl<'a, L: Leaf> NodeLayout<'a, L> for IndexNodeLayout
+where
+    L: HasStaticPrefix<KeyContext = TrieType>,
+    TreeHashFunctionImpl: TreeHashFunction<L>,
+{
+    type ChildData = ();
+    type DeserializationContext = IndexNodeContext;
+    type SubTree = IndexLayoutSubTree<'a>;
+    fn deserialize_node(
+        value: &DbValue,
+        deserialize_context: &Self::DeserializationContext,
+    ) -> Result<FilledNode<L, ()>, DeserializationError> {
+        let filled_node: IndexFilledNode<L> = DBObject::deserialize(value, deserialize_context)?;
+        Ok(filled_node.0)
+    }
+    fn create_subtree(
+        sorted_leaf_indices: SortedLeafIndices<'a>,
+        root_index: NodeIndex,
+        _root_hash: HashOutput,
+    ) -> Self::SubTree {
+        IndexLayoutSubTree { sorted_leaf_indices, root_index }
+    }
+    fn generate_key_context(trie_type: TrieType) -> <L as HasStaticPrefix>::KeyContext {
+        trie_type
+    }
+}
+
+impl<'a, S: Storage> ForestReader<'a> for IndexDb<S> {
+    /// Creates an original skeleton forest that includes the storage tries of the modified
+    /// contracts, the classes trie and the contracts trie. Additionally, returns the original
+    /// contract states that are needed to compute the contract state tree.
+    async fn read(
+        &mut self,
+        contracts_trie_root_hash: HashOutput,
+        classes_trie_root_hash: HashOutput,
+        storage_updates: &'a HashMap<ContractAddress, LeafModifications<StarknetStorageValue>>,
+        classes_updates: &'a LeafModifications<CompiledClassHash>,
+        forest_sorted_indices: &'a ForestSortedIndices<'a>,
+        config: ConfigImpl,
+    ) -> ForestResult<(OriginalSkeletonForest<'a>, HashMap<NodeIndex, ContractState>)> {
+        let (contracts_trie, original_contracts_trie_leaves) =
+            create_contracts_trie::<IndexLayoutContractState, IndexNodeLayout>(
+                &mut self.storage,
+                contracts_trie_root_hash,
+                forest_sorted_indices.contracts_trie_sorted_indices,
+            )
+            .await?;
+        let storage_tries =
+            create_storage_tries::<IndexLayoutStarknetStorageValue, IndexNodeLayout>(
+                &mut self.storage,
+                storage_updates,
+                &original_contracts_trie_leaves,
+                &config,
+                &forest_sorted_indices.storage_tries_sorted_indices,
+            )
+            .await?;
+        let classes_trie = create_classes_trie::<IndexLayoutCompiledClassHash, IndexNodeLayout>(
+            &mut self.storage,
+            classes_updates,
+            classes_trie_root_hash,
+            &config,
+            forest_sorted_indices.classes_trie_sorted_indices,
+        )
+        .await?;
+
+        Ok((
+            OriginalSkeletonForest { classes_trie, contracts_trie, storage_tries },
+            original_contracts_trie_leaves,
+        ))
+    }
+}
+
+impl<S: Storage> ForestWriter for IndexDb<S> {
+    async fn write(&mut self, filled_forest: &FilledForest) -> usize {
+        filled_forest.write_to_storage(&mut self.storage).await
+    }
+}

crates/starknet_committer/src/db/index_db/mod.rs

@@ -1,2 +1,3 @@
+pub mod db;
 pub mod leaves;
 pub mod types;