mihomo/adapter/provider/healthcheck.go

package provider

import (
	"context"
	"github.com/Dreamacro/clash/common/singledo"
	"time"

	"github.com/Dreamacro/clash/common/batch"
	C "github.com/Dreamacro/clash/constant"

	"go.uber.org/atomic"
)

const (
	defaultURLTestTimeout = time.Second * 5
)

type HealthCheckOption struct {
	URL      string
	Interval uint
}

type HealthCheck struct {
	url       string
	proxies   []C.Proxy
	interval  uint
	lazy      bool
	lastTouch *atomic.Int64
	done      chan struct{}
	singleDo  *singledo.Single[struct{}]
}

func (hc *HealthCheck) process() {
	ticker := time.NewTicker(time.Duration(hc.interval) * time.Second)

	go func() {
		time.Sleep(30 * time.Second)
		hc.lazyCheck()
	}()

	for {
		select {
		case <-ticker.C:
			hc.lazyCheck()
		case <-hc.done:
			ticker.Stop()
			return
		}
	}
}

func (hc *HealthCheck) lazyCheck() bool {
	now := time.Now().Unix()
	if !hc.lazy || now-hc.lastTouch.Load() < int64(hc.interval) {
		hc.check()
		return true
	} else {
		return false
	}
}

func (hc *HealthCheck) setProxy(proxies []C.Proxy) {
	hc.proxies = proxies
}

func (hc *HealthCheck) auto() bool {
	return hc.interval != 0
}

func (hc *HealthCheck) touch() {
	hc.lastTouch.Store(time.Now().Unix())
}

func (hc *HealthCheck) check() {
	_, _, _ = hc.singleDo.Do(func() (struct{}, error) {
		b, _ := batch.New[bool](context.Background(), batch.WithConcurrencyNum[bool](10))
		for _, proxy := range hc.proxies {
			p := proxy
			b.Go(p.Name(), func() (bool, error) {
				ctx, cancel := context.WithTimeout(context.Background(), defaultURLTestTimeout)
				defer cancel()
				_, _ = p.URLTest(ctx, hc.url)
				return false, nil
			})
		}

		b.Wait()
		return struct{}{}, nil
	})
}

func (hc *HealthCheck) close() {
	hc.done <- struct{}{}
}

func NewHealthCheck(proxies []C.Proxy, url string, interval uint, lazy bool) *HealthCheck {
	return &HealthCheck{
		proxies:   proxies,
		url:       url,
		interval:  interval,
		lazy:      lazy,
		lastTouch: atomic.NewInt64(0),
		done:      make(chan struct{}, 1),
		singleDo:  singledo.NewSingle[struct{}](time.Second),
	}
}