Merge pull request #348 from ibuildthecloud/clientconfig

Fix x-api-host when setting ports
rancher · Feb 21, 2020 · 9bd5ca9 · 9bd5ca9
2 parents bf773d0 + b16f316
commit 9bd5ca9
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/urlbuilder/url.go b/urlbuilder/url.go
@@ -46,9 +46,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
 	}
@@ -68,7 +70,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}, ":")