新增下载测速功能

util.go

@@ -0,0 +1,121 @@
+package main
+
+import (
+	"encoding/csv"
+	"github.com/cheggaaa/pb/v3"
+	"log"
+	"math/rand"
+	"net"
+	"os"
+	"strconv"
+	"time"
+)
+
+type CloudflareIPData struct {
+	ip            net.IPAddr
+	pingCount     int
+	pingReceived  int
+	recvRate      float32
+	downloadSpeed float32
+	pingTime      float32
+}
+
+func (cf *CloudflareIPData) getRecvRate() float32 {
+	if cf.recvRate == 0 {
+		cf.recvRate = float32(cf.pingReceived) / float32(cf.pingCount)
+	}
+	return cf.recvRate
+}
+
+func ExportCsv(filePath string, data []CloudflareIPData) {
+	fp, err := os.Create(filePath)
+	if err != nil {
+		log.Fatalf("创建文件["+filePath+"]句柄失败,%v", err)
+		return
+	}
+	defer fp.Close()
+	w := csv.NewWriter(fp) //创建一个新的写入文件流
+	w.Write([]string{"IP Address", "Ping count", "Ping received", "Ping received rate", "Ping time", "Download Speed (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()
+	result[1] = strconv.Itoa(cf.pingCount)
+	result[2] = strconv.Itoa(cf.pingReceived)
+	result[3] = strconv.FormatFloat(float64(cf.getRecvRate()), 'f', 4, 32)
+	result[4] = strconv.FormatFloat(float64(cf.pingTime), 'f', 4, 32)
+	result[5] = strconv.FormatFloat(float64(cf.downloadSpeed)/1024/1024, 'f', 4, 32)
+	return result
+}
+
+func convertToString(data []CloudflareIPData) [][]string {
+	result := make([][]string, 0)
+	for _, v := range data {
+		result = append(result, v.toString())
+	}
+	return result
+}
+
+var pingTime int
+var pingRoutine int
+
+var ipEndWith uint8 = 0
+
+type progressEvent int
+
+const (
+	NoAvailableIPFound progressEvent = iota
+	AvailableIPFound
+	NormalPing
+)
+
+const url string = "https://apple.freecdn.workers.dev/105/media/us/iphone-11-pro/2019/3bd902e4-0752-4ac1-95f8-6225c32aec6d/films/product/iphone-11-pro-product-tpl-cc-us-2019_1280x720h.mp4"
+
+var downloadTestTime time.Duration
+
+const downloadBufferSize = 1024
+
+var downloadTestCount int
+
+const defaultTcpPort = 443
+const tcpConnectTimeout = time.Second * 1
+const failTime = 4
+
+type CloudflareIPDataSet []CloudflareIPData
+
+func initipEndWith() {
+	ipEndWith = uint8(rand.Intn(254) + 1)
+}
+
+func handleProgressGenerator(pb *pb.ProgressBar) func(e progressEvent) {
+	return func(e progressEvent) {
+		switch e {
+		case NoAvailableIPFound:
+			pb.Add(pingTime)
+		case AvailableIPFound:
+			pb.Add(failTime)
+		case NormalPing:
+			pb.Increment()
+		}
+	}
+}
+
+func (cfs CloudflareIPDataSet) Len() int {
+	return len(cfs)
+}
+
+func (cfs CloudflareIPDataSet) Less(i, j int) bool {
+	if (cfs)[i].getRecvRate() != cfs[j].getRecvRate() {
+		return cfs[i].getRecvRate() > cfs[j].getRecvRate()
+	}
+	return cfs[i].pingTime < cfs[j].pingTime
+}
+
+func (cfs CloudflareIPDataSet) Swap(i, j int) {
+	cfs[i], cfs[j] = cfs[j], cfs[i]
+}