Compare commits

..

4 Commits

Author SHA1 Message Date
amery 4a378c2334 cluster: assign valid rings.RegionID to each primary region
Signed-off-by: Alejandro Mery <amery@jpi.io>
2024-06-02 23:23:31 +00:00
amery 7b44f40416 cluster: use rings.ZoneID and rings.NodeID types
Signed-off-by: Alejandro Mery <amery@jpi.io>
2024-06-02 23:22:08 +00:00
amery 4812cb7d5d rings: fix NodeZeroMax/NodeMax to not accept .255
Signed-off-by: Alejandro Mery <amery@jpi.io>
2024-06-02 22:07:28 +00:00
amery fd3c9b5fa3 rings: add String() to RegionID, ZoneID, NodeID
Signed-off-by: Alejandro Mery <amery@jpi.io>
2024-06-02 21:28:16 +00:00
6 changed files with 22 additions and 48 deletions
+1 -2
View File
@@ -15,8 +15,7 @@ TMPDIR ?= .tmp
REVIVE ?= $(GOBIN)/revive REVIVE ?= $(GOBIN)/revive
REVIVE_CONF ?= $(TOOLSDIR)/revive.toml REVIVE_CONF ?= $(TOOLSDIR)/revive.toml
REVIVE_RUN_ARGS ?= -config $(REVIVE_CONF) -formatter friendly REVIVE_RUN_ARGS ?= -config $(REVIVE_CONF) -formatter friendly
REVIVE_VERSION ?= v1.3.7 REVIVE_INSTALL_URL ?= github.com/mgechev/revive@master
REVIVE_INSTALL_URL ?= github.com/mgechev/revive@$(REVIVE_VERSION)
GO_INSTALL_URLS = \ GO_INSTALL_URLS = \
$(REVIVE_INSTALL_URL) \ $(REVIVE_INSTALL_URL) \
+1 -1
View File
@@ -13,11 +13,11 @@ import (
func (m *Cluster) init(opts *ScanOptions) error { func (m *Cluster) init(opts *ScanOptions) error {
for _, fn := range []func(*ScanOptions) error{ for _, fn := range []func(*ScanOptions) error{
m.initZones, m.initZones,
m.initRegions,
m.scanZoneIDs, m.scanZoneIDs,
m.scanSort, m.scanSort,
m.scanGateways, m.scanGateways,
m.initCephMonitors, m.initCephMonitors,
m.initRegions,
} { } {
if err := fn(opts); err != nil { if err := fn(opts); err != nil {
return err return err
+1 -1
View File
@@ -25,9 +25,9 @@ func (m *Cluster) scan(opts *ScanOptions) error {
for _, fn := range []func(*ScanOptions) error{ for _, fn := range []func(*ScanOptions) error{
m.scanDirectory, m.scanDirectory,
m.scanMachines, m.scanMachines,
m.initRegions,
m.scanZoneIDs, m.scanZoneIDs,
m.scanSort, m.scanSort,
m.initRegions,
m.scanGateways, m.scanGateways,
m.scanCephMonitors, m.scanCephMonitors,
} { } {
+12 -4
View File
@@ -163,10 +163,18 @@ func genEnvZoneNodes(z *Zone) string {
} }
func genEnvZoneRegion(z *Zone) string { func genEnvZoneRegion(z *Zone) string {
if z != nil && z.region != nil { var region string
return z.region.Name
} z.ForEachRegion(func(r *Region) bool {
return "" if r.IsPrimary() {
region = r.Name
return true
}
return false
})
return region
} }
func genEnvZoneCephMonNames(m Machines) string { func genEnvZoneCephMonNames(m Machines) string {
+7 -39
View File
@@ -31,7 +31,7 @@ type Region struct {
Regions []string `json:",omitempty" yaml:",omitempty"` Regions []string `json:",omitempty" yaml:",omitempty"`
} }
// IsPrimary indicates the region is primary and corresponds // IsPrimary() indicates the region is primary and corresponds
// to a kubernetes cluster. // to a kubernetes cluster.
func (r *Region) IsPrimary() bool { func (r *Region) IsPrimary() bool {
return r != nil && r.Cluster != nil return r != nil && r.Cluster != nil
@@ -100,10 +100,7 @@ func (m *Cluster) initRegions(_ *ScanOptions) error {
m.finishRegion(r) m.finishRegion(r)
} }
m.sortRegions() return m.finishRegions()
m.scanRegionID()
m.computeZonesRegion()
return nil
} }
func (m *Cluster) setRegionZones(name string, zones ...*Zone) { func (m *Cluster) setRegionZones(name string, zones ...*Zone) {
@@ -221,12 +218,12 @@ func (m *Cluster) finishRegion(r *Region) {
r.Regions = sub r.Regions = sub
} }
// revive:disable:cognitive-complexity func (m *Cluster) finishRegions() error {
func (m *Cluster) scanRegionID() {
// revive:enable:cognitive-complexity
var max rings.RegionID var max rings.RegionID
var missing bool var missing bool
m.sortRegions()
// check IDs // check IDs
ids := make(map[rings.RegionID]bool) ids := make(map[rings.RegionID]bool)
fn := func(r *Region) bool { fn := func(r *Region) bool {
@@ -234,8 +231,7 @@ func (m *Cluster) scanRegionID() {
switch { switch {
case !r.IsPrimary(): case !r.IsPrimary():
// secondary, no ID. // ignore secondary
r.ID = 0
case !r.ID.Valid(): case !r.ID.Valid():
// primary without ID // primary without ID
missing = true missing = true
@@ -276,35 +272,7 @@ func (m *Cluster) scanRegionID() {
m.ForEachRegion(fn) m.ForEachRegion(fn)
} }
} return nil
func (m *Cluster) computeZonesRegion() {
fn := func(r *Region, z *Zone) {
if z.region != nil {
m.error(nil).
WithField("zone", z.Name).
WithField("region", []string{
z.region.Name,
r.Name,
}).
Print("zone in two regions")
} else {
z.region = r
}
}
m.ForEachRegion(func(r *Region) bool {
var term bool
if r.IsPrimary() {
r.ForEachZone(func(z *Zone) bool {
fn(r, z)
return term
})
}
return term
})
} }
func (m *Cluster) getRegion(name string) (*Region, bool) { func (m *Cluster) getRegion(name string) (*Region, bool) {
-1
View File
@@ -19,7 +19,6 @@ type ZoneIterator interface {
// affinity. // affinity.
type Zone struct { type Zone struct {
zones *Cluster zones *Cluster
region *Region
logger `json:"-" yaml:"-"` logger `json:"-" yaml:"-"`
ID rings.ZoneID ID rings.ZoneID