package cluster import ( "bytes" "fmt" "io" "strings" "darvaza.org/core" "git.jpi.io/amery/jpictl/pkg/rings" ) // Env is a shell environment factory for this cluster type Env struct { ZoneIterator RegionIterator cephFSID string export bool } // Env returns a shell environment factory func (m *Cluster) Env(export bool) (*Env, error) { fsid, err := m.GetCephFSID() if err != nil { return nil, err } env := &Env{ ZoneIterator: m, RegionIterator: m, cephFSID: fsid.String(), export: export, } return env, nil } // 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 iter := core.IIf[ZoneIterator](r != nil, r, m) iter.ForEachZone(func(z *Zone) bool { zones = append(zones, z.ID) return false }) core.SliceSortOrdered(zones) return zones } // RegionsNames returns a sorted list of primary regions names func (m *Env) RegionsNames() []string { var regions []string m.ForEachRegion(func(r *Region) bool { if r.IsPrimary() { regions = append(regions, r.Name) } return false }) core.SliceSortOrdered(regions) 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.SliceSortOrdered(regions) return regions } // WriteTo generates environment variables for shell scripts func (m *Env) WriteTo(w io.Writer) (int64, error) { var buf bytes.Buffer if m.cephFSID != "" { m.writeEnvVar(&buf, m.cephFSID, "FSID") } 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") 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 }) core.SliceSortFn(out, func(a, b *Region) bool { return 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, r *Region, z *Zone) { zonePrefix := fmt.Sprintf("REGION%v_ZONE%v", r.ID, z.ID) monPrefix := zonePrefix + "_MON" // REGION{regionID}_ZONE{zoneID} m.writeEnvVar(w, genEnvZoneNodes(z), zonePrefix) // REGION{regionID}_ZONE{zoneID}_NAME m.writeEnvVar(w, z.Name, zonePrefix+"_NAME") // REGION{regionID}_ZONE{zoneID}_GW gateways, _ := z.GatewayIDs() m.writeEnvVar(w, genEnvInts(gateways), zonePrefix+"_GW") // Ceph monitors := z.GetCephMonitors() // 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) { var prefix string if m.export { prefix = "export " } if len(args) > 0 { name = fmt.Sprintf(name, args...) } if name != "" { value = strings.TrimSpace(value) if value == "" { _, _ = fmt.Fprintf(w, "%s%s=\n", prefix, name) } else { _, _ = fmt.Fprintf(w, "%s%s=%q\n", prefix, name, value) } } } 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", int64(v))) } return buf.String() } func genEnvStrings(values []string) string { return strings.Join(values, " ") } func genEnvZoneNodes(z *Zone) string { if n := z.Len(); n > 0 { s := make([]string, 0, n) z.ForEachMachine(func(p *Machine) bool { s = append(s, p.Name) return false }) return genEnvStrings(s) } return "" } func genEnvZoneCephMonNames(m Machines) string { var buf strings.Builder m.ForEachMachine(func(p *Machine) bool { if buf.Len() > 0 { _, _ = buf.WriteRune(' ') } _, _ = buf.WriteString(p.Name) return false }) return buf.String() } func genEnvZoneCephMonIPs(m Machines) string { var buf strings.Builder m.ForEachMachine(func(p *Machine) bool { addr := p.RingOneAddress() if buf.Len() > 0 { _, _ = buf.WriteRune(' ') } _, _ = buf.WriteString(addr.String()) return false }) return buf.String() } func genEnvZoneCephMonIDs(m Machines) string { var buf strings.Builder m.ForEachMachine(func(p *Machine) bool { if buf.Len() > 0 { _, _ = buf.WriteRune(' ') } _, _ = fmt.Fprintf(&buf, "%v", p.ID) return false }) return buf.String() }