diff --git a/internal/metrics/uptime/uptime.go b/internal/metrics/uptime/uptime.go index 825ebff7..df7059a9 100644 --- a/internal/metrics/uptime/uptime.go +++ b/internal/metrics/uptime/uptime.go @@ -4,12 +4,12 @@ import ( "context" "encoding/json" "net/url" - "strings" "time" "slices" "github.com/lithammer/fuzzysearch/fuzzy" + config "github.com/yusing/go-proxy/internal/config/types" "github.com/yusing/go-proxy/internal/metrics/period" metricsutils "github.com/yusing/go-proxy/internal/metrics/utils" "github.com/yusing/go-proxy/internal/route/routes" @@ -23,18 +23,20 @@ type ( } // @name RouteStatusesByAlias Status struct { Status types.HealthStatus `json:"status" swaggertype:"string" enums:"healthy,unhealthy,unknown,napping,starting"` - Latency int64 `json:"latency"` + Latency int32 `json:"latency"` Timestamp int64 `json:"timestamp"` } // @name RouteStatus - RouteStatuses map[string][]*Status // @name RouteStatuses + RouteStatuses map[string][]Status // @name RouteStatuses RouteAggregate struct { - Alias string `json:"alias"` - DisplayName string `json:"display_name"` - Uptime float64 `json:"uptime"` - Downtime float64 `json:"downtime"` - Idle float64 `json:"idle"` - AvgLatency float64 `json:"avg_latency"` - Statuses []*Status `json:"statuses"` + Alias string `json:"alias"` + DisplayName string `json:"display_name"` + Uptime float32 `json:"uptime"` + Downtime float32 `json:"downtime"` + Idle float32 `json:"idle"` + AvgLatency float32 `json:"avg_latency"` + IsDocker bool `json:"is_docker"` + CurrentStatus types.HealthStatus `json:"current_status" swaggertype:"string" enums:"healthy,unhealthy,unknown,napping,starting"` + Statuses []Status `json:"statuses"` } // @name RouteUptimeAggregate Aggregated []RouteAggregate ) @@ -64,9 +66,9 @@ func aggregateStatuses(entries []*StatusByAlias, query url.Values) (int, Aggrega statuses := make(RouteStatuses) for _, entry := range entries { for alias, status := range entry.Map { - statuses[alias] = append(statuses[alias], &Status{ + statuses[alias] = append(statuses[alias], Status{ Status: status.Status, - Latency: status.Latency.Milliseconds(), + Latency: int32(status.Latency.Milliseconds()), Timestamp: entry.Timestamp, }) } @@ -81,12 +83,12 @@ func aggregateStatuses(entries []*StatusByAlias, query url.Values) (int, Aggrega return len(statuses), statuses.aggregate(limit, offset) } -func (rs RouteStatuses) calculateInfo(statuses []*Status) (up float64, down float64, idle float64, _ float64) { +func (rs RouteStatuses) calculateInfo(statuses []Status) (up float32, down float32, idle float32, _ float32) { if len(statuses) == 0 { return 0, 0, 0, 0 } - total := float64(0) - latency := float64(0) + total := float32(0) + latency := float32(0) for _, status := range statuses { // ignoring unknown; treating napping and starting as downtime if status.Status == types.StatusUnknown { @@ -101,7 +103,7 @@ func (rs RouteStatuses) calculateInfo(statuses []*Status) (up float64, down floa down++ } total++ - latency += float64(status.Latency) + latency += float32(status.Latency) } if total == 0 { return 0, 0, 0, 0 @@ -121,34 +123,41 @@ func (rs RouteStatuses) aggregate(limit int, offset int) Aggregated { sortedAliases[i] = alias i++ } - // unknown statuses are at the end, then sort by alias - slices.SortFunc(sortedAliases, func(a, b string) int { - if rs[a][len(rs[a])-1].Status == types.StatusUnknown { - return 1 - } - if rs[b][len(rs[b])-1].Status == types.StatusUnknown { - return -1 - } - return strings.Compare(a, b) - }) + slices.Sort(sortedAliases) sortedAliases = sortedAliases[beg:end] result := make(Aggregated, len(sortedAliases)) for i, alias := range sortedAliases { statuses := rs[alias] up, down, idle, latency := rs.calculateInfo(statuses) - result[i] = RouteAggregate{ - Alias: alias, - Uptime: up, - Downtime: down, - Idle: idle, - AvgLatency: latency, - Statuses: statuses, - } + + displayName := alias r, ok := routes.Get(alias) - if ok { - result[i].DisplayName = r.HomepageConfig().Name - } else { - result[i].DisplayName = alias + if !ok { + // also search for excluded routes + r = config.GetInstance().SearchRoute(alias) + } + if r != nil { + displayName = r.DisplayName() + } + + status := types.StatusUnknown + if r != nil { + mon := r.HealthMonitor() + if mon != nil { + status = mon.Status() + } + } + + result[i] = RouteAggregate{ + Alias: alias, + DisplayName: displayName, + Uptime: up, + Downtime: down, + Idle: idle, + AvgLatency: latency, + CurrentStatus: status, + Statuses: statuses, + IsDocker: r != nil && r.ContainerInfo() != nil, } } return result