From f5765e4dde5fb2bb2e20bacd88d1784928fc37d4 Mon Sep 17 00:00:00 2001
From: Brian Waldon <bcwaldon@gmail.com>
Date: Tue, 18 Mar 2014 09:36:31 -0700
Subject: [PATCH] feat(etcd): Default etcd name to /etc/machine-id

---
 initialize/etcd.go      |  7 +++++++
 initialize/etcd_test.go | 44 +++++++++++++++++++++++++++++++++++++++++
 system/systemd.go       |  6 ++++++
 system/systemd_test.go  | 14 +++++++++++++
 test.yaml               |  1 -
 5 files changed, 71 insertions(+), 1 deletion(-)

diff --git a/initialize/etcd.go b/initialize/etcd.go
index a55be07..ad93cb6 100644
--- a/initialize/etcd.go
+++ b/initialize/etcd.go
@@ -31,11 +31,18 @@ func (ec EtcdEnvironment) String() (out string) {
 
 		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 name := system.MachineID(root); name != "" {
+			env["name"] = name
+		}
+	}
+
 	file := system.File{
 		Path: path.Join(root, "run", "systemd", "system", "etcd.service.d", "20-cloudinit.conf"),
 		RawFilePermissions: "0644",
diff --git a/initialize/etcd_test.go b/initialize/etcd_test.go
index f1c3e94..56cd88c 100644
--- a/initialize/etcd_test.go
+++ b/initialize/etcd_test.go
@@ -46,6 +46,7 @@ Environment="ETCD_PEER_BIND_ADDR=192.0.2.13:7001"
 
 func TestEtcdEnvironmentWrittenToDisk(t *testing.T) {
 	ec := EtcdEnvironment{
+		"name": "node001",
 		"discovery_url": "http://disco.example.com/foobar",
 		"peer-bind-addr": "127.0.0.1:7002",
 	}
@@ -76,6 +77,7 @@ func TestEtcdEnvironmentWrittenToDisk(t *testing.T) {
 	}
 
 	expect := `[Service]
+Environment="ETCD_NAME=node001"
 Environment="ETCD_DISCOVERY_URL=http://disco.example.com/foobar"
 Environment="ETCD_PEER_BIND_ADDR=127.0.0.1:7002"
 `
@@ -84,6 +86,48 @@ Environment="ETCD_PEER_BIND_ADDR=127.0.0.1:7002"
 	}
 }
 
+func TestEtcdEnvironmentWrittenToDiskDefaults(t *testing.T) {
+	ec := EtcdEnvironment{}
+	dir, err := ioutil.TempDir(os.TempDir(), "coreos-cloudinit-")
+	if err != nil {
+		t.Fatalf("Unable to create tempdir: %v", err)
+	}
+	defer syscall.Rmdir(dir)
+
+	os.Mkdir(path.Join(dir, "etc"), os.FileMode(0755))
+	err = ioutil.WriteFile(path.Join(dir, "etc", "machine-id"), []byte("node007"), os.FileMode(0444))
+	if err != nil {
+		t.Fatalf("Failed writing out /etc/machine-id: %v", err)
+	}
+
+	if err := WriteEtcdEnvironment(ec, dir); err != nil {
+		t.Fatalf("Processing of EtcdEnvironment failed: %v", err)
+	}
+
+	fullPath := path.Join(dir, "etc", "systemd", "system", "etcd.service.d", "20-cloudinit.conf")
+
+	fi, err := os.Stat(fullPath)
+	if err != nil {
+		t.Fatalf("Unable to stat file: %v", err)
+	}
+
+	if fi.Mode() != os.FileMode(0644) {
+		t.Errorf("File has incorrect mode: %v", fi.Mode())
+	}
+
+	contents, err := ioutil.ReadFile(fullPath)
+	if err != nil {
+		t.Fatalf("Unable to read expected file: %v", err)
+	}
+
+	expect := `[Service]
+Environment="ETCD_NAME=node007"
+`
+	if string(contents) != expect {
+		t.Fatalf("File has incorrect contents")
+	}
+}
+
 func rmdir(path string) error {
     cmd := exec.Command("rm", "-rf", path)
     return cmd.Run()
diff --git a/system/systemd.go b/system/systemd.go
index 5861238..5b77f9f 100644
--- a/system/systemd.go
+++ b/system/systemd.go
@@ -2,6 +2,7 @@ package system
 
 import (
 	"fmt"
+	"io/ioutil"
 	"log"
 	"os"
 	"os/exec"
@@ -161,3 +162,8 @@ func ExecuteScript(scriptPath string) (string, error) {
 func SetHostname(hostname string) error {
 	return exec.Command("hostnamectl", "set-hostname", hostname).Run()
 }
+
+func MachineID(root string) string {
+	contents, _ := ioutil.ReadFile(path.Join(root, "etc", "machine-id"))
+	return strings.TrimSpace(string(contents))
+}
diff --git a/system/systemd_test.go b/system/systemd_test.go
index 02924b0..6635d0e 100644
--- a/system/systemd_test.go
+++ b/system/systemd_test.go
@@ -100,3 +100,17 @@ Where=/media/state
 	}
 }
 
+func TestMachineID(t *testing.T) {
+	dir, err := ioutil.TempDir(os.TempDir(), "coreos-cloudinit-")
+	if err != nil {
+		t.Fatalf("Unable to create tempdir: %v", err)
+	}
+	defer syscall.Rmdir(dir)
+
+	os.Mkdir(path.Join(dir, "etc"), os.FileMode(0755))
+	ioutil.WriteFile(path.Join(dir, "etc", "machine-id"), []byte("node007\n"), os.FileMode(0444))
+
+	if MachineID(dir) != "node007" {
+		t.Fatalf("File has incorrect contents")
+	}
+}
diff --git a/test.yaml b/test.yaml
index 8941ecd..cbf0170 100644
--- a/test.yaml
+++ b/test.yaml
@@ -4,4 +4,3 @@ coreos:
     etcd:
         discovery_url: https://discovery.etcd.io/0022cb5027f8f5167a874794c3a13e0d
         bind-addr: $public_ipv4:4001
-        name: polvi