63 lines
1.4 KiB
Go
63 lines
1.4 KiB
Go
package initialize
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/coreos/coreos-cloudinit/system"
|
|
)
|
|
|
|
type EtcdEnvironment map[string]string
|
|
|
|
func (ec EtcdEnvironment) normalized() map[string]string {
|
|
out := make(map[string]string, len(ec))
|
|
for key, val := range ec {
|
|
key = strings.ToUpper(key)
|
|
key = strings.Replace(key, "-", "_", -1)
|
|
out[key] = val
|
|
}
|
|
return out
|
|
}
|
|
|
|
func (ec EtcdEnvironment) String() (out string) {
|
|
norm := ec.normalized()
|
|
|
|
if val, ok := norm["DISCOVERY_URL"]; ok {
|
|
delete(norm, "DISCOVERY_URL")
|
|
if _, ok := norm["DISCOVERY"]; !ok {
|
|
norm["DISCOVERY"] = val
|
|
}
|
|
}
|
|
|
|
out += "[Service]\n"
|
|
|
|
for key, val := range norm {
|
|
out += fmt.Sprintf("Environment=\"ETCD_%s=%s\"\n", key, val)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Write an EtcdEnvironment to the appropriate path on disk for etcd.service
|
|
func WriteEtcdEnvironment(env EtcdEnvironment, root string) error {
|
|
if _, ok := env["name"]; !ok {
|
|
if machineID := system.MachineID(root); machineID != "" {
|
|
env["name"] = machineID
|
|
} else if hostname, err := system.Hostname(); err == nil {
|
|
env["name"] = hostname
|
|
} else {
|
|
return errors.New("Unable to determine default etcd name")
|
|
}
|
|
}
|
|
|
|
file := system.File{
|
|
Path: path.Join(root, "run", "systemd", "system", "etcd.service.d", "20-cloudinit.conf"),
|
|
RawFilePermissions: "0644",
|
|
Content: env.String(),
|
|
}
|
|
|
|
return system.WriteFile(&file)
|
|
}
|