Archive/echo/internal/metrics/ping.go

package metrics

import (
	"math"
	"runtime"
	"time"

	"github.com/Ehco1996/ehco/internal/config"
	"github.com/go-ping/ping"
	"go.uber.org/zap"
)

func (pg *PingGroup) newPinger(ruleLabel string, remote string, addr string) (*ping.Pinger, error) {
	pinger := ping.New(addr)
	if err := pinger.Resolve(); err != nil {
		pg.logger.Error("failed to resolve pinger", zap.String("addr", addr), zap.Error(err))
		return nil, err
	}
	pinger.Interval = pingInterval
	pinger.Timeout = time.Duration(math.MaxInt64)
	pinger.RecordRtts = false

	switch runtime.GOOS {
	case "darwin":
	case "linux":
		pinger.SetPrivileged(true)
	default:
		pinger.SetPrivileged(true)
		pg.logger.Warn("Attempting to set privileged mode for unknown OS", zap.String("OS", runtime.GOOS))
	}

	pinger.OnRecv = func(pkt *ping.Packet) {
		ip := pkt.IPAddr.String()
		PingResponseDurationMilliseconds.WithLabelValues(
			ruleLabel, remote, ip).Observe(float64(pkt.Rtt.Milliseconds()))
		pg.logger.Sugar().Infof("%d bytes from %s icmp_seq=%d time=%v ttl=%v",
			pkt.Nbytes, pkt.Addr, pkt.Seq, pkt.Rtt, pkt.Ttl)
	}
	return pinger, nil
}

type PingGroup struct {
	logger *zap.Logger

	Pingers []*ping.Pinger
}

func NewPingGroup(cfg *config.Config) *PingGroup {
	pg := &PingGroup{logger: zap.L().Named("pinger"), Pingers: make([]*ping.Pinger, 0)}

	for _, relayCfg := range cfg.RelayConfigs {
		for _, remote := range relayCfg.GetAllRemotes() {
			addr, err := remote.GetAddrHost()
			if err != nil {
				pg.logger.Error("try parse host error", zap.Error(err))
				continue
			}
			if pinger, err := pg.newPinger(relayCfg.Label, remote.Address, addr); err != nil {
				pg.logger.Error("new pinger meet error", zap.Error(err))
			} else {
				pg.Pingers = append(pg.Pingers, pinger)
			}
		}
	}
	return pg
}

func (pg *PingGroup) Run() {
	if len(pg.Pingers) <= 0 {
		return
	}
	pg.logger.Sugar().Infof("Start Ping Group now total pinger: %d", len(pg.Pingers))
	splay := time.Duration(pingInterval.Nanoseconds() / int64(len(pg.Pingers)))
	for _, pinger := range pg.Pingers {
		go func() {
			if err := pinger.Run(); err != nil {
				pg.logger.Error("Starting pinger meet err", zap.Error(err), zap.String("addr", pinger.Addr()))
			}
		}()
		time.Sleep(splay)
	}
}