mirror of
https://github.com/grafana/grafana.git
synced 2025-02-15 01:53:33 -06:00
e7ff04a167
* [Alerting]: Implement test rule API route * Apply suggestions from code review * Call /query instead of /query_range
80 lines
2.3 KiB
Go
80 lines
2.3 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
apimodels "github.com/grafana/alerting-api/pkg/api"
|
|
"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"
|
|
"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 response.Error(http.StatusBadRequest, "unexpected payload", nil)
|
|
}
|
|
return conditionEval(c, *body.GrafanaManagedCondition, srv.DatasourceCache, srv.DataService, srv.Cfg)
|
|
}
|
|
|
|
if body.Type() != apimodels.LoTexRulerBackend {
|
|
return response.Error(http.StatusBadRequest, "unexpected payload", nil)
|
|
}
|
|
|
|
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 response.Error(http.StatusInternalServerError, "failed to get datasource", err)
|
|
}
|
|
|
|
switch ds.Type {
|
|
case "loki":
|
|
path = "loki/api/v1/query"
|
|
case "prometheus":
|
|
path = "api/v1/query"
|
|
default:
|
|
return response.Error(http.StatusBadRequest, fmt.Sprintf("unexpected recipient type %s", ds.Type), nil)
|
|
}
|
|
}
|
|
|
|
t := timeNow()
|
|
queryURL, err := url.Parse(path)
|
|
if err != nil {
|
|
return response.Error(http.StatusInternalServerError, "failed to parse url", err)
|
|
}
|
|
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,
|
|
jsonExtractor(nil),
|
|
nil,
|
|
)
|
|
}
|