mirror of https://github.com/grafana/grafana.git
99 lines
3.1 KiB
Go
99 lines
3.1 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
"github.com/grafana/grafana/pkg/api/response"
|
|
"github.com/grafana/grafana/pkg/expr"
|
|
"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/services/secrets"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/grafana/grafana/pkg/web"
|
|
)
|
|
|
|
type TestingApiSrv struct {
|
|
*AlertingProxy
|
|
Cfg *setting.Cfg
|
|
ExpressionService *expr.Service
|
|
DatasourceCache datasources.CacheService
|
|
log log.Logger
|
|
secretsService secrets.Service
|
|
}
|
|
|
|
func (srv TestingApiSrv) RouteTestRuleConfig(c *models.ReqContext, body apimodels.TestRulePayload) response.Response {
|
|
recipient := web.Params(c.Req)[":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.ExpressionService, srv.secretsService, srv.Cfg, srv.log)
|
|
}
|
|
|
|
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(context.Background(), 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(c.Req.Context(), cmd.Data, c.SignedInUser, c.SkipCache, srv.DatasourceCache); err != nil {
|
|
return ErrResp(http.StatusBadRequest, err, "invalid queries or expressions")
|
|
}
|
|
|
|
evaluator := eval.NewEvaluator(srv.Cfg, srv.log, srv.DatasourceCache, srv.secretsService)
|
|
evalResults, err := evaluator.QueriesAndExpressionsEval(c.SignedInUser.OrgId, cmd.Data, now, srv.ExpressionService)
|
|
if err != nil {
|
|
return ErrResp(http.StatusBadRequest, err, "Failed to evaluate queries and expressions")
|
|
}
|
|
|
|
return response.JSONStreaming(http.StatusOK, evalResults)
|
|
}
|