From c069f32bbe2f45a2b4a80e17b323f0cc2ab21e76 Mon Sep 17 00:00:00 2001
From: Darren Shepherd <darren@rancher.com>
Date: Fri, 21 Feb 2020 13:17:42 -0700
Subject: [PATCH] Fix x-api-host when port is set

---
 pkg/schemaserver/urlbuilder/base.go | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/pkg/schemaserver/urlbuilder/base.go b/pkg/schemaserver/urlbuilder/base.go
index 23872d0..48c12d7 100644
--- a/pkg/schemaserver/urlbuilder/base.go
+++ b/pkg/schemaserver/urlbuilder/base.go
@@ -17,9 +17,11 @@ func ParseRequestURL(r *http.Request) string {
 
 func GetHost(r *http.Request, scheme string) string {
 	host := r.Header.Get(ForwardedAPIHostHeader)
-	if host == "" {
-		host = strings.Split(r.Header.Get(ForwardedHostHeader), ",")[0]
+	if host != "" {
+		return host
 	}
+
+	host = strings.Split(r.Header.Get(ForwardedHostHeader), ",")[0]
 	if host == "" {
 		host = r.Host
 	}
@@ -39,7 +41,7 @@ func GetHost(r *http.Request, scheme string) string {
 
 	hostname, _, err := net.SplitHostPort(host)
 	if err != nil {
-		return host
+		hostname = host
 	}
 
 	return strings.Join([]string{hostname, port}, ":")