You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
2.7 KiB
122 lines
2.7 KiB
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 |
|
}
|
|
|