From d31571fc5075372d6f60cb2df1efe8b59da05fd5 Mon Sep 17 00:00:00 2001 From: Bastian Doetsch Date: Sun, 6 Mar 2022 19:20:21 +0100 Subject: [PATCH] cleanup (country default now comes from flag) --- main.go | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) diff --git a/main.go b/main.go index ce1a179..8bf274a 100644 --- a/main.go +++ b/main.go @@ -24,7 +24,7 @@ func main() { flag.Parse() servers := getServers(*typeFlag) - bestIndex := selectBestServerIndex(servers, countryFlag) + bestIndex := selectBestServerIndex(servers, *countryFlag) best := servers[bestIndex] log.Debug().Interface("server", best).Msg("Best latency server found.") hostname := strings.TrimSuffix(best.Hostname, "-wireguard") @@ -39,19 +39,11 @@ func main() { } } -func selectBestServerIndex(servers []server, country *string) int { +func selectBestServerIndex(servers []server, country string) int { best := servers[0].Hostname bestIndex := -1 - allowedCountries := map[string]string{} - if *country == "" { - allowedCountries["de"] = "1" - allowedCountries["ch"] = "1" - allowedCountries["at"] = "1" - } else { - allowedCountries[*country] = "1" - } for i, server := range servers { - if server.Active && allowedCountries[server.CountryCode] != "" { + if server.Active && server.CountryCode == country { duration, err := serverLatency(server) if err == nil { pings[server.Hostname] = duration