⚡️ refactor: better parse address

gofiber · Sep 6, 2022 · 871faf5 · 871faf5 · ReneWerner87 · Sep 6, 2022
1 parent 9b6d62d
commit 871faf5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ctx.go b/ctx.go
@@ -618,12 +618,13 @@ func (c *Ctx) Host() string {
 func (c *Ctx) Hostname() string {
 	if c.IsProxyTrusted() {
 		if host := c.Get(HeaderXForwardedHost); len(host) > 0 {
-			return strings.Split(host, ":")[0]
+			addr, _ := parseAddr(c.Host())
+			return addr
 		}
 	}
 
-	hn := utils.UnsafeString(c.fasthttp.Host())
-	return strings.Split(hn, ":")[0]
+	addr, _ := parseAddr(c.Host())
+	return addr
 }
 
 // Port returns the remote port of the request.