Fix block_revert and data_restore crates

derekzTu · Mar 28, 2022 · cfa54ca · cfa54ca
1 parent cdb1195
commit cfa54ca
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/core/bin/block_revert/src/main.rs b/core/bin/block_revert/src/main.rs
@@ -45,7 +45,7 @@ async fn revert_blocks_in_storage(
     println!("`pending_block` table is cleaned");
     transaction
         .chain()
-        .block_schema()
+        .tree_cache_schema_bincode()
         .remove_new_account_tree_cache(last_block)
         .await?;
     println!("`account_tree_cache` table is cleaned");

diff --git a/core/bin/data_restore/src/database_storage_interactor.rs b/core/bin/data_restore/src/database_storage_interactor.rs
@@ -350,7 +350,7 @@ impl<'a> DatabaseStorageInteractor<'a> {
         let tree_cache = self
             .storage
             .chain()
-            .block_schema()
+            .tree_cache_schema_json()
             .get_account_tree_cache_block(last_block)
             .await
             .expect("Failed to query the database for the tree cache");
@@ -390,7 +390,7 @@ impl<'a> DatabaseStorageInteractor<'a> {
 
         transaction
             .chain()
-            .block_schema()
+            .tree_cache_schema_json()
             .remove_old_account_tree_cache(block_number)
             .await
             .expect("Failed to remove old tree cache");
@@ -399,7 +399,7 @@ impl<'a> DatabaseStorageInteractor<'a> {
         // since on conflict it does nothing.
         transaction
             .chain()
-            .block_schema()
+            .tree_cache_schema_json()
             .store_account_tree_cache(block_number, tree_cache)
             .await
             .expect("Failed to store new tree cache");