komari-agent/monitoring/net.go

package monitoring

import (
	"fmt"

	"github.com/shirou/gopsutil/net"
)

func ConnectionsCount() (tcpCount, udpCount int, err error) {
	tcps, err := net.Connections("tcp")
	if err != nil {
		return 0, 0, fmt.Errorf("failed to get TCP connections: %w", err)
	}
	udps, err := net.Connections("udp")
	if err != nil {
		return 0, 0, fmt.Errorf("failed to get UDP connections: %w", err)
	}

	return len(tcps), len(udps), nil
}

var (
	lastUp   uint64
	lastDown uint64
)

func NetworkSpeed(interval int) (totalUp, totalDown, upSpeed, downSpeed uint64, err error) {
	// Get the network IO counters
	ioCounters, err := net.IOCounters(false)
	if err != nil {
		return 0, 0, 0, 0, fmt.Errorf("failed to get network IO counters: %w", err)
	}

	if len(ioCounters) == 0 {
		return 0, 0, 0, 0, fmt.Errorf("no network interfaces found")
	}

	for _, interfaceStats := range ioCounters {
		loopbackNames := []string{"lo", "lo0", "localhost", "brd0", "docker0", "docker1", "veth0", "veth1", "veth2", "veth3", "veth4", "veth5", "veth6", "veth7"}
		isLoopback := false
		for _, name := range loopbackNames {
			if interfaceStats.Name == name {
				isLoopback = true
				break
			}
		}
		if isLoopback {
			continue // Skip loopback interface
		}
		totalUp += interfaceStats.BytesSent
		totalDown += interfaceStats.BytesRecv

	}
	upSpeed = (totalUp - lastUp) / uint64(interval)
	downSpeed = (totalDown - lastDown) / uint64(interval)

	lastUp = totalUp
	lastDown = totalDown
	return totalUp, totalDown, upSpeed, downSpeed, nil
}