mirror of
https://github.com/grafana/grafana.git
synced 2025-02-15 10:03:33 -06:00
9aca032d10
* consolidates alertmanager api errors * util & testing consistent errors * consistent errors for rest of ngalert apis * updates expected errors in testware * bump ci * linting * unrelated: dashboard.go lint
101 lines
3.1 KiB
Go
101 lines
3.1 KiB
Go
package api
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
"github.com/grafana/grafana/pkg/api/response"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/datasources"
|
|
apimodels "github.com/grafana/grafana/pkg/services/ngalert/api/tooling/definitions"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/eval"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/grafana/grafana/pkg/tsdb"
|
|
"github.com/grafana/grafana/pkg/util"
|
|
)
|
|
|
|
type TestingApiSrv struct {
|
|
*AlertingProxy
|
|
Cfg *setting.Cfg
|
|
DataService *tsdb.Service
|
|
DatasourceCache datasources.CacheService
|
|
log log.Logger
|
|
}
|
|
|
|
func (srv TestingApiSrv) RouteTestReceiverConfig(c *models.ReqContext, body apimodels.ExtendedReceiver) response.Response {
|
|
srv.log.Info("RouteTestReceiverConfig: ", "body", body)
|
|
return response.JSON(http.StatusOK, util.DynMap{"message": "success"})
|
|
}
|
|
|
|
func (srv TestingApiSrv) RouteTestRuleConfig(c *models.ReqContext, body apimodels.TestRulePayload) response.Response {
|
|
recipient := c.Params("Recipient")
|
|
if recipient == apimodels.GrafanaBackend.String() {
|
|
if body.Type() != apimodels.GrafanaBackend || body.GrafanaManagedCondition == nil {
|
|
return ErrResp(http.StatusBadRequest, errors.New("unexpected payload"), "")
|
|
}
|
|
return conditionEval(c, *body.GrafanaManagedCondition, srv.DatasourceCache, srv.DataService, srv.Cfg)
|
|
}
|
|
|
|
if body.Type() != apimodels.LoTexRulerBackend {
|
|
return ErrResp(http.StatusBadRequest, errors.New("unexpected payload"), "")
|
|
}
|
|
|
|
var path string
|
|
if datasourceID, err := strconv.ParseInt(recipient, 10, 64); err == nil {
|
|
ds, err := srv.DatasourceCache.GetDatasource(datasourceID, c.SignedInUser, c.SkipCache)
|
|
if err != nil {
|
|
return ErrResp(http.StatusInternalServerError, err, "failed to get datasource")
|
|
}
|
|
|
|
switch ds.Type {
|
|
case "loki":
|
|
path = "loki/api/v1/query"
|
|
case "prometheus":
|
|
path = "api/v1/query"
|
|
default:
|
|
return ErrResp(http.StatusBadRequest, fmt.Errorf("unexpected recipient type %s", ds.Type), "")
|
|
}
|
|
}
|
|
|
|
t := timeNow()
|
|
queryURL, err := url.Parse(path)
|
|
if err != nil {
|
|
return ErrResp(http.StatusInternalServerError, err, "failed to parse url")
|
|
}
|
|
params := queryURL.Query()
|
|
params.Set("query", body.Expr)
|
|
params.Set("time", strconv.FormatInt(t.Unix(), 10))
|
|
queryURL.RawQuery = params.Encode()
|
|
return srv.withReq(
|
|
c,
|
|
http.MethodGet,
|
|
queryURL,
|
|
nil,
|
|
instantQueryResultsExtractor,
|
|
nil,
|
|
)
|
|
}
|
|
|
|
func (srv TestingApiSrv) RouteEvalQueries(c *models.ReqContext, cmd apimodels.EvalQueriesPayload) response.Response {
|
|
now := cmd.Now
|
|
if now.IsZero() {
|
|
now = timeNow()
|
|
}
|
|
|
|
if _, err := validateQueriesAndExpressions(cmd.Data, c.SignedInUser, c.SkipCache, srv.DatasourceCache); err != nil {
|
|
return ErrResp(http.StatusBadRequest, err, "invalid queries or expressions")
|
|
}
|
|
|
|
evaluator := eval.Evaluator{Cfg: srv.Cfg}
|
|
evalResults, err := evaluator.QueriesAndExpressionsEval(c.SignedInUser.OrgId, cmd.Data, now, srv.DataService)
|
|
if err != nil {
|
|
return ErrResp(http.StatusBadRequest, err, "Failed to evaluate queries and expressions")
|
|
}
|
|
|
|
return response.JSONStreaming(http.StatusOK, evalResults)
|
|
}
|