From 94aa0ec3e9f7145cdf177ad6f6d3d8b7d5bdbef7 Mon Sep 17 00:00:00 2001
From: Luke Shumaker <lukeshu@lukeshu.com>
Date: Wed, 29 Mar 2023 10:00:18 -0600
Subject: btrfstree: Change the NodeSource API so that it can do caching

---
 lib/btrfsutil/graph.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'lib/btrfsutil/graph.go')

diff --git a/lib/btrfsutil/graph.go b/lib/btrfsutil/graph.go
index 4148f89..ea33be7 100644
--- a/lib/btrfsutil/graph.go
+++ b/lib/btrfsutil/graph.go
@@ -207,9 +207,10 @@ func (g Graph) FinalCheck(ctx context.Context, fs btrfstree.NodeSource) error {
 	progressWriter.Set(stats)
 	for laddr := range g.EdgesTo {
 		if _, ok := g.Nodes[laddr]; !ok {
-			_, err := fs.ReadNode(ctx, laddr, btrfstree.NodeExpectations{
+			node, err := fs.AcquireNode(ctx, laddr, btrfstree.NodeExpectations{
 				LAddr: containers.OptionalValue(laddr),
 			})
+			fs.ReleaseNode(node)
 			if err == nil {
 				progressWriter.Done()
 				return fmt.Errorf("node@%v exists but was not in node scan results", laddr)
-- 
cgit v1.2.3-54-g00ecf