GoProxy/checker/checker.go

package checker

import (
	"log"
	"time"

	"goproxy/config"
	"goproxy/storage"
	"goproxy/validator"
)

type Checker struct {
	storage *storage.Storage
}

func New(s *storage.Storage, _ *validator.Validator, _ *config.Config) *Checker {
	return &Checker{storage: s}
}

func (c *Checker) Start() {
	go func() {
		for {
			cfg := config.Get()
			time.Sleep(time.Duration(cfg.CheckInterval) * time.Minute)
			c.run()
		}
	}()
	log.Printf("health checker started, interval: %d min", config.Get().CheckInterval)
}

func (c *Checker) run() {
	log.Println("[checker] start health check...")

	proxies, err := c.storage.GetAll()
	if err != nil {
		log.Printf("[checker] get proxies error: %v", err)
		return
	}
	if len(proxies) == 0 {
		log.Println("[checker] no proxies to check")
		return
	}

	// 每次用最新配置创建 validator
	cfg := config.Get()
	validate := validator.New(cfg.ValidateConcurrency, cfg.ValidateTimeout, cfg.ValidateURL)

	log.Printf("[checker] checking %d proxies...", len(proxies))
	results := validate.ValidateAll(proxies)

	valid, invalid := 0, 0
	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 {
				log.Printf("[checker] delete error: %v", err)
			}
		}
	}

	count, _ := c.storage.Count()
	log.Printf("[checker] done: valid=%d invalid(deleted)=%d remaining=%d", valid, invalid, count)
}