wg-portal/internal/domain/statistics.go

package domain

import (
	"time"
)

type PeerStatus struct {
	PeerId    PeerIdentifier `gorm:"primaryKey;column:identifier" json:"PeerId"`
	UpdatedAt time.Time      `gorm:"column:updated_at" json:"-"`

	IsConnected bool `gorm:"column:connected" json:"IsConnected"` // indicates if the peer is connected based on the last handshake or ping

	IsPingable bool       `gorm:"column:pingable" json:"IsPingable"`
	LastPing   *time.Time `gorm:"column:last_ping" json:"LastPing"`

	BytesReceived    uint64 `gorm:"column:received" json:"BytesReceived"`
	BytesTransmitted uint64 `gorm:"column:transmitted" json:"BytesTransmitted"`

	LastHandshake    *time.Time `gorm:"column:last_handshake" json:"LastHandshake"`
	Endpoint         string     `gorm:"column:endpoint" json:"Endpoint"`
	LastSessionStart *time.Time `gorm:"column:last_session_start" json:"LastSessionStart"`
}

func (s *PeerStatus) CalcConnected() {
	oldestHandshakeTime := time.Now().Add(-2 * time.Minute) // if a handshake is older than 2 minutes, the peer is no longer connected

	handshakeValid := false
	if s.LastHandshake != nil {
		handshakeValid = !s.LastHandshake.Before(oldestHandshakeTime)
	}

	s.IsConnected = s.IsPingable || handshakeValid
}

type InterfaceStatus struct {
	InterfaceId InterfaceIdentifier `gorm:"primaryKey;column:identifier"`
	UpdatedAt   time.Time           `gorm:"column:updated_at"`

	BytesReceived    uint64 `gorm:"column:received"`
	BytesTransmitted uint64 `gorm:"column:transmitted"`
}

type PingerResult struct {
	PacketsRecv int
	PacketsSent int
	Rtts        []time.Duration
}

func (r PingerResult) IsPingable() bool {
	return r.PacketsRecv > 0 && r.PacketsSent > 0 && len(r.Rtts) > 0
}

func (r PingerResult) AverageRtt() time.Duration {
	if len(r.Rtts) == 0 {
		return 0
	}

	var total time.Duration
	for _, rtt := range r.Rtts {
		total += rtt
	}
	return total / time.Duration(len(r.Rtts))
}

type TrafficDelta struct {
	EntityId                  string `json:"EntityId"` // Either peerId or interfaceId
	BytesReceivedPerSecond    uint64 `json:"BytesReceived"`
	BytesTransmittedPerSecond uint64 `json:"BytesTransmitted"`
}

func CalculateTrafficDelta(id string, oldTime, newTime time.Time, oldTx, newTx, oldRx, newRx uint64) TrafficDelta {
	timeDiff := uint64(newTime.Sub(oldTime).Seconds())
	if timeDiff == 0 {
		return TrafficDelta{
			EntityId:                  id,
			BytesReceivedPerSecond:    0,
			BytesTransmittedPerSecond: 0,
		}
	}
	return TrafficDelta{
		EntityId:                  id,
		BytesReceivedPerSecond:    (newRx - oldRx) / timeDiff,
		BytesTransmittedPerSecond: (newTx - oldTx) / timeDiff,
	}
}