|
|
|
|
@ -7,6 +7,8 @@ import (
|
|
|
|
|
"sort" |
|
|
|
|
"strings" |
|
|
|
|
|
|
|
|
|
"darvaza.org/core" |
|
|
|
|
|
|
|
|
|
"git.jpi.io/amery/jpictl/pkg/rings" |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
@ -36,20 +38,24 @@ func (m *Cluster) Env(export bool) (*Env, error) {
|
|
|
|
|
return env, nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Zones returns the list of Zone IDs
|
|
|
|
|
func (m *Env) Zones() []rings.ZoneID { |
|
|
|
|
// Zones returns the list of Zone IDs of a region,
|
|
|
|
|
// or from all if none is specified.
|
|
|
|
|
func (m *Env) Zones(r *Region) []rings.ZoneID { |
|
|
|
|
var zones []rings.ZoneID |
|
|
|
|
|
|
|
|
|
m.ForEachZone(func(z *Zone) bool { |
|
|
|
|
iter := core.IIf[ZoneIterator](r != nil, r, m) |
|
|
|
|
|
|
|
|
|
iter.ForEachZone(func(z *Zone) bool { |
|
|
|
|
zones = append(zones, z.ID) |
|
|
|
|
return false |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
core.SliceSort(zones, cmpOrdered[rings.ZoneID]) |
|
|
|
|
return zones |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Regions returns the list of primary regions
|
|
|
|
|
func (m *Env) Regions() []string { |
|
|
|
|
// RegionsNames returns a sorted list of primary regions names
|
|
|
|
|
func (m *Env) RegionsNames() []string { |
|
|
|
|
var regions []string |
|
|
|
|
|
|
|
|
|
m.ForEachRegion(func(r *Region) bool { |
|
|
|
|
@ -64,6 +70,20 @@ func (m *Env) Regions() []string {
|
|
|
|
|
return regions |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Regions returns a sorted list of primary regions IDs
|
|
|
|
|
func (m *Env) Regions() (regions []rings.RegionID) { |
|
|
|
|
m.ForEachRegion(func(r *Region) bool { |
|
|
|
|
if r.IsPrimary() { |
|
|
|
|
regions = append(regions, r.ID) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
return false |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
core.SliceSort(regions, cmpOrdered[rings.RegionID]) |
|
|
|
|
return regions |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// WriteTo generates environment variables for shell scripts
|
|
|
|
|
func (m *Env) WriteTo(w io.Writer) (int64, error) { |
|
|
|
|
var buf bytes.Buffer |
|
|
|
|
@ -72,41 +92,75 @@ func (m *Env) WriteTo(w io.Writer) (int64, error) {
|
|
|
|
|
m.writeEnvVar(&buf, m.cephFSID, "FSID") |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
m.writeEnvVar(&buf, genEnvStrings(m.Regions()), "REGIONS") |
|
|
|
|
m.writeEnvVar(&buf, genEnvInts(m.Zones()), "ZONES") |
|
|
|
|
regions := m.getRegions() |
|
|
|
|
ids := core.SliceMap(regions, func(_ []rings.RegionID, r *Region) (out []rings.RegionID) { |
|
|
|
|
return append(out, r.ID) |
|
|
|
|
}) |
|
|
|
|
names := core.SliceMap(regions, func(_ []string, r *Region) (out []string) { |
|
|
|
|
return append(out, r.Name) |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
m.writeEnvVar(&buf, genEnvInts(ids), "REGIONS") |
|
|
|
|
m.writeEnvVar(&buf, genEnvStrings(names), "REGIONS_NAMES") |
|
|
|
|
|
|
|
|
|
m.ForEachZone(func(z *Zone) bool { |
|
|
|
|
m.writeEnvZone(&buf, z) |
|
|
|
|
for _, r := range regions { |
|
|
|
|
m.writeEnvRegion(&buf, r) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
return buf.WriteTo(w) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (m *Env) getRegions() (out []*Region) { |
|
|
|
|
m.ForEachRegion(func(r *Region) bool { |
|
|
|
|
if r.IsPrimary() { |
|
|
|
|
out = append(out, r) |
|
|
|
|
} |
|
|
|
|
return false |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
return buf.WriteTo(w) |
|
|
|
|
core.SliceSort(out, func(a, b *Region) int { |
|
|
|
|
return cmpOrdered(a.ID, b.ID) |
|
|
|
|
}) |
|
|
|
|
|
|
|
|
|
return out |
|
|
|
|
} |
|
|
|
|
func (m *Env) writeEnvRegion(w io.Writer, r *Region) { |
|
|
|
|
regionID := r.ID |
|
|
|
|
|
|
|
|
|
// REGION{regionID}_NAME
|
|
|
|
|
m.writeEnvVar(w, r.Name, "REGION%v_%s", regionID, "NAME") |
|
|
|
|
|
|
|
|
|
// REGION{regionID}_ZONES
|
|
|
|
|
m.writeEnvVar(w, genEnvInts(m.Zones(r)), "REGION%v_%s", regionID, "ZONES") |
|
|
|
|
|
|
|
|
|
r.ForEachZone(func(z *Zone) bool { |
|
|
|
|
m.writeEnvZone(w, r, z) |
|
|
|
|
return false |
|
|
|
|
}) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (m *Env) writeEnvZone(w io.Writer, z *Zone) { |
|
|
|
|
zoneID := z.ID |
|
|
|
|
func (m *Env) writeEnvZone(w io.Writer, r *Region, z *Zone) { |
|
|
|
|
zonePrefix := fmt.Sprintf("REGION%v_ZONE%v", r.ID, z.ID) |
|
|
|
|
monPrefix := zonePrefix + "_MON" |
|
|
|
|
|
|
|
|
|
// ZONE{zoneID}
|
|
|
|
|
m.writeEnvVar(w, genEnvZoneNodes(z), "ZONE%v", zoneID) |
|
|
|
|
// REGION{regionID}_ZONE{zoneID}
|
|
|
|
|
m.writeEnvVar(w, genEnvZoneNodes(z), zonePrefix) |
|
|
|
|
|
|
|
|
|
// ZONE{zoneID}_NAME
|
|
|
|
|
m.writeEnvVar(w, z.Name, "ZONE%v_%s", zoneID, "NAME") |
|
|
|
|
// REGION{regionID}_ZONE{zoneID}_NAME
|
|
|
|
|
m.writeEnvVar(w, z.Name, zonePrefix+"_NAME") |
|
|
|
|
|
|
|
|
|
// ZONE{zoneID}_GW
|
|
|
|
|
// REGION{regionID}_ZONE{zoneID}_GW
|
|
|
|
|
gateways, _ := z.GatewayIDs() |
|
|
|
|
m.writeEnvVar(w, genEnvInts(gateways), "ZONE%v_%s", zoneID, "GW") |
|
|
|
|
|
|
|
|
|
// ZONE{zoneID}_REGION
|
|
|
|
|
m.writeEnvVar(w, genEnvZoneRegion(z), "ZONE%v_%s", zoneID, "REGION") |
|
|
|
|
m.writeEnvVar(w, genEnvInts(gateways), zonePrefix+"_GW") |
|
|
|
|
|
|
|
|
|
// Ceph
|
|
|
|
|
monitors := z.GetCephMonitors() |
|
|
|
|
// MON{zoneID}_NAME
|
|
|
|
|
m.writeEnvVar(w, genEnvZoneCephMonNames(monitors), "MON%v_%s", zoneID, "NAME") |
|
|
|
|
// MON{zoneID}_IP
|
|
|
|
|
m.writeEnvVar(w, genEnvZoneCephMonIPs(monitors), "MON%v_%s", zoneID, "IP") |
|
|
|
|
// MON{zoneID}_ID
|
|
|
|
|
m.writeEnvVar(w, genEnvZoneCephMonIDs(monitors), "MON%v_%s", zoneID, "ID") |
|
|
|
|
// REGION{regionID}_MON{zone_ID}
|
|
|
|
|
m.writeEnvVar(w, genEnvZoneCephMonNames(monitors), monPrefix) |
|
|
|
|
// REGION{regionID}_MON{zone_ID}_IP
|
|
|
|
|
m.writeEnvVar(w, genEnvZoneCephMonIPs(monitors), monPrefix+"_IP") |
|
|
|
|
// REGION{regionID}_MON{zone_ID}_ID
|
|
|
|
|
m.writeEnvVar(w, genEnvZoneCephMonIDs(monitors), monPrefix+"_ID") |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (m *Env) writeEnvVar(w io.Writer, value string, name string, args ...any) { |
|
|
|
|
@ -131,14 +185,15 @@ func (m *Env) writeEnvVar(w io.Writer, value string, name string, args ...any) {
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func genEnvInts[T ~int | ~uint](values []T) string { |
|
|
|
|
func genEnvInts[T core.Signed](values []T) string { |
|
|
|
|
var buf bytes.Buffer |
|
|
|
|
|
|
|
|
|
for _, v := range values { |
|
|
|
|
if buf.Len() > 0 { |
|
|
|
|
_, _ = buf.WriteRune(' ') |
|
|
|
|
} |
|
|
|
|
_, _ = buf.WriteString(fmt.Sprintf("%v", v)) |
|
|
|
|
|
|
|
|
|
_, _ = buf.WriteString(fmt.Sprintf("%v", int64(v))) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
return buf.String() |
|
|
|
|
@ -209,3 +264,14 @@ func genEnvZoneCephMonIDs(m Machines) string {
|
|
|
|
|
}) |
|
|
|
|
return buf.String() |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func cmpOrdered[T core.Ordered](a, b T) int { |
|
|
|
|
switch { |
|
|
|
|
case a == b: |
|
|
|
|
return 0 |
|
|
|
|
case a < b: |
|
|
|
|
return -1 |
|
|
|
|
default: |
|
|
|
|
return 1 |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
