✨ More detailed metrics

pkg/navi/config.go

@@ -6,8 +6,6 @@ import (
 	"path/filepath"
 	"strings"
 
-	"github.com/spf13/viper"
-
 	"github.com/pelletier/go-toml/v2"
 )
 
@@ -16,7 +14,12 @@ var R *RoadApp
 func ReadInConfig(root string) error {
 	instance := &RoadApp{
 		Regions: make([]*Region, 0),
-		Traces:  make([]RoadTrace, 0, viper.GetInt("performance.traces_limit")),
+		Metrics: &RoadMetrics{
+			Traces:       make([]RoadTrace, 0),
+			Traffic:      make(map[string]int64),
+			TrafficFrom:  make(map[string]int64),
+			TotalTraffic: 0,
+		},
 	}
 
 	if err := filepath.Walk(root, func(fp string, info os.FileInfo, _ error) error {

pkg/navi/metrics.go

@@ -2,6 +2,14 @@ package navi
 
 import "github.com/spf13/viper"
 
+type RoadMetrics struct {
+	Traces []RoadTrace `json:"-"`
+
+	Traffic      map[string]int64 `json:"traffic"`
+	TrafficFrom  map[string]int64 `json:"traffic_from"`
+	TotalTraffic int64            `json:"total_traffic"`
+}
+
 type RoadTrace struct {
 	Region      string         `json:"region"`
 	Location    string         `json:"location"`
@@ -17,8 +25,28 @@ type RoadTraceError struct {
 	Message string `json:"message"`
 }
 
-func (v *RoadApp) AddTrace(trace RoadTrace) {
+func (v *RoadMetrics) AddTrace(trace RoadTrace) {
+	v.TotalTraffic++
+	if _, ok := v.Traffic[trace.Region]; !ok {
+		v.Traffic[trace.Region] = 0
+	} else {
+		v.Traffic[trace.Region]++
+	}
+	if _, ok := v.TrafficFrom[trace.IpAddress]; !ok {
+		v.TrafficFrom[trace.IpAddress] = 0
+	} else {
+		v.TrafficFrom[trace.IpAddress]++
+	}
+
 	v.Traces = append(v.Traces, trace)
+
+	// Garbage recycle
+	if len(v.Traffic) > viper.GetInt("performance.traces_limit") {
+		v.Traffic = make(map[string]int64)
+	}
+	if len(v.TrafficFrom) > viper.GetInt("performance.traces_limit") {
+		v.TrafficFrom = make(map[string]int64)
+	}
 	if len(v.Traces) > viper.GetInt("performance.traces_limit") {
 		v.Traces = v.Traces[1:]
 	}

pkg/navi/route.go

@@ -10,8 +10,8 @@ import (
 )
 
 type RoadApp struct {
-	Regions []*Region   `json:"regions"`
-	Traces  []RoadTrace `json:"traces"`
+	Regions []*Region    `json:"regions"`
+	Metrics *RoadMetrics `json:"metrics"`
 }
 
 func (v *RoadApp) Forward(c *fiber.Ctx, dest *Destination) error {