From 40b22f660a3b9a1f37d524fcd0b93dfb664f7445 Mon Sep 17 00:00:00 2001 From: xiu2 <54703944+XIU2@users.noreply.github.com> Date: Wed, 11 Nov 2020 18:10:53 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=20=E6=8C=87=E5=AE=9A?= =?UTF-8?q?=E5=BB=B6=E8=BF=9F=E6=97=B6=E9=97=B4=E4=B8=8A=E9=99=90=E3=80=81?= =?UTF-8?q?=E4=B8=8B=E8=BD=BD=E9=80=9F=E5=BA=A6=E4=B8=8B=E9=99=90=E6=9D=A1?= =?UTF-8?q?=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 35 ++++++++++++++++++++++------------ main.go | 57 +++++++++++++++++++++++++++++++++++++++++++++---------- 2 files changed, 70 insertions(+), 22 deletions(-) diff --git a/README.md b/README.md index 7e9f70a..64a7587 100644 --- a/README.md +++ b/README.md @@ -38,6 +38,7 @@ IP 地址 已发送 已接收 丢包率 平均延迟 下载速度 104.22.78.104 4 4 0.00 146.75 13.00 104.22.69.208 4 4 0.00 147.00 19.07 104.27.194.10 4 4 0.00 148.02 21.05 +... ``` 完整结果保存在当前目录下的 `result.csv` 文件中，用**记事本/表格软件**打开，排序为**延迟由低到高**，分别是： @@ -56,8 +57,6 @@ IP 地址, 已发送, 已接收, 丢包率, 平均延迟, 下载速度 (MB/s) 直接双击运行使用的是默认参数，如果想要测试速度更快、测试结果更全面，可以自定义参数。 -> **提示：Linux 系统**只需要把下面命令中的 **.exe 删除**即可通用。 - ``` cmd C:\>CloudflareST.exe -h @@ -78,6 +77,10 @@ https://github.com/XIU2/CloudflareSpeedTest 下载测速时间；单个 IP 测速最长时间，单位：秒；(默认 5) -url https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png 下载测速地址；用来 Cloudflare CDN 测速的文件地址，如含有空格请加上引号； + -tl 200 + 延迟时间上限；只输出指定延迟时间以下的结果，数量为 -dn 参数的值，单位：ms； + -sl 5 + 下载速度下限；只输出指定下载速度以上的结果，数量为 -dn 参数的值，单位：MB/s； -p 20 显示结果数量；测速后直接显示指定数量的结果，为 0 时不显示结果直接退出；(默认 20) -f ip.txt @@ -97,27 +100,35 @@ https://github.com/XIU2/CloudflareSpeedTest ### 使用示例 在 CMD 中运行，或者把启动参数添加到快捷方式中。 -> **注意：** 不需要加上所有参数，按需选择，参数前后顺序随意。 > **提示： Linux 系统**只需要把下面命令中的 **.exe 删除**即可通用。 -``` cmd -# 命令行示例 -CloudflareST.exe -n 500 -t 4 -dn 20 -dt 5 +``` bash +# 注意：各参数均有默认值，只有不使用默认值时，才需要手动指定参数的值（按需选择），参数不分前后顺序。 +# 提示： Linux 系统只需要把下面命令中的 .exe 删除即可通用。 # 指定 IP数据文件，不显示结果直接退出（-p 值为 0） -CloudflareST.exe -n 500 -t 4 -dn 20 -dt 5 -p 0 -f "ip.txt" -dd +CloudflareST.exe -p 0 -f ip.txt -dd -# 指定 IP数据文件，不输出结果到文件，直接显示结果（-p 值为 20 条） -CloudflareST.exe -n 500 -t 4 -dn 20 -dt 5 -p 20 -f "ip.txt" -o " " -dd +# 指定 IP数据文件，不输出结果到文件，直接显示结果（-p 值为 10 条） +CloudflareST.exe -p 10 -f ip.txt -o " " -dd # 指定 IP数据文件 及 输出结果到文件（相对路径，即当前目录下，如果包含空格请加上引号） -CloudflareST.exe -n 500 -t 4 -dn 20 -dt 5 -f ip.txt -o result.csv -dd +CloudflareST.exe -f ip.txt -o result.csv -dd # 指定 IP数据文件 及 输出结果到文件（绝对路径，即 C:\abc\ 目录下，如果包含空格请加上引号） -CloudflareST.exe -n 500 -t 4 -dn 20 -dt 5 -f C:\abc\ip.txt -o C:\abc\result.csv -dd +CloudflareST.exe -f C:\abc\ip.txt -o C:\abc\result.csv -dd # 指定下载测速地址（要求：可以直接下载的文件、文件大小超过 200MB、网站用的是 Cloudflare CDN），如果包含空格请加上引号 -CloudflareST.exe -n 500 -t 4 -dn 20 -dt 5 -url https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png +CloudflareST.exe -url https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png + +# 指定测速条件，延迟时间上限：200 ms，下载速度下限：0 MB/s，数量：10 个（只有同时满足三个条件时才会停止测速） +CloudflareST.exe -tl 200 -dn 10 + +# 指定测速条件，延迟时间上限：0 ms，下载速度下限：5 MB/s，数量：10 个（只有同时满足三个条件时才会停止测速） +CloudflareST.exe -sl 5 -dn 10 + +# 指定测速条件，延迟时间上限：200 ms，下载速度下限：5 MB/s，数量：10 个（只有同时满足三个条件时才会停止测速） +CloudflareST.exe -tl 200 -sl 5 -dn 10 ``` ``` cmd diff --git a/main.go b/main.go index c7bf230..4a80b12 100644 --- a/main.go +++ b/main.go @@ -18,6 +18,8 @@ var tcpPort int var ipFile string var outputFile string var printResultNum int +var timeLimit int +var speedLimit int func init() { var downloadSecond int64 @@ -40,6 +42,10 @@ https://github.com/XIU2/CloudflareSpeedTest 下载测速时间；单个 IP 测速最长时间，单位：秒；(默认 5) -url https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png 下载测速地址；用来 Cloudflare CDN 测速的文件地址，如含有空格请加上引号； + -tl 200 + 延迟时间上限；只输出指定延迟时间以下的结果，数量为 -dn 参数的值，单位：ms； + -sl 5 + 下载速度下限；只输出指定下载速度以上的结果，数量为 -dn 参数的值，单位：MB/s； -p 20 显示结果数量；测速后直接显示指定数量的结果，值为 0 时不显示结果直接退出；(默认 20) -f ip.txt @@ -59,6 +65,8 @@ https://github.com/XIU2/CloudflareSpeedTest flag.IntVar(&downloadTestCount, "dn", 20, "下载测速数量") flag.Int64Var(&downloadSecond, "dt", 5, "下载测速时间") flag.StringVar(&url, "url", "https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png", "下载测速地址") + flag.IntVar(&timeLimit, "tl", 0, "延迟时间上限") + flag.IntVar(&speedLimit, "sl", 0, "下载速度下限") flag.IntVar(&printResultNum, "p", 20, "显示结果数量") flag.BoolVar(&disableDownload, "dd", false, "禁用下载测速") flag.StringVar(&ipFile, "f", "ip.txt", "IP 数据文件") @@ -91,6 +99,12 @@ https://github.com/XIU2/CloudflareSpeedTest if url == "" { url = "https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png" } + if timeLimit <= 0 { + timeLimit = 9999 + } + if speedLimit < 0 { + speedLimit = 0 + } if printResultNum < 0 { printResultNum = 20 } @@ -111,6 +125,7 @@ func main() { var wg sync.WaitGroup var mu sync.Mutex var data = make([]CloudflareIPData, 0) + var data_2 = make([]CloudflareIPData, 0) fmt.Println("# XIU2/CloudflareSpeedTest " + version + "\n") fmt.Println("开始延迟测速（模式：TCP，端口：" + strconv.Itoa(tcpPort) + "）：") @@ -130,15 +145,31 @@ func main() { if !disableDownload { // 如果禁用下载测速就跳过 if len(data) > 0 { // IP数组长度(IP数量) 大于 0 时继续 if len(data) < downloadTestCount { // 如果IP数组长度(IP数量) 小于 下载测速次数，则次数改为IP数 + //fmt.Println("\n[信息] IP 数量小于下载测速次数（" + strconv.Itoa(downloadTestCount) + " < " + strconv.Itoa(len(data)) + "），下载测速次数改为IP数。\n") downloadTestCount = len(data) - fmt.Println("\n[信息] IP数量小于下载测速次数，下载测速次数改为IP数。\n") + } - bar = pb.Simple.Start(downloadTestCount) - fmt.Println("开始下载测速：") - for i := 0; i < downloadTestCount; i++ { + var downloadTestCount_2 int // 临时的下载测速次数 + if timeLimit == 9999 && speedLimit == 0 { + downloadTestCount_2 = downloadTestCount // 如果没有指定条件，则临时的下载次数变量为下载测速次数 + fmt.Println("开始下载测速：") + } else if timeLimit > 0 || speedLimit >= 0 { + downloadTestCount_2 = len(data) // 如果指定了任意一个条件，则临时的下载次数变量改为总数量 + fmt.Println("开始下载测速（延迟时间上限：" + strconv.Itoa(timeLimit) + " ms，下载速度下限：" + strconv.Itoa(speedLimit) + " MB/s）：") + } + bar = pb.Simple.Start(downloadTestCount_2) + for i := 0; i < downloadTestCount_2; i++ { _, speed := DownloadSpeedHandler(data[i].ip) data[i].downloadSpeed = speed bar.Add(1) + if int(data[i].pingTime) <= timeLimit && int(float64(speed)/1024/1024) >= speedLimit { + data_2 = append(data_2, data[i]) // 延迟和速度均满足条件时，添加到新数组中 + if len(data_2) == downloadTestCount { // 满足条件的 IP =下载测速次数，则跳出循环 + break + } + } else if int(data[i].pingTime) > timeLimit { + break + } } bar.Finish() } else { @@ -146,11 +177,17 @@ func main() { } } - if outputFile != "" { - ExportCsv(outputFile, data) // 输出结果到文件 + if len(data_2) > 0 { // 如果该数字有内容，说明进行过指定条件的下载测速 + if outputFile != "" { + ExportCsv(outputFile, data_2) // 输出结果到文件（指定延迟时间或下载速度的） + } + printResult(data_2) // 显示最快结果（指定延迟时间或下载速度的） + } else { + if outputFile != "" { + ExportCsv(outputFile, data) // 输出结果到文件 + } + printResult(data) // 显示最快结果 } - - printResult(data) // 显示最快结果 } // 显示最快结果 @@ -159,15 +196,15 @@ func printResult(data []CloudflareIPData) { dateString := convertToString(data) // 转为多维数组 [][]String if len(dateString) > 0 { // IP数组长度(IP数量) 大于 0 时继续 if len(dateString) < printResultNum { // 如果IP数组长度(IP数量) 小于 打印次数，则次数改为IP数量 + //fmt.Println("\n[信息] IP 数量小于显示结果数量（" + strconv.Itoa(printResultNum) + " < " + strconv.Itoa(len(dateString)) + "），显示结果数量改为IP数量。\n") printResultNum = len(dateString) - fmt.Println("\n[信息] IP数量小于显示结果数量，显示结果数量改为IP数量。\n") } fmt.Printf("%-16s%-5s%-5s%-5s%-6s%-11s\n", "IP 地址", "已发送", "已接收", "丢包率", "平均延迟", "下载速度 (MB/s)") for i := 0; i < printResultNum; i++ { fmt.Printf("%-18s%-8s%-8s%-8s%-10s%-15s\n", ipPadding(dateString[i][0]), dateString[i][1], dateString[i][2], dateString[i][3], dateString[i][4], dateString[i][5]) } if outputFile != "" { - fmt.Printf("\n完整测速结果已写入 %v 文件，请使用记事本/表格软件查看。按下 回车键 或 Ctrl+C 退出。", outputFile) + fmt.Printf("\n完整测速结果已写入 %v 文件，请使用记事本/表格软件查看。\n按下 回车键 或 Ctrl+C 退出。", outputFile) } else { fmt.Printf("\n请按 回车键 或 Ctrl+C 退出。") }