From 5590315b11557aa48999d09700631ad9239ce03b Mon Sep 17 00:00:00 2001
From: Luke Shumaker <lukeshu@lukeshu.com>
Date: Tue, 27 Dec 2022 01:14:34 -0700
Subject: rebuildnodes: Optimize: Try to avoid disk access for DIR_INDEX

---
 lib/btrfsprogs/btrfsinspect/rebuildnodes/keyio/keyio.go | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'lib/btrfsprogs/btrfsinspect/rebuildnodes/keyio')

diff --git a/lib/btrfsprogs/btrfsinspect/rebuildnodes/keyio/keyio.go b/lib/btrfsprogs/btrfsinspect/rebuildnodes/keyio/keyio.go
index 24c3dcf..eb03004 100644
--- a/lib/btrfsprogs/btrfsinspect/rebuildnodes/keyio/keyio.go
+++ b/lib/btrfsprogs/btrfsinspect/rebuildnodes/keyio/keyio.go
@@ -39,6 +39,7 @@ type Handle struct {
 	sb      btrfstree.Superblock
 	graph   graph.Graph
 
+	Names map[ItemPtr][]byte
 	Sizes map[ItemPtr]SizeAndErr
 
 	cache *containers.LRUCache[btrfsvol.LogicalAddr, *diskio.Ref[btrfsvol.LogicalAddr, btrfstree.Node]]
@@ -49,6 +50,7 @@ func NewHandle(file diskio.File[btrfsvol.LogicalAddr], sb btrfstree.Superblock)
 		rawFile: file,
 		sb:      sb,
 
+		Names: make(map[ItemPtr][]byte),
 		Sizes: make(map[ItemPtr]SizeAndErr),
 
 		cache: containers.NewLRUCache[btrfsvol.LogicalAddr, *diskio.Ref[btrfsvol.LogicalAddr, btrfstree.Node]](textui.Tunable(8)),
@@ -62,6 +64,10 @@ func (o *Handle) InsertNode(nodeRef *diskio.Ref[btrfsvol.LogicalAddr, btrfstree.
 			Idx:  i,
 		}
 		switch itemBody := item.Body.(type) {
+		case btrfsitem.DirEntry:
+			if item.Key.ItemType == btrfsprim.DIR_INDEX_KEY {
+				o.Names[ptr] = append([]byte(nil), itemBody.Name...)
+			}
 		case btrfsitem.ExtentCSum:
 			o.Sizes[ptr] = SizeAndErr{
 				Size: uint64(itemBody.Size()),
-- 
cgit v1.2.3-2-g168b