From e1611ccf4796e4653cb8f992d02ca0eacfeb1449 Mon Sep 17 00:00:00 2001
From: guzeng <gz@flyrise.cn>
Date: Sun, 7 Feb 2021 11:45:33 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=BC=BA=E5=A4=B1=E6=96=B9?=
 =?UTF-8?q?=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 site.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/site.go b/site.go
index 7b5bf47..aca2d02 100644
--- a/site.go
+++ b/site.go
@@ -99,3 +99,28 @@ func GetSiteInfoFromReq(master string, req *http.Request) (map[string]string, er
 	}
 	return siteInfo, nil
 }
+
+/**
+ * 取访问主机名
+ */
+func GetHost(req *http.Request) string {
+
+	var hostlist []string = req.Header.Values("X-Forwarded-Host")
+	var host string
+	if len(hostlist) > 1 {
+		host = hostlist[len(hostlist)-1] //取最后一次转发的
+	} else if len(hostlist) == 1 {
+		host = hostlist[0]
+	} else {
+		host = ""
+	}
+	if host != "" {
+		hosts := strings.Split(host, ",")
+		host = strings.Trim(hosts[len(hosts)-1], " ")
+	}
+
+	if host == "" {
+		host = req.Host
+	}
+	return host
+}