feat: ✨ init

test/test_proxy.go

@@ -0,0 +1,122 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+	"os"
+	"os/signal"
+	"strings"
+	"syscall"
+	"time"
+)
+
+const (
+	proxyHost    = "127.0.0.1"
+	testURL      = "http://ip-api.com/json/?fields=countryCode,query"
+	delaySeconds = 1
+)
+
+var proxyPort = "7777"
+
+type IPResponse struct {
+	Query       string `json:"query"`
+	CountryCode string `json:"countryCode"`
+}
+
+var (
+	totalCount   = 0
+	successCount = 0
+)
+
+// 国家代码转 emoji 旗帜
+func countryToEmoji(countryCode string) string {
+	if countryCode == "" {
+		return "🌐"
+	}
+	
+	countryCode = strings.ToUpper(countryCode)
+	if len(countryCode) != 2 {
+		return "🌐"
+	}
+	
+	// 将国家代码转换为 emoji
+	// A=127462, 所以 'US' -> 🇺🇸
+	first := rune(countryCode[0]) - 'A' + 127462
+	second := rune(countryCode[1]) - 'A' + 127462
+	
+	return string([]rune{first, second})
+}
+
+func printStats() {
+	fmt.Println()
+	fmt.Println("---")
+	lossCount := totalCount - successCount
+	lossRate := 0.0
+	if totalCount > 0 {
+		lossRate = float64(lossCount) / float64(totalCount) * 100
+	}
+	fmt.Printf("%d requests transmitted, %d received, %d failed, %.1f%% packet loss\n",
+		totalCount, successCount, lossCount, lossRate)
+	os.Exit(0)
+}
+
+func testHTTPProxyContinuous() {
+	fmt.Printf("PROXY %s:%s (%s): continuous mode\n", proxyHost, proxyPort, testURL)
+	fmt.Println()
+
+	proxyURL, _ := url.Parse(fmt.Sprintf("http://%s:%s", proxyHost, proxyPort))
+	client := &http.Client{
+		Transport: &http.Transport{
+			Proxy: http.ProxyURL(proxyURL),
+		},
+		Timeout: 15 * time.Second,
+	}
+
+	// 捕获 Ctrl+C 信号
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
+	go func() {
+		<-sigChan
+		printStats()
+	}()
+
+	for {
+		totalCount++
+
+		start := time.Now()
+		resp, err := client.Get(testURL)
+		elapsed := time.Since(start).Milliseconds()
+
+		if err != nil {
+			fmt.Printf("proxy #%d: request failed (%v)\n", totalCount, err)
+			time.Sleep(delaySeconds * time.Second)
+			continue
+		}
+
+		body, _ := io.ReadAll(resp.Body)
+		resp.Body.Close()
+
+		var ipResp IPResponse
+		if err := json.Unmarshal(body, &ipResp); err == nil {
+			flag := countryToEmoji(ipResp.CountryCode)
+			fmt.Printf("proxy from %s %s: seq=%d time=%dms\n", flag, ipResp.Query, totalCount, elapsed)
+			successCount++
+		} else {
+			fmt.Printf("proxy #%d: parse error\n", totalCount)
+		}
+
+		time.Sleep(delaySeconds * time.Second)
+	}
+}
+
+func main() {
+	// 支持指定端口号
+	if len(os.Args) > 1 {
+		proxyPort = os.Args[1]
+	}
+	
+	testHTTPProxyContinuous()
+}