rings: introduce RingID and DecodeAddress() #50
Merged
amery
merged 3 commits from pr-amery-rings
into main
3 months ago
3 changed files with 201 additions and 0 deletions
@ -0,0 +1,122 @@ |
|||||||
|
package rings |
||||||
|
|
||||||
|
import ( |
||||||
|
"net/netip" |
||||||
|
) |
||||||
|
|
||||||
|
// DecodeAddress extracts ring address fields from a given 10.0.0.0/8
|
||||||
|
// address.
|
||||||
|
//
|
||||||
|
// revive:disable:function-result-limit
|
||||||
|
func DecodeAddress[T ~uint | NodeID](addr netip.Addr) (RingID, RegionID, ZoneID, T) { |
||||||
|
// revive:enable:function-result-limit
|
||||||
|
if addr.IsValid() { |
||||||
|
if addr.Is4In6() { |
||||||
|
addr = addr.Unmap() |
||||||
|
} |
||||||
|
|
||||||
|
if addr.Is4() { |
||||||
|
a4 := addr.As4() |
||||||
|
return unsafeDecodeAddress[T](a4[0], a4[1], a4[2], a4[3]) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return UnspecifiedRingID, 0, 0, 0 |
||||||
|
} |
||||||
|
|
||||||
|
// revive:disable:function-result-limit
|
||||||
|
func unsafeDecodeAddress[T ~uint | NodeID](a, b, c, d byte) (RingID, RegionID, ZoneID, T) { |
||||||
|
// revive:enable:function-result-limit
|
||||||
|
switch { |
||||||
|
case a != 10: |
||||||
|
return UnspecifiedRingID, 0, 0, 0 |
||||||
|
case b == 0x00: |
||||||
|
// 10.00.RZ.dd
|
||||||
|
k := RingZeroID |
||||||
|
r := RegionID(c >> 4) |
||||||
|
z := ZoneID(c & 0xf) |
||||||
|
n := T(d) |
||||||
|
|
||||||
|
return k, r, z, n |
||||||
|
case b&0xf0 != 0: |
||||||
|
// 10.Rb.cc.dd
|
||||||
|
k := RingThreeID |
||||||
|
r := RegionID(b >> 4) |
||||||
|
|
||||||
|
n2 := T(b & 0x0f) |
||||||
|
n1 := T(c) |
||||||
|
n0 := T(d) |
||||||
|
n := n0 + n1<<8 + n2<<16 |
||||||
|
|
||||||
|
return k, r, 0, n |
||||||
|
case c&0xf0 != 0: |
||||||
|
// 10.0R.Zc.dd
|
||||||
|
k := RingOneID |
||||||
|
r := RegionID(b) |
||||||
|
z := ZoneID(c >> 4) |
||||||
|
|
||||||
|
n1 := T(c & 0x0f) |
||||||
|
n0 := T(d) |
||||||
|
n := n0 + n1<<8 |
||||||
|
|
||||||
|
return k, r, z, n |
||||||
|
default: |
||||||
|
// 10.0R.0c.dd
|
||||||
|
k := RingTwoID |
||||||
|
r := RegionID(b) |
||||||
|
|
||||||
|
n1 := T(c & 0x0f) |
||||||
|
n0 := T(d) |
||||||
|
n := n0 + n1<<8 |
||||||
|
|
||||||
|
return k, r, 0, n |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// DecodeRingZeroAddress attempts to extract region, zone and node identifiers
|
||||||
|
// from a given ring 0 address.
|
||||||
|
//
|
||||||
|
// revive:disable:function-result-limit
|
||||||
|
func DecodeRingZeroAddress(addr netip.Addr) (RegionID, ZoneID, NodeID, bool) { |
||||||
|
// revive:enable:function-result-limit
|
||||||
|
k, r, z, n := DecodeAddress[NodeID](addr) |
||||||
|
if k == RingZeroID { |
||||||
|
return r, z, n, true |
||||||
|
} |
||||||
|
|
||||||
|
return 0, 0, 0, false |
||||||
|
} |
||||||
|
|
||||||
|
// DecodeRingOneAddress attempts to extract region, zone and node identifiers
|
||||||
|
// from a given ring 1 address.
|
||||||
|
//
|
||||||
|
// revive:disable:function-result-limit
|
||||||
|
func DecodeRingOneAddress(addr netip.Addr) (RegionID, ZoneID, NodeID, bool) { |
||||||
|
// revive:enable:function-result-limit
|
||||||
|
k, r, z, n := DecodeAddress[NodeID](addr) |
||||||
|
if k == RingOneID { |
||||||
|
return r, z, n, true |
||||||
|
} |
||||||
|
|
||||||
|
return 0, 0, 0, false |
||||||
|
} |
||||||
|
|
||||||
|
// DecodeRingTwoAddress attempts to extract region and unique identifier for
|
||||||
|
// a kubernetes service from a given ring 2 address.
|
||||||
|
func DecodeRingTwoAddress(addr netip.Addr) (RegionID, uint, bool) { |
||||||
|
k, r, _, n := DecodeAddress[uint](addr) |
||||||
|
if k == RingTwoID { |
||||||
|
return r, n, true |
||||||
|
} |
||||||
|
return 0, 0, false |
||||||
|
} |
||||||
|
|
||||||
|
// DecodeRingThreeAddress attempts to extract region and unique identifier for
|
||||||
|
// a kubernetes pod from a given ring 3 address.
|
||||||
|
func DecodeRingThreeAddress(addr netip.Addr) (RegionID, uint, bool) { |
||||||
|
k, r, _, n := DecodeAddress[uint](addr) |
||||||
|
if k == RingThreeID { |
||||||
|
return r, n, true |
||||||
|
} |
||||||
|
return 0, 0, false |
||||||
|
} |
@ -0,0 +1,53 @@ |
|||||||
|
package rings |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"net/netip" |
||||||
|
"testing" |
||||||
|
) |
||||||
|
|
||||||
|
func TestDecodeRingZeroAddress(t *testing.T) { |
||||||
|
RZNDecodeTest(t, "DecodeRingZeroAddress", DecodeRingZeroAddress, []RZNDecodeTestCase{ |
||||||
|
{1, 1, 50, MustParseAddr("10.0.17.50"), true}, |
||||||
|
{1, 2, 50, MustParseAddr("10.0.18.50"), true}, |
||||||
|
{2, 3, 1, MustParseAddr("10.0.35.1"), true}, |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
func TesDecodetRingOneAddress(t *testing.T) { |
||||||
|
RZNDecodeTest(t, "DecodeRingOneAddress", DecodeRingOneAddress, []RZNDecodeTestCase{ |
||||||
|
{1, 1, 50, MustParseAddr("10.1.16.50"), true}, |
||||||
|
{1, 2, 50, MustParseAddr("10.1.32.50"), true}, |
||||||
|
{2, 3, 300, MustParseAddr("10.2.49.44"), true}, |
||||||
|
}) |
||||||
|
} |
||||||
|
|
||||||
|
type RZNDecodeTestCase struct { |
||||||
|
region RegionID |
||||||
|
zone ZoneID |
||||||
|
node NodeID |
||||||
|
addr netip.Addr |
||||||
|
ok bool |
||||||
|
} |
||||||
|
|
||||||
|
func RZNDecodeTest(t *testing.T, |
||||||
|
fnName string, fn func(netip.Addr) (RegionID, ZoneID, NodeID, bool), |
||||||
|
cases []RZNDecodeTestCase) { |
||||||
|
//
|
||||||
|
for i, tc := range cases { |
||||||
|
s := fmt.Sprintf("%s(%q)", fnName, tc.addr) |
||||||
|
|
||||||
|
r, z, n, ok := fn(tc.addr) |
||||||
|
|
||||||
|
switch { |
||||||
|
case ok != tc.ok, r != tc.region, z != tc.zone, n != tc.node: |
||||||
|
t.Errorf("ERROR: [%v/%v]: %s → %v %v %v %v (expected %v %v %v %v)", |
||||||
|
i, len(cases), s, |
||||||
|
r, z, n, ok, |
||||||
|
tc.region, tc.zone, tc.node, tc.ok) |
||||||
|
default: |
||||||
|
t.Logf("[%v/%v]: %s → %v %v %v %v", i, len(cases), s, |
||||||
|
r, z, n, ok) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue