Alejandro Mery
1 year ago
1 changed files with 97 additions and 0 deletions
@ -0,0 +1,97 @@ |
|||||||
|
package zones |
||||||
|
|
||||||
|
import ( |
||||||
|
"bytes" |
||||||
|
"fmt" |
||||||
|
"io" |
||||||
|
"strings" |
||||||
|
) |
||||||
|
|
||||||
|
// WriteEnv generates environment variables for shell scripts
|
||||||
|
func (m *Zones) WriteEnv(w io.Writer) error { |
||||||
|
var buf bytes.Buffer |
||||||
|
|
||||||
|
m.writeEnvVarFn(&buf, genEnvZones, "ZONES") |
||||||
|
m.ForEachZone(func(z *Zone) bool { |
||||||
|
m.writeEnvZone(&buf, z) |
||||||
|
return false |
||||||
|
}) |
||||||
|
|
||||||
|
_, err := buf.WriteTo(w) |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
func (m *Zones) writeEnvZone(w io.Writer, z *Zone) { |
||||||
|
zoneID := z.ID |
||||||
|
|
||||||
|
// ZONE{zoneID}
|
||||||
|
m.writeEnvVar(w, genEnvZoneNodes(z), "ZONE%v", zoneID) |
||||||
|
|
||||||
|
// ZONE{zoneID}_NAME
|
||||||
|
m.writeEnvVar(w, z.Name, "ZONE%v_%s", zoneID, "NAME") |
||||||
|
|
||||||
|
// ZONE{zoneID}_GW
|
||||||
|
firstNodeID := 0 |
||||||
|
gatewayID := 0 |
||||||
|
z.ForEachMachine(func(p *Machine) bool { |
||||||
|
nodeID := p.ID() |
||||||
|
|
||||||
|
if firstNodeID == 0 { |
||||||
|
firstNodeID = nodeID |
||||||
|
} |
||||||
|
|
||||||
|
if _, found := p.getRingInfo(0); found { |
||||||
|
gatewayID = nodeID |
||||||
|
} |
||||||
|
|
||||||
|
return gatewayID != 0 |
||||||
|
}) |
||||||
|
|
||||||
|
if gatewayID == 0 { |
||||||
|
gatewayID = firstNodeID |
||||||
|
} |
||||||
|
m.writeEnvVar(w, fmt.Sprintf("%v", gatewayID), "ZONE%v_%s", zoneID, "GW") |
||||||
|
|
||||||
|
// ZONE{zoneID}_IP
|
||||||
|
ip, _ := RingZeroAddress(zoneID, gatewayID) |
||||||
|
m.writeEnvVar(w, ip.String(), "ZONE%v_%s", zoneID, "IP") |
||||||
|
} |
||||||
|
|
||||||
|
func (m *Zones) writeEnvVarFn(w io.Writer, fn func(*Zones) string, name string, args ...any) { |
||||||
|
var value string |
||||||
|
|
||||||
|
if fn != nil { |
||||||
|
value = fn(m) |
||||||
|
} |
||||||
|
|
||||||
|
m.writeEnvVar(w, value, name, args...) |
||||||
|
} |
||||||
|
|
||||||
|
func (*Zones) writeEnvVar(w io.Writer, value string, name string, args ...any) { |
||||||
|
if len(args) > 0 { |
||||||
|
name = fmt.Sprintf(name, args...) |
||||||
|
} |
||||||
|
|
||||||
|
if name != "" { |
||||||
|
value = strings.TrimSpace(value) |
||||||
|
|
||||||
|
_, _ = fmt.Fprintf(w, "%s=%q\n", name, value) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func genEnvZones(m *Zones) string { |
||||||
|
s := make([]string, 0, len(m.Zones)) |
||||||
|
for _, z := range m.Zones { |
||||||
|
s = append(s, fmt.Sprintf("%v", z.ID)) |
||||||
|
} |
||||||
|
|
||||||
|
return strings.Join(s, " ") |
||||||
|
} |
||||||
|
|
||||||
|
func genEnvZoneNodes(z *Zone) string { |
||||||
|
s := make([]string, 0, len(z.Machines)) |
||||||
|
for _, p := range z.Machines { |
||||||
|
s = append(s, p.Name) |
||||||
|
} |
||||||
|
return strings.Join(s, " ") |
||||||
|
} |
Loading…
Reference in new issue