re-enable snapshot

gballet · Jul 3, 2023 · 8c6f9bf · 8c6f9bf
1 parent ad2fdb4
commit 8c6f9bf
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/core/state/statedb.go b/core/state/statedb.go
@@ -175,6 +175,17 @@ func New(root common.Hash, db Database, snaps *snapshot.Tree) (*StateDB, error)
 			sdb.snapDestructs = make(map[common.Hash]struct{})
 			sdb.snapAccounts = make(map[common.Hash][]byte)
 			sdb.snapStorage = make(map[common.Hash]map[common.Hash][]byte)
+		} else {
+			if fdb, ok := db.(*ForkingDB); ok {
+				trans := fdb.getTranslation(root)
+				if trans != (common.Hash{}) {
+					if sdb.snap = sdb.snaps.Snapshot(trans); sdb.snap != nil {
+						sdb.snapDestructs = make(map[common.Hash]struct{})
+						sdb.snapAccounts = make(map[common.Hash][]byte)
+						sdb.snapStorage = make(map[common.Hash]map[common.Hash][]byte)
+					}
+				}
+			}
 		}
 	}
 	return sdb, nil