From 30be1eacbe4ff253c82c6e6163234dc20b4de550 Mon Sep 17 00:00:00 2001
From: Luke Shumaker <lukeshu@lukeshu.com>
Date: Wed, 21 Dec 2022 05:30:25 -0700
Subject: rebuildnodes: Refactor existing key-io code in to a sub-package

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

(limited to 'lib/btrfsprogs/btrfsinspect/rebuildnodes/scan.go')

diff --git a/lib/btrfsprogs/btrfsinspect/rebuildnodes/scan.go b/lib/btrfsprogs/btrfsinspect/rebuildnodes/scan.go
index 3575534..7e01693 100644
--- a/lib/btrfsprogs/btrfsinspect/rebuildnodes/scan.go
+++ b/lib/btrfsprogs/btrfsinspect/rebuildnodes/scan.go
@@ -16,6 +16,7 @@ import (
 	"git.lukeshu.com/btrfs-progs-ng/lib/btrfs/btrfsvol"
 	"git.lukeshu.com/btrfs-progs-ng/lib/btrfsprogs/btrfsinspect"
 	"git.lukeshu.com/btrfs-progs-ng/lib/btrfsprogs/btrfsinspect/rebuildnodes/graph"
+	"git.lukeshu.com/btrfs-progs-ng/lib/btrfsprogs/btrfsinspect/rebuildnodes/keyio"
 	"git.lukeshu.com/btrfs-progs-ng/lib/btrfsprogs/btrfsinspect/rebuildnodes/uuidmap"
 	"git.lukeshu.com/btrfs-progs-ng/lib/containers"
 	"git.lukeshu.com/btrfs-progs-ng/lib/maps"
@@ -25,6 +26,7 @@ import (
 type scanResult struct {
 	uuidMap   *uuidmap.UUIDMap
 	nodeGraph *graph.Graph
+	keyIO     *keyio.Handle
 }
 
 type scanStats struct {
@@ -56,6 +58,7 @@ func ScanDevices(ctx context.Context, fs *btrfs.FS, scanResults btrfsinspect.Sca
 		uuidMap:   uuidmap.New(),
 		nodeGraph: graph.New(*sb),
 	}
+	ret.keyIO = keyio.NewHandle(fs, *sb, ret.nodeGraph)
 
 	progress(done, total)
 	for _, devResults := range scanResults {
@@ -72,6 +75,7 @@ func ScanDevices(ctx context.Context, fs *btrfs.FS, scanResults btrfsinspect.Sca
 			}
 
 			ret.nodeGraph.InsertNode(nodeRef)
+			ret.keyIO.InsertNode(nodeRef)
 
 			done++
 			progress(done, total)
-- 
cgit v1.2.3-2-g168b