improved host lookup

pull/194/head
t4skforce 2019-05-27 19:35:51 +02:00
parent 1950439b78
commit a187d23e3b
1 changed files with 2 additions and 8 deletions

View File

@ -90,17 +90,11 @@ func (s *Service) parseHost() string {
if s.Type == "tcp" || s.Type == "udp" {
return s.Domain
} else {
domain := s.Domain
hasPort, _ := regexp.MatchString(`\:([0-9]+)`, domain)
if hasPort {
splitDomain := strings.Split(s.Domain, ":")
domain = splitDomain[len(splitDomain)-2]
}
host, err := url.Parse(domain)
u, err := url.Parse(s.Domain)
if err != nil {
return s.Domain
}
return host.Host
return strings.Split(u.Host, ":")[0]
}
}