mirror of
https://github.com/prymitive/karma
synced 2026-05-05 03:16:51 +00:00
129ffb7e17
httptest.newRequest is suppose to be used for testing request handing, http.newRequest is for preparing outgoing requests, this allows to fix cache key selection and increase test coverage
94 lines
2.0 KiB
Go
94 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"sort"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/prymitive/unsee/internal/alertmanager"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// knownLabelNames allows querying known label names
|
|
func knownLabelNames(c *gin.Context) {
|
|
noCache(c)
|
|
start := time.Now()
|
|
|
|
cacheKey := c.Request.RequestURI
|
|
|
|
data, found := apiCache.Get(cacheKey)
|
|
if found {
|
|
c.Data(http.StatusOK, gin.MIMEJSON, data.([]byte))
|
|
logAlertsView(c, "HIT", time.Since(start))
|
|
return
|
|
}
|
|
|
|
labels := alertmanager.DedupKnownLabels()
|
|
acData := []string{}
|
|
|
|
term, found := c.GetQuery("term")
|
|
if !found || term == "" {
|
|
// return everything
|
|
sort.Strings(labels)
|
|
acData = labels
|
|
} else {
|
|
// return what matches
|
|
for _, key := range labels {
|
|
if strings.Contains(key, term) {
|
|
acData = append(acData, key)
|
|
}
|
|
}
|
|
sort.Strings(acData)
|
|
}
|
|
|
|
data, err := json.Marshal(acData)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
panic(err)
|
|
}
|
|
|
|
apiCache.Set(cacheKey, data, time.Second*15)
|
|
|
|
c.Data(http.StatusOK, gin.MIMEJSON, data.([]byte))
|
|
logAlertsView(c, "MIS", time.Since(start))
|
|
}
|
|
|
|
func knownLabelValues(c *gin.Context) {
|
|
noCache(c)
|
|
start := time.Now()
|
|
|
|
cacheKey := c.Request.RequestURI
|
|
|
|
data, found := apiCache.Get(cacheKey)
|
|
if found {
|
|
c.Data(http.StatusOK, gin.MIMEJSON, data.([]byte))
|
|
logAlertsView(c, "HIT", time.Since(start))
|
|
return
|
|
}
|
|
|
|
name, found := c.GetQuery("name")
|
|
if !found || name == "" {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "missing name=<token> parameter"})
|
|
log.Infof("[%s] <%d> %s %s took %s", c.ClientIP(), http.StatusBadRequest, c.Request.Method, c.Request.RequestURI, time.Since(start))
|
|
return
|
|
}
|
|
|
|
values := alertmanager.DedupKnownLabelValues(name)
|
|
sort.Strings(values)
|
|
|
|
data, err := json.Marshal(values)
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
panic(err)
|
|
}
|
|
|
|
apiCache.Set(cacheKey, data, time.Second*15)
|
|
|
|
c.Data(http.StatusOK, gin.MIMEJSON, data.([]byte))
|
|
logAlertsView(c, "MIS", time.Since(start))
|
|
}
|