improve: use IndexByte to find the colon of ip address (#92)

cmd/agent/main.go

@@ -575,7 +575,7 @@ func handleTcpPingTask(task *pb.Task, result *pb.TaskResult) {
 		result.Data = err.Error()
 		return
 	}
-	if strings.Contains(ipAddr, ":") {
+	if strings.IndexByte(ipAddr, ':') != -1 {
 		ipAddr = fmt.Sprintf("[%s]", ipAddr)
 	}
 	printf("TCP-Ping Task: Pinging %s:%s", ipAddr, port)

pkg/monitor/myip.go

@@ -91,11 +91,11 @@ func fetchIP(servers []string, isV6 bool) string {
 				}
 			}
 			// 没取到 v6 IP
-			if isV6 && !strings.Contains(newIP, ":") {
+			if isV6 && strings.IndexByte(newIP, ':') == -1 {
 				continue
 			}
 			// 没取到 v4 IP
-			if !isV6 && !strings.Contains(newIP, ".") {
+			if !isV6 && strings.IndexByte(newIP, '.') == -1 {
 				continue
 			}
 			ip = newIP

pkg/util/http.go

@@ -72,12 +72,12 @@ func resolveIP(addr string, ipv6 bool) (string, error) {
 
 	for i := 0; i < len(res); i++ {
 		ip := res[i].String()
-		if strings.Contains(ip, ".") && !ipv6 {
+		if strings.IndexByte(ip, '.') != -1 && !ipv6 {
 			ipv4Resolved = true
 			url[0] = ip
 			break
 		}
-		if strings.Contains(ip, ":") && ipv6 {
+		if strings.IndexByte(ip, ':') != -1 && ipv6 {
 			ipv6Resolved = true
 			url[0] = "[" + ip + "]"
 			break