From 6a1b5c780b8fbb9ca0285286036096960458d4e6 Mon Sep 17 00:00:00 2001 From: Luke Shumaker Date: Sun, 26 Jun 2022 15:53:49 -0600 Subject: Add an AddrDelta type for safer arithmentic --- pkg/btrfs/internal/addr.go | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'pkg/btrfs/internal') diff --git a/pkg/btrfs/internal/addr.go b/pkg/btrfs/internal/addr.go index 7067982..a3f41ed 100644 --- a/pkg/btrfs/internal/addr.go +++ b/pkg/btrfs/internal/addr.go @@ -9,6 +9,7 @@ import ( type ( PhysicalAddr int64 LogicalAddr int64 + AddrDelta int64 ) func formatAddr(addr int64, f fmt.State, verb rune) { @@ -23,3 +24,10 @@ func formatAddr(addr int64, f fmt.State, verb rune) { func (a PhysicalAddr) Format(f fmt.State, verb rune) { formatAddr(int64(a), f, verb) } func (a LogicalAddr) Format(f fmt.State, verb rune) { formatAddr(int64(a), f, verb) } +func (d AddrDelta) Format(f fmt.State, verb rune) { formatAddr(int64(d), f, verb) } + +func (a PhysicalAddr) Sub(b PhysicalAddr) AddrDelta { return AddrDelta(a - b) } +func (a LogicalAddr) Sub(b LogicalAddr) AddrDelta { return AddrDelta(a - b) } + +func (a PhysicalAddr) Add(b AddrDelta) PhysicalAddr { return a + PhysicalAddr(b) } +func (a LogicalAddr) Add(b AddrDelta) LogicalAddr { return a + LogicalAddr(b) } -- cgit v1.2.3-2-g168b