fix

main.go

@@ -3,13 +3,15 @@ package main
 import (
 	"flag"
 	"fmt"
+	"net"
 	"net/http"
+	"strings"
 )
 
 func main() {
 	var host string
 	var port int
-	flag.StringVar(&host, "h", "127.0.0.1", "IP地址")
+	flag.StringVar(&host, "h", GetIp(), "IP地址")
 	flag.IntVar(&port, "p", 9000, "端口")
 	flag.Parse()
 	address := fmt.Sprintf("%s:%d", host, port)
@@ -24,3 +26,24 @@ func main() {
 		panic(err)
 	}
 }
+
+func GetIp() string {
+	addrs, err := net.InterfaceAddrs()
+	if err != nil {
+		return "127.0.0.1"
+	}
+	retIp := ""
+	for _, address := range addrs {
+		// 检查ip地址判断是否回环地址
+		if ipnet, ok := address.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
+			if ipnet.IP.To4() != nil {
+				ip := ipnet.IP.String()
+				if strings.HasPrefix(ip, "172.") {
+					retIp = ip
+					break
+				}
+			}
+		}
+	}
+	return retIp
+}