amend

robin-near · Nov 3, 2023 · ee6d560 · ee6d560
1 parent 2285336
commit ee6d560
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/core/store/src/trie/mem/loading.rs b/core/store/src/trie/mem/loading.rs
@@ -258,7 +258,7 @@ mod tests {
             // Check that the accessed nodes are consistent with those from disk.
             for (node_hash, serialized_node) in nodes_accessed {
                 let expected_serialized_node =
-                    trie.internal_retrieve_trie_node(&node_hash, false).unwrap().to_vec();
+                    trie.internal_retrieve_trie_node(&node_hash, false).unwrap();
                 assert_eq!(expected_serialized_node, serialized_node);
             }
         }

diff --git a/core/store/src/trie/mem/lookup.rs b/core/store/src/trie/mem/lookup.rs
@@ -1,3 +1,5 @@
+use std::sync::Arc;
+
 use super::metrics::MEM_TRIE_NUM_LOOKUPS;
 use super::node::{MemTrieNodePtr, MemTrieNodeView};
 use crate::NibbleSlice;
@@ -19,7 +21,7 @@ use near_primitives::state::FlatStateValue;
 pub fn memtrie_lookup(
     root: MemTrieNodePtr<'_>,
     key: &[u8],
-    mut nodes_accessed: Option<&mut Vec<(CryptoHash, Vec<u8>)>>,
+    mut nodes_accessed: Option<&mut Vec<(CryptoHash, Arc<[u8]>)>>,
 ) -> Option<FlatStateValue> {
     MEM_TRIE_NUM_LOOKUPS.inc();
     let mut nibbles = NibbleSlice::new(key);
@@ -29,7 +31,7 @@ pub fn memtrie_lookup(
         let view = node.view();
         if let Some(nodes_accessed) = &mut nodes_accessed {
             let raw_node_serialized = borsh::to_vec(&view.to_raw_trie_node_with_size()).unwrap();
-            nodes_accessed.push((view.node_hash(), raw_node_serialized));
+            nodes_accessed.push((view.node_hash(), raw_node_serialized.into()));
         }
         match view {
             MemTrieNodeView::Leaf { extension, value } => {