diff --git a/server.go b/server.go index 7c5522a..e40b0d1 100644 --- a/server.go +++ b/server.go @@ -145,11 +145,10 @@ func (s *Server) handleConnection(ctx context.Context, conn net.Conn) { fields[log.FnError] = err.Error() _ = s.logger.Warn("peekHTTP failed", fields) } else if host != "" { - if !strings.Contains(host, ":") { - addr = host + addr[strings.Index(addr, ":"):] - } else { - addr = host + if strings.Contains(host, ":") { + host = host[:strings.Index(host, ":")] } + addr = host + addr[strings.Index(addr, ":"):] } reader = reader_n3 }