v1.2.3 新增 测速后 直接显示结果 功能

README.md

@@ -18,7 +18,7 @@
 1. 下载编译好的可执行文件 [蓝奏云](https://www.lanzoux.com/b0742hkxe) / [Github](https://github.com/XIU2/CloudflareSpeedTest/releases) 并解压。  
 2. 双击运行 `CloudflareST.exe`文件（Windows系统），等待测速...  
 
-测速完毕后，会把结果保存在当前目录下的 `result.csv` 文件中（只输出丢包率 50% 以下的），用记事本打开，排序为**延迟由低到高**，每一列用逗号分隔，分别是：  
+测速完毕后，会把结果保存在当前目录下的 `result.csv` 文件中，用记事本打开，排序为**延迟由低到高**，每一列用逗号分隔，分别是：  
 ```
 IP 地址, 测试次数, 成功次数, 成功比率, 平均延迟, 下载速度 (MB/s)
 104.27.70.18, 4, 4, 1.00, 150.79, 12.89
@@ -38,15 +38,19 @@ https://github.com/XIU2/CloudflareSpeedTest
 
 参数：
     -n 500
-        测速线程数量；数值越大速度越快，请勿超过1000(结果误差大)；(默认 500)
+        测速线程数量；数值越大速度越快，请勿超过 1000(结果误差大)；(默认 500)
     -t 4
         延迟测速次数；单个 IP 测速次数，为 1 时将过滤丢包的IP，TCP协议；(默认 4)
     -dn 20
-        下载测速数量；延迟测速并排序后，从最低延迟起测试下载速度的数量，请勿太多(速度慢)；(默认 20)
+        下载测速数量；延迟测速并排序后，从最低延迟起下载测速数量，请勿太多(速度慢)；(默认 20)
     -dt 10
         下载测速时间；单个 IP 测速最长时间，单位：秒；(默认 10)
+    -p 20
+        直接显示结果；测速后直接显示指定数量的结果，为 -1 时不显示结果直接退出；(默认 20)
     -f ip.txt
-        IP 数据文件；支持相对路径和绝对路径，如果包含空格请前后加上引号；(默认 ip.txt)
+        IP 数据文件；相对/绝对路径，如包含空格请加上引号；支持其他 CDN IP段，记得禁用下载测速；(默认 ip.txt)
+    -o result.csv
+        输出结果文件；相对/绝对路径，如包含空格请加上引号；允许 .txt 等后缀；(默认 result.csv)
     -dd
         禁用下载测速；如果带上该参数就是禁用下载测速；(默认 启用)
     -v
@@ -55,24 +59,31 @@ https://github.com/XIU2/CloudflareSpeedTest
         打印帮助说明
 
 示例：
-    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -p 20
-    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -f "C:\abc\ip.txt" -dd
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -p -1 -f "ip.txt" -o "result.csv" -dd
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -p -1 -f "C:\abc\ip.txt" -o "C:\abc\result.csv" -dd
 ```
 
 #### 使用示例
 
 在 CMD 中运行，或者把启动参数添加到快捷方式中。  
-> **注意：** 不需要通顺加上所有参数，按需选择，参数前后顺序随意。  
+> **注意：** 不需要加上所有参数，按需选择，参数前后顺序随意。  
 
 ``` cmd
 # CMD 示例
-CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10
+CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -p 20
+
+# 指定 IP数据文件 及 输出结果文件（相对路径，即当前目录下）
+CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -p -1 -f "ip.txt" -o "result.csv" -dd
+
+# 指定 IP数据文件 及 输出结果文件（绝对路径，即 C:\abc\ 目录下）
+CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -p -1 -f "C:\abc\ip.txt" -o "C:\abc\result.csv" -dd
 ```
 
 ``` cmd
 # 快捷方式示例（右键快捷方式 - 目标）
 ## 如果有引号就放在引号外面，记得引号和 - 之间有空格。
-"D:\Program Files\CloudflareST\CloudflareST.exe" -n 500 -t 4 -dn 20 -dt 10
+"D:\Program Files\CloudflareST\CloudflareST.exe" -n 500 -t 4 -dn 20 -dt 10 -p 20
 ```
 
 ****

go.mod

@@ -1,4 +1,4 @@
-module CloudflareIPScanner
+module CloudflareSpeedTest
 
 go 1.14
 

main.go

@@ -14,6 +14,8 @@ import (
 var version string
 var disableDownload bool
 var ipFile string
+var outputFile string
+var printResult int
 
 func init() {
 	var downloadSecond int64
@@ -25,15 +27,19 @@ https://github.com/XIU2/CloudflareSpeedTest
 
 参数：
     -n 500
-        测速线程数量；数值越大速度越快，请勿超过1000(结果误差大)；(默认 500)
+        测速线程数量；数值越大速度越快，请勿超过 1000(结果误差大)；(默认 500)
     -t 4
         延迟测速次数；单个 IP 测速次数，为 1 时将过滤丢包的IP，TCP协议；(默认 4)
     -dn 20
-        下载测速数量；延迟测速并排序后，从最低延迟起测试下载速度的数量，请勿太多(速度慢)；(默认 20)
+        下载测速数量；延迟测速并排序后，从最低延迟起下载测速数量，请勿太多(速度慢)；(默认 20)
     -dt 10
         下载测速时间；单个 IP 测速最长时间，单位：秒；(默认 10)
+    -p 20
+        直接显示结果；测速后直接显示指定数量的结果，为 -1 时不显示结果直接退出；(默认 20)
     -f ip.txt
-        IP 数据文件；支持相对路径和绝对路径，如果包含空格请前后加上引号；(默认 ip.txt)
+        IP 数据文件；相对/绝对路径，如包含空格请加上引号；支持其他 CDN IP段，记得禁用下载测速；(默认 ip.txt)
+    -o result.csv
+        输出结果文件；相对/绝对路径，如包含空格请加上引号；允许 .txt 等后缀；(默认 result.csv)
     -dd
         禁用下载测速；如果带上该参数就是禁用下载测速；(默认 启用)
     -v
@@ -42,15 +48,18 @@ https://github.com/XIU2/CloudflareSpeedTest
         打印帮助说明
 
 示例：
-    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -p 20
-    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -f "C:\abc\ip.txt" -dd`
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -p -1 -f "ip.txt" -o "result.csv" -dd
+    CloudflareST.exe -n 500 -t 4 -dn 20 -dt 10 -p -1 -f "C:\abc\ip.txt" -o "C:\abc\result.csv" -dd`
 
 	flag.IntVar(&pingRoutine, "n", 500, "测速线程数量")
 	flag.IntVar(&pingTime, "t", 4, "延迟测速次数")
 	flag.IntVar(&downloadTestCount, "dn", 20, "下载测速数量")
 	flag.Int64Var(&downloadSecond, "dt", 10, "下载测速时间")
+	flag.IntVar(&printResult, "p", 20, "直接显示结果")
 	flag.BoolVar(&disableDownload, "dd", false, "禁用下载测速")
 	flag.StringVar(&ipFile, "f", "ip.txt", "IP 数据文件")
+	flag.StringVar(&outputFile, "o", "result.csv", "输出结果文件")
 	flag.BoolVar(&printVersion, "v", false, "打印程序版本")
 
 	downloadTestTime = time.Duration(downloadSecond) * time.Second
@@ -61,6 +70,27 @@ https://github.com/XIU2/CloudflareSpeedTest
 		println(version)
 		os.Exit(0)
 	}
+	if pingRoutine <= 0 {
+		pingRoutine = 500
+	}
+	if pingTime <= 0 {
+		pingTime = 4
+	}
+	if downloadTestCount <= 0 {
+		downloadTestCount = 20
+	}
+	if downloadSecond <= 0 {
+		downloadSecond = 10
+	}
+	if printResult == 0 {
+		printResult = 20
+	}
+	if ipFile == "" {
+		ipFile = "ip.txt"
+	}
+	if outputFile == "" {
+		outputFile = "result.csv"
+	}
 }
 
 func main() {
@@ -74,27 +104,58 @@ func main() {
 	var data = make([]CloudflareIPData, 0)
 
 	fmt.Println("开始延迟测速(TCP)：")
-
 	control := make(chan bool, pingRoutine)
 	for _, ip := range ips {
 		wg.Add(1)
 		control <- false
-		handleProgress := handleProgressGenerator(bar)
+		handleProgress := handleProgressGenerator(bar) // 多线程进度条
 		go tcpingGoroutine(&wg, &mu, ip, pingTime, &data, control, handleProgress)
 	}
 	wg.Wait()
 	bar.Finish()
 
 	sort.Sort(CloudflareIPDataSet(data)) // 排序
-	if !disableDownload {                // 如果禁用下载测速就跳过
+
-		bar = pb.Simple.Start(downloadTestCount)
+	// 下载测速
-		fmt.Println("开始下载测速：")
+	if !disableDownload { // 如果禁用下载测速就跳过
-		for i := 0; i < downloadTestCount; i++ {
+		if len(data) > 0 { // IP数组长度(IP数量) 大于 0 时继续
-			_, speed := DownloadSpeedHandler(data[i].ip)
+			if len(data) < downloadTestCount { // 如果IP数组长度(IP数量) 小于 下载测速次数，则次数改为IP数
-			data[i].downloadSpeed = speed
+				downloadTestCount = len(data)
-			bar.Add(1)
+				fmt.Println("\n[信息] IP数量小于下载测速次数，下载测速次数改为IP数。\n")
+			}
+			bar = pb.Simple.Start(downloadTestCount)
+			fmt.Println("开始下载测速：")
+			for i := 0; i < downloadTestCount; i++ {
+				_, speed := DownloadSpeedHandler(data[i].ip)
+				data[i].downloadSpeed = speed
+				bar.Add(1)
+			}
+			bar.Finish()
+		} else {
+			fmt.Println("\n[信息] IP数量为 0，跳过下载测速。")
 		}
-		bar.Finish()
 	}
-	ExportCsv("./result.csv", data) // 输出结果
+
+	// 直接输出结果
+	if printResult > 0 { // 如果禁用下载测速就跳过
+		dateString := convertToString(data) // 转为多维数组 [][]String
+		if len(dateString) > 0 {            // IP数组长度(IP数量) 大于 0 时继续
+			if len(dateString) < printResult { // 如果IP数组长度(IP数量) 小于  打印次数，则次数改为IP数量
+				printResult = len(dateString)
+				fmt.Println("\n[信息] IP数量小于显示结果数量，显示结果数量改为IP数量。\n")
+			}
+			fmt.Println("\nIP 地址    \t", "测试次数\t", "成功次数\t", "成功比率\t", "平均延迟\t", "下载速度 (MB/s)")
+			for i := 0; i < printResult; i++ {
+				fmt.Println(dateString[i][0], "\t", dateString[i][1], "\t\t", dateString[i][2], "\t\t", dateString[i][3], "\t\t", dateString[i][4], "\t", dateString[i][5])
+			}
+			fmt.Printf("\n完整内容请查看 %v 文件。请按 回车键 或 Ctrl+C 退出。", outputFile)
+			var pause int
+			fmt.Scanln(&pause)
+		} else {
+			fmt.Println("\n[信息] IP数量为 0，跳过输出结果。")
+		}
+	}
+
+	// 输出结果到文件
+	ExportCsv(outputFile, data)
 }

tcping.go

@@ -2,13 +2,14 @@ package main
 
 import (
 	"context"
-	"github.com/VividCortex/ewma"
 	"io"
 	"net"
 	"net/http"
 	"strconv"
 	"sync"
 	"time"
+
+	"github.com/VividCortex/ewma"
 )
 
 //bool connectionSucceed float32 time
@@ -130,7 +131,7 @@ func DownloadSpeedHandler(ip net.IPAddr) (bool, float32) {
 			var nextTime = timeStart.Add(timeSlice * time.Duration(timeCounter))
 			e := ewma.NewMovingAverage()
 
-			for ; contentLength != contentRead; {
+			for contentLength != contentRead {
 				var currentTime = time.Now()
 				if currentTime.After(nextTime) {
 					timeCounter += 1

util.go

@@ -36,13 +36,11 @@ func ExportCsv(filePath string, data []CloudflareIPData) {
 	}
 	defer fp.Close()
 	w := csv.NewWriter(fp) //创建一个新的写入文件流
-	w.Write([]string{"IP 地址", "Ping 发送次数", "Ping 接收次数", "Ping 接收率", "平均延迟", "下载速度 (MB/s)"})
+	w.Write([]string{"IP 地址", "测试次数", "成功次数", "成功比率", "平均延迟", "下载速度 (MB/s)"})
 	w.WriteAll(convertToString(data))
 	w.Flush()
 }
 
-//"IP Address","Ping Count","Ping received","Ping received rate","Ping time","Download speed"
-
 func (cf *CloudflareIPData) toString() []string {
 	result := make([]string, 6)
 	result[0] = cf.ip.String()