From f1e8040bc33e9057bd7a756a09c431c3f0d86226 Mon Sep 17 00:00:00 2001
From: Luke Shumaker <lukeshu@lukeshu.com>
Date: Tue, 12 Jul 2022 23:52:37 -0600
Subject: Add an `Ordered` interface for the Cmp method, have rbtree use it

---
 lib/btrfs/btrfsvol/lvm.go | 36 ++++++++++++++++++------------------
 1 file changed, 18 insertions(+), 18 deletions(-)

(limited to 'lib/btrfs/btrfsvol')

diff --git a/lib/btrfs/btrfsvol/lvm.go b/lib/btrfs/btrfsvol/lvm.go
index 65636f6..ca46ad3 100644
--- a/lib/btrfs/btrfsvol/lvm.go
+++ b/lib/btrfs/btrfsvol/lvm.go
@@ -19,8 +19,8 @@ type LogicalVolume[PhysicalVolume util.File[PhysicalAddr]] struct {
 
 	id2pv map[DeviceID]PhysicalVolume
 
-	logical2physical *rbtree.Tree[LogicalAddr, chunkMapping]
-	physical2logical map[DeviceID]*rbtree.Tree[PhysicalAddr, devextMapping]
+	logical2physical *rbtree.Tree[util.NativeOrdered[LogicalAddr], chunkMapping]
+	physical2logical map[DeviceID]*rbtree.Tree[util.NativeOrdered[PhysicalAddr], devextMapping]
 }
 
 var _ util.File[LogicalAddr] = (*LogicalVolume[util.File[PhysicalAddr]])(nil)
@@ -30,20 +30,20 @@ func (lv *LogicalVolume[PhysicalVolume]) init() {
 		lv.id2pv = make(map[DeviceID]PhysicalVolume)
 	}
 	if lv.logical2physical == nil {
-		lv.logical2physical = &rbtree.Tree[LogicalAddr, chunkMapping]{
-			KeyFn: func(chunk chunkMapping) LogicalAddr {
-				return chunk.LAddr
+		lv.logical2physical = &rbtree.Tree[util.NativeOrdered[LogicalAddr], chunkMapping]{
+			KeyFn: func(chunk chunkMapping) util.NativeOrdered[LogicalAddr] {
+				return util.NativeOrdered[LogicalAddr]{Val: chunk.LAddr}
 			},
 		}
 	}
 	if lv.physical2logical == nil {
-		lv.physical2logical = make(map[DeviceID]*rbtree.Tree[PhysicalAddr, devextMapping], len(lv.id2pv))
+		lv.physical2logical = make(map[DeviceID]*rbtree.Tree[util.NativeOrdered[PhysicalAddr], devextMapping], len(lv.id2pv))
 	}
 	for devid := range lv.id2pv {
 		if _, ok := lv.physical2logical[devid]; !ok {
-			lv.physical2logical[devid] = &rbtree.Tree[PhysicalAddr, devextMapping]{
-				KeyFn: func(ext devextMapping) PhysicalAddr {
-					return ext.PAddr
+			lv.physical2logical[devid] = &rbtree.Tree[util.NativeOrdered[PhysicalAddr], devextMapping]{
+				KeyFn: func(ext devextMapping) util.NativeOrdered[PhysicalAddr] {
+					return util.NativeOrdered[PhysicalAddr]{Val: ext.PAddr}
 				},
 			}
 		}
@@ -74,9 +74,9 @@ func (lv *LogicalVolume[PhysicalVolume]) AddPhysicalVolume(id DeviceID, dev Phys
 			lv, dev.Name(), other.Name(), id)
 	}
 	lv.id2pv[id] = dev
-	lv.physical2logical[id] = &rbtree.Tree[PhysicalAddr, devextMapping]{
-		KeyFn: func(ext devextMapping) PhysicalAddr {
-			return ext.PAddr
+	lv.physical2logical[id] = &rbtree.Tree[util.NativeOrdered[PhysicalAddr], devextMapping]{
+		KeyFn: func(ext devextMapping) util.NativeOrdered[PhysicalAddr] {
+			return util.NativeOrdered[PhysicalAddr]{Val: ext.PAddr}
 		},
 	}
 	return nil
@@ -148,13 +148,13 @@ func (lv *LogicalVolume[PhysicalVolume]) AddMapping(m Mapping) error {
 
 	// logical2physical
 	for _, chunk := range logicalOverlaps {
-		lv.logical2physical.Delete(chunk.LAddr)
+		lv.logical2physical.Delete(util.NativeOrdered[LogicalAddr]{Val: chunk.LAddr})
 	}
 	lv.logical2physical.Insert(newChunk)
 
 	// physical2logical
 	for _, ext := range physicalOverlaps {
-		lv.physical2logical[m.PAddr.Dev].Delete(ext.PAddr)
+		lv.physical2logical[m.PAddr.Dev].Delete(util.NativeOrdered[PhysicalAddr]{Val: ext.PAddr})
 	}
 	lv.physical2logical[m.PAddr.Dev].Insert(newExt)
 
@@ -173,7 +173,7 @@ func (lv *LogicalVolume[PhysicalVolume]) AddMapping(m Mapping) error {
 }
 
 func (lv *LogicalVolume[PhysicalVolume]) fsck() error {
-	physical2logical := make(map[DeviceID]*rbtree.Tree[PhysicalAddr, devextMapping])
+	physical2logical := make(map[DeviceID]*rbtree.Tree[util.NativeOrdered[PhysicalAddr], devextMapping])
 	if err := lv.logical2physical.Walk(func(node *rbtree.Node[chunkMapping]) error {
 		chunk := node.Value
 		for _, stripe := range chunk.PAddrs {
@@ -182,9 +182,9 @@ func (lv *LogicalVolume[PhysicalVolume]) fsck() error {
 					lv, stripe.Dev)
 			}
 			if _, exists := physical2logical[stripe.Dev]; !exists {
-				physical2logical[stripe.Dev] = &rbtree.Tree[PhysicalAddr, devextMapping]{
-					KeyFn: func(ext devextMapping) PhysicalAddr {
-						return ext.PAddr
+				physical2logical[stripe.Dev] = &rbtree.Tree[util.NativeOrdered[PhysicalAddr], devextMapping]{
+					KeyFn: func(ext devextMapping) util.NativeOrdered[PhysicalAddr] {
+						return util.NativeOrdered[PhysicalAddr]{Val: ext.PAddr}
 					},
 				}
 			}
-- 
cgit v1.2.3-2-g168b