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))
}