feat: ✨ init

checker/checker.go

@@ -4,9 +4,9 @@ import (
 	"log"
 	"time"
 
-	"proxy-pool/config"
-	"proxy-pool/storage"
-	"proxy-pool/validator"
+	"goproxy/config"
+	"goproxy/storage"
+	"goproxy/validator"
 )
 
 type Checker struct {
@@ -52,6 +52,19 @@ func (c *Checker) run() {
 	for _, r := range results {
 		if r.Valid {
 			valid++
+			// 更新出口 IP、位置和延迟信息
+			latencyMs := int(r.Latency.Milliseconds())
+			if r.ExitIP != "" && r.Proxy.ExitIP == "" {
+				// 如果之前没有出口 IP 信息，更新完整信息
+				if err := c.storage.UpdateExitInfo(r.Proxy.Address, r.ExitIP, r.ExitLocation, latencyMs); err != nil {
+					log.Printf("[checker] update exit info error: %v", err)
+				}
+			} else if r.Latency > 0 {
+				// 否则只更新延迟
+				if err := c.storage.UpdateLatency(r.Proxy.Address, latencyMs); err != nil {
+					log.Printf("[checker] update latency error: %v", err)
+				}
+			}
 		} else {
 			invalid++
 			if err := c.storage.Delete(r.Proxy.Address); err != nil {

checker/health_checker.go

@@ -0,0 +1,103 @@
+package checker
+
+import (
+	"log"
+	"time"
+
+	"goproxy/config"
+	"goproxy/pool"
+	"goproxy/storage"
+	"goproxy/validator"
+)
+
+// HealthChecker 健康检查器
+type HealthChecker struct {
+	storage   *storage.Storage
+	validator *validator.Validator
+	cfg       *config.Config
+	poolMgr   *pool.Manager
+}
+
+func NewHealthChecker(s *storage.Storage, v *validator.Validator, cfg *config.Config, pm *pool.Manager) *HealthChecker {
+	return &HealthChecker{
+		storage:   s,
+		validator: v,
+		cfg:       cfg,
+		poolMgr:   pm,
+	}
+}
+
+// RunOnce 执行一次健康检查
+func (hc *HealthChecker) RunOnce() {
+	start := time.Now()
+	log.Println("[health] 开始健康检查...")
+
+	// 获取池子状态
+	status, err := hc.poolMgr.GetStatus()
+	if err != nil {
+		log.Printf("[health] 获取状态失败: %v", err)
+		return
+	}
+
+	// 健康状态且S级占比高时，跳过S级代理检查
+	skipSGrade := status.State == "healthy"
+	dist, _ := hc.storage.GetQualityDistribution()
+	sGradeCount := dist["S"]
+	totalCount := status.Total
+	if totalCount > 0 && float64(sGradeCount)/float64(totalCount) > 0.3 {
+		skipSGrade = true
+	}
+
+	// 批量获取需要检查的代理
+	proxies, err := hc.storage.GetBatchForHealthCheck(hc.cfg.HealthCheckBatchSize, skipSGrade)
+	if err != nil {
+		log.Printf("[health] 获取检查批次失败: %v", err)
+		return
+	}
+
+	if len(proxies) == 0 {
+		log.Println("[health] 无需检查的代理")
+		return
+	}
+
+	log.Printf("[health] 检查 %d 个代理（跳过S级=%v）", len(proxies), skipSGrade)
+
+	// 执行验证
+	validCount := 0
+	removeCount := 0
+	updateCount := 0
+
+	for result := range hc.validator.ValidateStream(proxies) {
+		if result.Valid {
+			validCount++
+			// 更新延迟和质量等级
+			latencyMs := int(result.Latency.Milliseconds())
+			if err := hc.storage.UpdateExitInfo(result.Proxy.Address, result.ExitIP, result.ExitLocation, latencyMs); err == nil {
+				updateCount++
+			}
+		} else {
+			// 失败次数+1
+			hc.storage.IncrementFailCount(result.Proxy.Address)
+			// 如果失败次数 >= 3，删除
+			if result.Proxy.FailCount+1 >= 3 {
+				hc.storage.Delete(result.Proxy.Address)
+				removeCount++
+			}
+		}
+	}
+
+	elapsed := time.Since(start)
+	log.Printf("[health] 完成: 验证%d 有效%d 更新%d 移除%d 耗时%v",
+		len(proxies), validCount, updateCount, removeCount, elapsed)
+}
+
+// StartBackground 后台定时健康检查
+func (hc *HealthChecker) StartBackground() {
+	ticker := time.NewTicker(time.Duration(hc.cfg.HealthCheckInterval) * time.Minute)
+	go func() {
+		for range ticker.C {
+			hc.RunOnce()
+		}
+	}()
+	log.Printf("[health] 健康检查器已启动，间隔 %d 分钟", hc.cfg.HealthCheckInterval)
+}