From 6630e28213a6c5f506d6a6d6f3e764a42c967163 Mon Sep 17 00:00:00 2001
From: Luke Shumaker <lukeshu@lukeshu.com>
Date: Sun, 5 Mar 2023 11:40:32 -0700
Subject: btrfsutil: RebuiltForrest: Add a .RebuiltAddRoots() method,
 cmd/btrfs-rec: Add a --trees flag

---
 lib/btrfsutil/rebuilt_tree.go | 4 ++++
 1 file changed, 4 insertions(+)

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

diff --git a/lib/btrfsutil/rebuilt_tree.go b/lib/btrfsutil/rebuilt_tree.go
index 97308a3..f002ea6 100644
--- a/lib/btrfsutil/rebuilt_tree.go
+++ b/lib/btrfsutil/rebuilt_tree.go
@@ -700,6 +700,10 @@ func (tree *RebuiltTree) RebuiltAddRoot(ctx context.Context, rootNode btrfsvol.L
 	tree.mu.Lock()
 	defer tree.mu.Unlock()
 
+	if tree.Roots.Has(rootNode) {
+		return
+	}
+
 	ctx = dlog.WithField(ctx, "btrfs.util.rebuilt-tree.add-root", fmt.Sprintf("tree=%v rootNode=%v", tree.ID, rootNode))
 	dlog.Info(ctx, "adding root...")
 
-- 
cgit v1.2.3-2-g168b