k3s/pkg/agent/loadbalancer/utility.go

50 lines
975 B
Go
Raw Normal View History

2019-07-24 07:22:31 +00:00
package loadbalancer
import (
"errors"
"net/url"
"sort"
"strings"
)
func parseURL(serverURL, newHost string) (string, string, error) {
parsedURL, err := url.Parse(serverURL)
if err != nil {
return "", "", err
}
if parsedURL.Host == "" {
return "", "", errors.New("Initial server URL host is not defined for load balancer")
}
address := parsedURL.Host
if parsedURL.Port() == "" {
if strings.ToLower(parsedURL.Scheme) == "http" {
address += ":80"
}
if strings.ToLower(parsedURL.Scheme) == "https" {
address += ":443"
}
}
parsedURL.Host = newHost
return address, parsedURL.String(), nil
}
func sortServers(input []string, search string) ([]string, bool) {
result := []string{}
found := false
skip := map[string]bool{"": true}
for _, entry := range input {
if skip[entry] {
continue
}
if search == entry {
found = true
}
skip[entry] = true
result = append(result, entry)
}
sort.Strings(result)
return result, found
}