support linux dns resolve, short domain name resolve

dns/dns_linux.go

@@ -0,0 +1,30 @@
+// +build linux
+
+package dns
+
+import (
+	log "github.com/sirupsen/logrus"
+	"os"
+	"os/exec"
+)
+
+func DNS(ip string, namespace string) error {
+	tunName := os.Getenv("tunName")
+	if len(tunName) == 0 {
+		tunName = "tun0"
+	}
+	cmd := exec.Command("systemd-resolve", []string{
+		"--set-dns",
+		ip,
+		"--interface",
+		tunName,
+		"--set-domain=" + namespace + ".svc.cluster.local",
+		"--set-domain=svc.cluster.local",
+	}...)
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		log.Warnf("cmd: %s, output: %s, error: %v\n", cmd.Args, string(output), err)
+	}
+
+	return nil
+}

dns/dns_unix.go

@@ -1,4 +1,4 @@
-// +build !windows
+// +build darwin
 
 package dns
 

pkg/main.go

@@ -170,11 +170,7 @@ func main() {
 	if err := start(); err != nil {
 		log.Fatal(err)
 	}
-	//time.Sleep(time.Second * 5)
+
-	dnsServiceIp := util.GetDNSServiceIpFromPod(clientset, restclient, config, util.TrafficManager, namespace)
-	if err := dns.DNS(dnsServiceIp, namespace); err != nil {
-		log.Fatal(err)
-	}
 	if runtime.GOOS == "windows" {
 		if !util.FindRule() {
 			util.AddFirewallRule()
@@ -183,6 +179,12 @@ func main() {
 	}
 	log.Info("dns service ok")
 	_ = exec.Command("ping", "-c", "4", "223.254.254.100").Run()
+
+	//time.Sleep(time.Second * 5)
+	dnsServiceIp := util.GetDNSServiceIpFromPod(clientset, restclient, config, util.TrafficManager, namespace)
+	if err := dns.DNS(dnsServiceIp, namespace); err != nil {
+		log.Fatal(err)
+	}
 	select {}
 }