mirror of
https://github.com/TwiN/gatus.git
synced 2024-12-14 11:58:04 +00:00
118 lines
3.3 KiB
Go
118 lines
3.3 KiB
Go
package api
|
|
|
|
import (
|
|
"log"
|
|
"math"
|
|
"net/http"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/TwiN/gatus/v5/storage/store"
|
|
"github.com/TwiN/gatus/v5/storage/store/common"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/wcharczuk/go-chart/v2"
|
|
"github.com/wcharczuk/go-chart/v2/drawing"
|
|
)
|
|
|
|
const timeFormat = "3:04PM"
|
|
|
|
var (
|
|
gridStyle = chart.Style{
|
|
StrokeColor: drawing.Color{R: 119, G: 119, B: 119, A: 40},
|
|
StrokeWidth: 1.0,
|
|
}
|
|
axisStyle = chart.Style{
|
|
FontColor: drawing.Color{R: 119, G: 119, B: 119, A: 255},
|
|
}
|
|
transparentStyle = chart.Style{
|
|
FillColor: drawing.Color{R: 255, G: 255, B: 255, A: 0},
|
|
}
|
|
)
|
|
|
|
func ResponseTimeChart(c *fiber.Ctx) error {
|
|
duration := c.Params("duration")
|
|
var from time.Time
|
|
switch duration {
|
|
case "7d":
|
|
from = time.Now().Truncate(time.Hour).Add(-24 * 7 * time.Hour)
|
|
case "24h":
|
|
from = time.Now().Truncate(time.Hour).Add(-24 * time.Hour)
|
|
default:
|
|
return c.Status(400).SendString("Durations supported: 7d, 24h")
|
|
}
|
|
hourlyAverageResponseTime, err := store.Get().GetHourlyAverageResponseTimeByKey(c.Params("key"), from, time.Now())
|
|
if err != nil {
|
|
if err == common.ErrEndpointNotFound {
|
|
return c.Status(404).SendString(err.Error())
|
|
} else if err == common.ErrInvalidTimeRange {
|
|
return c.Status(400).SendString(err.Error())
|
|
}
|
|
return c.Status(500).SendString(err.Error())
|
|
}
|
|
if len(hourlyAverageResponseTime) == 0 {
|
|
return c.Status(204).SendString("")
|
|
}
|
|
series := chart.TimeSeries{
|
|
Name: "Average response time per hour",
|
|
Style: chart.Style{
|
|
StrokeWidth: 1.5,
|
|
DotWidth: 2.0,
|
|
},
|
|
}
|
|
keys := make([]int, 0, len(hourlyAverageResponseTime))
|
|
earliestTimestamp := int64(0)
|
|
for hourlyTimestamp := range hourlyAverageResponseTime {
|
|
keys = append(keys, int(hourlyTimestamp))
|
|
if earliestTimestamp == 0 || hourlyTimestamp < earliestTimestamp {
|
|
earliestTimestamp = hourlyTimestamp
|
|
}
|
|
}
|
|
for earliestTimestamp > from.Unix() {
|
|
earliestTimestamp -= int64(time.Hour.Seconds())
|
|
keys = append(keys, int(earliestTimestamp))
|
|
}
|
|
sort.Ints(keys)
|
|
var maxAverageResponseTime float64
|
|
for _, key := range keys {
|
|
averageResponseTime := float64(hourlyAverageResponseTime[int64(key)])
|
|
if maxAverageResponseTime < averageResponseTime {
|
|
maxAverageResponseTime = averageResponseTime
|
|
}
|
|
series.XValues = append(series.XValues, time.Unix(int64(key), 0))
|
|
series.YValues = append(series.YValues, averageResponseTime)
|
|
}
|
|
graph := chart.Chart{
|
|
Canvas: transparentStyle,
|
|
Background: transparentStyle,
|
|
Width: 1280,
|
|
Height: 300,
|
|
XAxis: chart.XAxis{
|
|
ValueFormatter: chart.TimeValueFormatterWithFormat(timeFormat),
|
|
GridMajorStyle: gridStyle,
|
|
GridMinorStyle: gridStyle,
|
|
Style: axisStyle,
|
|
NameStyle: axisStyle,
|
|
},
|
|
YAxis: chart.YAxis{
|
|
Name: "Average response time",
|
|
GridMajorStyle: gridStyle,
|
|
GridMinorStyle: gridStyle,
|
|
Style: axisStyle,
|
|
NameStyle: axisStyle,
|
|
Range: &chart.ContinuousRange{
|
|
Min: 0,
|
|
Max: math.Ceil(maxAverageResponseTime * 1.25),
|
|
},
|
|
},
|
|
Series: []chart.Series{series},
|
|
}
|
|
c.Set("Content-Type", "image/svg+xml")
|
|
c.Set("Cache-Control", "no-cache, no-store")
|
|
c.Set("Expires", "0")
|
|
c.Status(http.StatusOK)
|
|
if err := graph.Render(chart.SVG, c); err != nil {
|
|
log.Println("[api.ResponseTimeChart] Failed to render response time chart:", err.Error())
|
|
return c.Status(500).SendString(err.Error())
|
|
}
|
|
return nil
|
|
}
|