fix: 兼容旧版Go — atomic.Int64→int64+atomic, any→interface{}, os→ioutil

search-proxy/main.go

@@ -5,7 +5,7 @@ import (
 	"encoding/json"
 	"flag"
 	"fmt"
-	"io"
+	"io/ioutil"
 	"log"
 	"net"
 	"net/http"
@@ -27,14 +27,14 @@ const version = "2.2.0"
 
 // ── CLI ──
 var (
-	cliPort    = flag.Int("port", 7899, "监听端口")
-	cliCache   = flag.Int("cache", 5, "缓存时间(分钟)")
-	cliRate    = flag.Int("rate", 30, "限流速率(req/s)")
-	cliBurst   = flag.Int("burst", 60, "限流突发容量")
-	cliTimeout = flag.Int("timeout", 12, "搜索引擎请求超时(秒)")
+	cliPort    = flag.Int("port", 7899, "")
+	cliCache   = flag.Int("cache", 5, "")
+	cliRate    = flag.Int("rate", 30, "")
+	cliBurst   = flag.Int("burst", 60, "")
+	cliTimeout = flag.Int("timeout", 12, "")
 	cliVersion = flag.Bool("version", false, "显示版本")
-	cliStop    = flag.Bool("stop", false, "停止后台运行的实例")
-	cliStart   = flag.Bool("start", false, "后台启动（nohup模式）")
+	cliStop    = flag.Bool("stop", false, "停止后台实例")
+	cliStart   = flag.Bool("start", false, "后台启动")
 	cliStatus  = flag.Bool("status", false, "查看运行状态")
 	cliRestart = flag.Bool("restart", false, "重启后台实例")
 )
@@ -43,11 +43,11 @@ var pidFile = "/tmp/metona-search-proxy.pid"
 
 // ── 统计 ──
 type statsData struct {
-	ReqTotal   atomic.Int64
-	ReqRate    atomic.Int64   // 被限流的
-	CacheHit   atomic.Int64
-	EngineOK   [4]atomic.Int64 // google/bing/baidu/ddg 成功
-	EngineFail [4]atomic.Int64 // google/bing/baidu/ddg 失败
+	ReqTotal   int64
+	ReqRate    int64
+	CacheHit   int64
+	EngineOK   [4]int64
+	EngineFail [4]int64
 	StartTime  time.Time
 }
 
@@ -55,43 +55,43 @@ var stats = &statsData{StartTime: time.Now()}
 
 // ── 限流器 ──
 type RateLimiter struct {
-	tokens   atomic.Int64
+	tokens   int64
 	capacity int64
 	rate     int64
-	lastFill atomic.Int64
+	lastFill int64
 }
 
 func newRateLimiter(rate, cap int64) *RateLimiter {
 	rl := &RateLimiter{rate: rate, capacity: cap}
-	rl.tokens.Store(cap)
-	rl.lastFill.Store(time.Now().UnixNano())
+	atomic.StoreInt64(&rl.tokens, cap)
+	atomic.StoreInt64(&rl.lastFill, time.Now().UnixNano())
 	return rl
 }
 
 func (rl *RateLimiter) Allow() bool {
 	now := time.Now().UnixNano()
-	last := rl.lastFill.Load()
+	last := atomic.LoadInt64(&rl.lastFill)
 	elapsed := float64(now-last) / 1e9
 	newTokens := int64(elapsed * float64(rl.rate))
 	if newTokens > 0 {
-		rl.lastFill.Store(now)
+		atomic.StoreInt64(&rl.lastFill, now)
 		for {
-			current := rl.tokens.Load()
+			current := atomic.LoadInt64(&rl.tokens)
 			next := current + newTokens
 			if next > rl.capacity {
 				next = rl.capacity
 			}
-			if rl.tokens.CompareAndSwap(current, next) {
+			if atomic.CompareAndSwapInt64(&rl.tokens, current, next) {
 				break
 			}
 		}
 	}
 	for {
-		current := rl.tokens.Load()
+		current := atomic.LoadInt64(&rl.tokens)
 		if current <= 0 {
 			return false
 		}
-		if rl.tokens.CompareAndSwap(current, current-1) {
+		if atomic.CompareAndSwapInt64(&rl.tokens, current, current-1) {
 			return true
 		}
 	}
@@ -105,14 +105,12 @@ type cacheEntry struct {
 	ExpiresAt int64
 }
 
-var cacheSize atomic.Int64
-
 func cleanCacheLoop(interval time.Duration) {
 	for {
 		time.Sleep(interval)
 		now := time.Now().Unix()
 		count := 0
-		cache.Range(func(key, value any) bool {
+		cache.Range(func(key, value interface{}) bool {
 			ce := value.(cacheEntry)
 			if now >= ce.ExpiresAt {
 				cache.Delete(key)
@@ -121,12 +119,12 @@ func cleanCacheLoop(interval time.Duration) {
 			return true
 		})
 		if count > 0 {
-			log.Printf("cache cleaned: %d entries removed", count)
+			log.Printf("cache cleaned: %d", count)
 		}
 	}
 }
 
-// ── 正则预编译（全局，避免每次搜索重新编译）──
+// ── 预编译正则 ──
 var (
 	reGoogleBlock = regexp.MustCompile(`<(?:div|li)[^>]*class="[^"]*(?:Gx5Zad|g|kvH3mc)[^"]*"[^>]*>([\s\S]*?)(?=<(?:div|li)[^>]*class="[^"]*(?:Gx5Zad|kvH3mc)[^"]*"|<(?:div|g-section)|\z)`)
 	reGoogleTitle = regexp.MustCompile(`<a[^>]*href="(https?://[^"]*)"[^>]*>[\s\S]*?<h3[^>]*>([\s\S]*?)</h3>`)
@@ -163,7 +161,6 @@ type SearchResp struct {
 	FromCache bool     `json:"from_cache,omitempty"`
 }
 
-// ── 全局 ──
 var (
 	httpClient *http.Client
 	ua         = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36"
@@ -172,15 +169,14 @@ var (
 	cacheTTL   int64
 )
 
-// ── PID 管理 ──
-
+// ── PID ──
 func writePID(port int) {
 	pid := os.Getpid()
-	os.WriteFile(pidFile, []byte(fmt.Sprintf("%d\n%d", pid, port)), 0644)
+	ioutil.WriteFile(pidFile, []byte(fmt.Sprintf("%d\n%d", pid, port)), 0644)
 }
 
 func readPID() (pid int, port int, ok bool) {
-	data, err := os.ReadFile(pidFile)
+	data, err := ioutil.ReadFile(pidFile)
 	if err != nil {
 		return 0, 0, false
 	}
@@ -294,7 +290,6 @@ func doRestart() {
 }
 
 // ── recovery ──
-
 func recoverWrap(h http.HandlerFunc) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
 		defer func() {
@@ -307,9 +302,9 @@ func recoverWrap(h http.HandlerFunc) http.HandlerFunc {
 	}
 }
 
-// ── 引擎实现 ──
-
+// ── 引擎 ──
 var engineNames = []string{"google", "bing", "baidu", "ddg"}
+var engineIdx = map[string]int{"google": 0, "bing": 1, "baidu": 2, "ddg": 3}
 
 func doGet(u string) (*http.Response, error) {
 	req, _ := http.NewRequest("GET", u, nil)
@@ -318,20 +313,37 @@ func doGet(u string) (*http.Response, error) {
 	return httpClient.Do(req)
 }
 
+func codeLog(engine string, code int) {
+	if code == 0 {
+		log.Printf("%s: network error", engine)
+		return
+	}
+	switch code {
+	case 429:
+		log.Printf("%s: 429 rate limited", engine)
+	case 503:
+		log.Printf("%s: 503 unavailable", engine)
+	case 403:
+		log.Printf("%s: 403 blocked (可能需要验证码)", engine)
+	default:
+		log.Printf("%s: HTTP %d", engine, code)
+	}
+}
+
 func searchGoogle(q string, n int) []Result {
 	u := fmt.Sprintf("https://www.google.com/search?q=%s&num=%d&hl=zh-CN", url.QueryEscape(q), n)
 	resp, err := doGet(u)
 	if err != nil || resp.StatusCode != 200 {
 		if resp != nil {
-			resp.Body.Close()
 			codeLog("google", resp.StatusCode)
+			resp.Body.Close()
 		} else {
 			codeLog("google", 0)
 		}
 		return nil
 	}
 	defer resp.Body.Close()
-	body, _ := io.ReadAll(resp.Body)
+	body, _ := ioutil.ReadAll(resp.Body)
 	html := string(body)
 
 	var out []Result
@@ -361,15 +373,15 @@ func searchBing(q string, n int) []Result {
 	resp, err := doGet(u)
 	if err != nil || resp.StatusCode != 200 {
 		if resp != nil {
-			resp.Body.Close()
 			codeLog("bing", resp.StatusCode)
+			resp.Body.Close()
 		} else {
 			codeLog("bing", 0)
 		}
 		return nil
 	}
 	defer resp.Body.Close()
-	body, _ := io.ReadAll(resp.Body)
+	body, _ := ioutil.ReadAll(resp.Body)
 	html := string(body)
 
 	var out []Result
@@ -399,15 +411,15 @@ func searchDDG(q string, n int) []Result {
 	resp, err := doGet(u)
 	if err != nil || resp.StatusCode != 200 {
 		if resp != nil {
-			resp.Body.Close()
 			codeLog("ddg", resp.StatusCode)
+			resp.Body.Close()
 		} else {
 			codeLog("ddg", 0)
 		}
 		return nil
 	}
 	defer resp.Body.Close()
-	body, _ := io.ReadAll(resp.Body)
+	body, _ := ioutil.ReadAll(resp.Body)
 	html := string(body)
 
 	var out []Result
@@ -430,15 +442,15 @@ func searchBaidu(q string, n int) []Result {
 	resp, err := doGet(u)
 	if err != nil || resp.StatusCode != 200 {
 		if resp != nil {
-			resp.Body.Close()
 			codeLog("baidu", resp.StatusCode)
+			resp.Body.Close()
 		} else {
 			codeLog("baidu", 0)
 		}
 		return nil
 	}
 	defer resp.Body.Close()
-	body, _ := io.ReadAll(resp.Body)
+	body, _ := ioutil.ReadAll(resp.Body)
 	html := string(body)
 
 	var out []Result
@@ -469,28 +481,7 @@ func searchBaidu(q string, n int) []Result {
 	return out
 }
 
-func codeLog(engine string, code int) {
-	if code == 0 {
-		log.Printf("%s: network error", engine)
-		return
-	}
-	switch code {
-	case 429:
-		log.Printf("%s: 429 rate limited", engine)
-	case 503:
-		log.Printf("%s: 503 unavailable", engine)
-	case 403:
-		log.Printf("%s: 403 blocked (可能需要验证码)", engine)
-	default:
-		log.Printf("%s: HTTP %d", engine, code)
-	}
-}
-
-// ── 引擎索引 ──
-var engineIdx = map[string]int{"google": 0, "bing": 1, "baidu": 2, "ddg": 3}
-
 // ── 工具 ──
-
 func clean(s string) string {
 	s = reHTMLTags.ReplaceAllString(s, "")
 	s = decodeEntities(s)
@@ -509,9 +500,9 @@ func decodeEntities(s string) string {
 func recordEngine(eng string, ok bool) {
 	if idx, exists := engineIdx[eng]; exists {
 		if ok {
-			stats.EngineOK[idx].Add(1)
+			atomic.AddInt64(&stats.EngineOK[idx], 1)
 		} else {
-			stats.EngineFail[idx].Add(1)
+			atomic.AddInt64(&stats.EngineFail[idx], 1)
 		}
 	}
 }
@@ -547,11 +538,10 @@ func parallelSearch(q string, n int) []Result {
 }
 
 // ── Handler ──
-
 func handleSearch(w http.ResponseWriter, r *http.Request) {
-	stats.ReqTotal.Add(1)
+	atomic.AddInt64(&stats.ReqTotal, 1)
 	if !limiter.Allow() {
-		stats.ReqRate.Add(1)
+		atomic.AddInt64(&stats.ReqRate, 1)
 		writeJSON(w, http.StatusTooManyRequests, SearchResp{Error: "rate limited"})
 		return
 	}
@@ -580,7 +570,7 @@ func handleSearch(w http.ResponseWriter, r *http.Request) {
 	if v, ok := cache.Load(ck); ok {
 		ce := v.(cacheEntry)
 		if time.Now().Unix() < ce.ExpiresAt {
-			stats.CacheHit.Add(1)
+			atomic.AddInt64(&stats.CacheHit, 1)
 			writeJSON(w, http.StatusOK, SearchResp{
 				Success: true, Query: q, Engine: engine,
 				Total: len(ce.Results), Results: ce.Results, FromCache: true,
@@ -629,9 +619,9 @@ func handleHealth(w http.ResponseWriter, r *http.Request) {
 
 func handleStats(w http.ResponseWriter, r *http.Request) {
 	uptime := time.Since(stats.StartTime).Round(time.Second).String()
-	total := stats.ReqTotal.Load()
-	cacheHit := stats.CacheHit.Load()
-	rate := stats.ReqRate.Load()
+	total := atomic.LoadInt64(&stats.ReqTotal)
+	cacheHit := atomic.LoadInt64(&stats.CacheHit)
+	rate := atomic.LoadInt64(&stats.ReqRate)
 
 	cacheRate := 0.0
 	if total > 0 {
@@ -645,10 +635,10 @@ func handleStats(w http.ResponseWriter, r *http.Request) {
 		"cache_hit":  cacheHit,
 		"cache_rate": fmt.Sprintf("%.1f%%", cacheRate),
 		"engines": map[string]interface{}{
-			"google": map[string]int64{"ok": stats.EngineOK[0].Load(), "fail": stats.EngineFail[0].Load()},
-			"bing":   map[string]int64{"ok": stats.EngineOK[1].Load(), "fail": stats.EngineFail[1].Load()},
-			"baidu":  map[string]int64{"ok": stats.EngineOK[2].Load(), "fail": stats.EngineFail[2].Load()},
-			"ddg":    map[string]int64{"ok": stats.EngineOK[3].Load(), "fail": stats.EngineFail[3].Load()},
+			"google": map[string]int64{"ok": atomic.LoadInt64(&stats.EngineOK[0]), "fail": atomic.LoadInt64(&stats.EngineFail[0])},
+			"bing":   map[string]int64{"ok": atomic.LoadInt64(&stats.EngineOK[1]), "fail": atomic.LoadInt64(&stats.EngineFail[1])},
+			"baidu":  map[string]int64{"ok": atomic.LoadInt64(&stats.EngineOK[2]), "fail": atomic.LoadInt64(&stats.EngineFail[2])},
+			"ddg":    map[string]int64{"ok": atomic.LoadInt64(&stats.EngineOK[3]), "fail": atomic.LoadInt64(&stats.EngineFail[3])},
 		},
 		"version": version,
 	})
@@ -661,6 +651,7 @@ func writeJSON(w http.ResponseWriter, status int, v interface{}) {
 	json.NewEncoder(w).Encode(v)
 }
 
+// ── main ──
 func main() {
 	flag.Parse()
 
@@ -690,9 +681,8 @@ func main() {
 		port = fmt.Sprintf("%d", *cliPort)
 	}
 	portNum, _ := strconv.Atoi(port)
-
 	if checkPort(portNum) {
-		log.Fatalf("端口 %s 已被占用。使用 metona-search-proxy -status 查看。", port)
+		log.Fatalf("端口 %s 已被占用", port)
 	}
 
 	httpClient = &http.Client{
@@ -706,7 +696,6 @@ func main() {
 	limiter = newRateLimiter(int64(*cliRate), int64(*cliBurst))
 	cacheTTL = int64(*cliCache)
 
-	// 后台清理过期缓存
 	go cleanCacheLoop(30 * time.Second)
 
 	writePID(portNum)