package navi import ( "github.com/spf13/viper" "time" ) type RoadMetrics struct { Traces []RoadTrace `json:"-"` Traffic map[string]int64 `json:"traffic"` TrafficFrom map[string]int64 `json:"traffic_from"` TotalTraffic int64 `json:"total_traffic"` StartupAt time.Time `json:"startup_at"` } type RoadTrace struct { Timestamp time.Time `json:"timestamp"` Region string `json:"region"` Location string `json:"location"` Destination string `json:"destination"` Uri string `json:"uri"` IpAddress string `json:"ip_address"` UserAgent string `json:"user_agent"` Error RoadTraceError `json:"error"` } type RoadTraceError struct { IsNull bool `json:"is_null"` Message string `json:"message"` } func (v *RoadMetrics) AddTrace(trace RoadTrace) { if viper.GetBool("performance.low_memory") { return } v.TotalTraffic++ trace.Timestamp = time.Now() 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") { clear(v.Traffic) } if len(v.TrafficFrom) > viper.GetInt("performance.traces_limit") { clear(v.TrafficFrom) } if len(v.Traces) > viper.GetInt("performance.traces_limit") { clear(v.Traces) } }