Merge pull request #21 from TRON-US/add-ip-util

add local ip util

network/ips.go

@@ -0,0 +1,42 @@
+package network
+
+import "net"
+
+func GetLocalIps() ([]string, error) {
+	ips := make([]string, 0)
+	ifaces, err := net.Interfaces()
+	if err != nil {
+		return ips, err
+	}
+	for _, i := range ifaces {
+		addrs, err := i.Addrs()
+		if err != nil {
+			return ips, err
+		}
+		for _, addr := range addrs {
+			var ip net.IP
+			switch v := addr.(type) {
+			case *net.IPNet:
+				ip = v.IP
+			case *net.IPAddr:
+				ip = v.IP
+			}
+			ips = append(ips, ip.String())
+		}
+	}
+	return ips, nil
+}
+
+func IsLocalIp(ip string) (bool, error) {
+	isLocal := false
+	ips, err := GetLocalIps()
+	if err != nil {
+		return isLocal, err
+	}
+	for _, item := range ips {
+		if item == ip {
+			return true, nil
+		}
+	}
+	return isLocal, nil
+}

network/ips_test.go

@@ -0,0 +1,14 @@
+package network
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestIsLocalIp(t *testing.T) {
+	isLocal, err := IsLocalIp("127.0.0.1")
+	if err != nil {
+		t.Fatal(err)
+	}
+	assert.True(t, isLocal)
+}