chore: refine some code

will-2012 · Jun 12, 2024 · 28bcfe0 · 28bcfe0
1 parent 00e8f9d
commit 28bcfe0
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/triedb/pathdb/difflayer.go b/triedb/pathdb/difflayer.go
@@ -166,19 +166,16 @@ func newDiffLayer(parent layer, root common.Hash, id uint64, block uint64, nodes
 	switch l := parent.(type) {
 	case *diskLayer:
 		dl.origin = l
+		dl.cache = &HashNodeCache{
+			cache: make(map[common.Hash]*RefTrieNode),
+		}
 	case *diffLayer:
-		dl.origin = l.origin
+		dl.origin = l.originDiskLayer()
 		dl.cache = l.cache
 	default:
 		panic("unknown parent type")
 	}
 
-	if dl.cache == nil {
-		dl.cache = &HashNodeCache{
-			cache: make(map[common.Hash]*RefTrieNode),
-		}
-	}
-
 	for _, subset := range nodes {
 		for path, n := range subset {
 			dl.memory += uint64(n.Size() + len(path))