feat: reth drop-stage merkle (paradigmxyz#1941)

smtmfft · Mar 23, 2023 · 1ce021f · 1ce021f
1 parent d580b5a
commit 1ce021f
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/bin/reth/src/db/mod.rs b/bin/reth/src/db/mod.rs
@@ -172,6 +172,7 @@ impl Command {
                     BlockTransitionIndex,
                     TxTransitionIndex,
                     SyncStage,
+                    SyncStageProgress,
                     Transactions
                 ]);
             }

diff --git a/bin/reth/src/drop_stage.rs b/bin/reth/src/drop_stage.rs
@@ -72,6 +72,18 @@ impl Command {
                     Ok::<_, eyre::Error>(())
                 })??;
             }
+            StageEnum::Merkle => {
+                tool.db.update(|tx| {
+                    tx.clear::<tables::AccountsTrie>()?;
+                    tx.clear::<tables::StoragesTrie>()?;
+                    tx.put::<tables::SyncStageProgress>(
+                        // TODO: Extract to constant in `TrieLoader` in trie/mod.rs
+                        "TrieLoader".to_string(),
+                        Vec::new(),
+                    )?;
+                    Ok::<_, eyre::Error>(())
+                })??;
+            }
             _ => {
                 info!("Nothing to do for stage {:?}", self.stage);
             }

diff --git a/bin/reth/src/lib.rs b/bin/reth/src/lib.rs
@@ -29,4 +29,5 @@ enum StageEnum {
     Bodies,
     Senders,
     Execution,
+    Merkle,
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -29,4 +29,5 @@ enum StageEnum { @@
         Bodies,
         Senders,
         Execution,
+        Merkle,
     }