添加remote ip

Author: yumaojun03

http/request/pagging_test.go

@@ -38,5 +38,4 @@ func TestOffSet(t *testing.T) {
 	pr = request.NewPageRequestFromHTTP(req)
 	shoud.Equal(int64(18), pr.Offset())
 	shoud.Equal(uint(20), pr.PageSize)
-
 }

http/request/remoteip.go

@@ -0,0 +1,47 @@
+package request
+
+import (
+	"net/http"
+	"strings"
+)
+
+var (
+	// DefaultScanForwareHeaderKey 协商forward ip 的 hander key名称
+	DefaultScanForwareHeaderKey = []string{"X-Forwarded-For", "X-Real-IP"}
+)
+
+// GetRemoteIP todo
+func GetRemoteIP(r *http.Request) string {
+	// 优先获取代理IP
+	var ip string
+	for _, key := range DefaultScanForwareHeaderKey {
+		value := r.Header.Get(key)
+
+		if strings.Contains(value, ", ") {
+			i := strings.Index(value, ", ")
+			if i == -1 {
+				i = len(value)
+			}
+
+			ip = value[:i]
+			break
+		}
+
+		if value != "" {
+			ip = value
+			break
+		}
+	}
+
+	if ip != "" {
+		return ip
+	}
+
+	// 如果没有获得代理IP则采用RemoteIP
+	addr := strings.Split(r.RemoteAddr, ":")
+	if len(addr) == 1 {
+		return addr[0]
+	}
+
+	return strings.Join(addr[0:len(addr)-1], ":")
+}

http/request/remoteip_test.go

@@ -0,0 +1,27 @@
+package request_test
+
+import (
+	"net/http"
+	"testing"
+
+	"github.com/infraboard/mcube/http/request"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetRemoteIpFromHeader(t *testing.T) {
+	shoud := assert.New(t)
+
+	req, _ := http.NewRequest("GET", "/", nil)
+	req.Header.Set("X-Forwarded-For", "10.10.10.10")
+	ip := request.GetRemoteIP(req)
+	shoud.Equal("10.10.10.10", ip)
+}
+
+func TestGetRemoteIpFromConn(t *testing.T) {
+	shoud := assert.New(t)
+
+	req, _ := http.NewRequest("GET", "/", nil)
+	req.RemoteAddr = "10.10.10.10"
+	ip := request.GetRemoteIP(req)
+	shoud.Equal("10.10.10.10", ip)
+}