feature: add node ip and sn into daemon labels

daemon/daemon.go

@@ -16,6 +16,7 @@ import (
 	"github.com/alibaba/pouch/internal"
 	"github.com/alibaba/pouch/network/mode"
 	"github.com/alibaba/pouch/pkg/meta"
+	"github.com/alibaba/pouch/pkg/system"
 
 	"github.com/gorilla/mux"
 	"github.com/pkg/errors"
@@ -167,6 +168,10 @@ func (d *Daemon) Run() error {
 	}
 	d.containerMgr = containerMgr
 
+	if err := d.addSystemLabels(); err != nil {
+		return err
+	}
+
 	d.server = server.Server{
 		Config:          d.config,
 		ContainerMgr:    containerMgr,
@@ -269,6 +274,27 @@ func (d *Daemon) ShutdownPlugin() error {
 	return nil
 }
 
+// addSystemLabels adds some system labels to daemon's config.
+// Currently, pouchd add node ip and serial number to pouchd with the format:
+// node_ip=192.168.0.1
+// SN=xxxxx
+func (d *Daemon) addSystemLabels() error {
+	d.config.Lock()
+	defer d.config.Unlock()
+	if d.config.Labels == nil {
+		d.config.Labels = make([]string, 0)
+	}
+	// get node IP
+	nodeIP := system.GetNodeIP()
+	d.config.Labels = append(d.config.Labels, fmt.Sprintf("node_ip=%s", nodeIP))
+
+	// get serial number
+	serialNo := system.GetSerialNumber()
+	d.config.Labels = append(d.config.Labels, fmt.Sprintf("SN=%s", serialNo))
+
+	return nil
+}
+
 // RunCriService start cri service if pouchd is specified with --enable-cri.
 func (d *Daemon) RunCriService(stopCh chan error) {
 	var err error

pkg/system/machine.go

@@ -0,0 +1,36 @@
+package system
+
+import (
+	"bufio"
+	"bytes"
+	"os"
+	"os/exec"
+	"strings"
+	"time"
+)
+
+// GetSerialNumber gets serial number or a machine.
+func GetSerialNumber() string {
+	var sn string
+	if b, e := exec.Command("dmidecode", "-s", "system-serial-number").CombinedOutput(); e == nil {
+		scanner := bufio.NewScanner(bytes.NewReader(b))
+		for scanner.Scan() {
+			sn = scanner.Text()
+		}
+	}
+	if len(strings.Fields(sn)) != 0 {
+		sn = strings.Fields(sn)[0]
+	}
+	for i := 0; i < 10; i++ {
+		if _, ex := os.Stat("/usr/alisys/dragoon/libexec/armory/bin/armoryinfo"); ex == nil {
+			if b, e := exec.Command("/usr/alisys/dragoon/libexec/armory/bin/armoryinfo", "sn").CombinedOutput(); e == nil {
+				sn = strings.TrimSpace(string(b))
+			}
+		}
+		if sn != "" {
+			break
+		}
+		time.Sleep(100 * time.Millisecond)
+	}
+	return sn
+}

pkg/system/network.go

@@ -0,0 +1,26 @@
+package system
+
+import (
+	"bufio"
+	"bytes"
+	"net"
+	"os/exec"
+)
+
+// GetNodeIP fetches node ip via command hostname.
+// If it fails to get this, return empty string directly.
+func GetNodeIP() string {
+	output, err := exec.Command("hostname", "-i").CombinedOutput()
+	if err != nil {
+		return ""
+	}
+
+	scanner := bufio.NewScanner(bytes.NewReader(output))
+	for scanner.Scan() {
+		ip := scanner.Text()
+		if net.ParseIP(ip) != nil {
+			return ip
+		}
+	}
+	return ""
+}

test/api_daemon_update_test.go

@@ -43,6 +43,18 @@ func (suite *APIDaemonUpdateSuite) TestUpdateDaemon(c *check.C) {
 	err = request.DecodeBody(&info, resp.Body)
 	c.Assert(err, check.IsNil)
 
-	c.Assert(info.Labels, check.DeepEquals, labels)
+	for _, label := range labels {
+		isContained := false
+		for _, infoLabel := range info.Labels {
+			if infoLabel == label {
+				isContained = true
+				break
+			}
+		}
+		if !isContained {
+			c.Fatalf("label %s should be in labels in info API", label)
+		}
+	}
+	//c.Assert(info.Labels, check.DeepEquals, labels)
 	// TODO: add checking image proxy
 }