mirror of
https://github.com/grafana/grafana.git
synced 2024-11-25 18:30:41 -06:00
79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package searchV2
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"io"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/grafana/grafana-plugin-sdk-go/data"
|
|
|
|
"github.com/grafana/grafana/pkg/api/response"
|
|
"github.com/grafana/grafana/pkg/api/routing"
|
|
"github.com/grafana/grafana/pkg/middleware"
|
|
contextmodel "github.com/grafana/grafana/pkg/services/contexthandler/model"
|
|
)
|
|
|
|
type SearchHTTPService interface {
|
|
RegisterHTTPRoutes(storageRoute routing.RouteRegister)
|
|
}
|
|
|
|
type searchHTTPService struct {
|
|
search SearchService
|
|
}
|
|
|
|
func ProvideSearchHTTPService(search SearchService) SearchHTTPService {
|
|
return &searchHTTPService{search: search}
|
|
}
|
|
|
|
func (s *searchHTTPService) RegisterHTTPRoutes(storageRoute routing.RouteRegister) {
|
|
storageRoute.Post("/", middleware.ReqSignedIn, routing.Wrap(s.doQuery))
|
|
}
|
|
|
|
func (s *searchHTTPService) doQuery(c *contextmodel.ReqContext) response.Response {
|
|
searchReadinessCheckResp := s.search.IsReady(c.Req.Context(), c.SignedInUser.GetOrgID())
|
|
if !searchReadinessCheckResp.IsReady {
|
|
dashboardSearchNotServedRequestsCounter.With(prometheus.Labels{
|
|
"reason": searchReadinessCheckResp.Reason,
|
|
}).Inc()
|
|
|
|
return response.JSON(200, &backend.DataResponse{
|
|
Frames: []*data.Frame{{
|
|
Name: "Loading",
|
|
}},
|
|
Error: nil,
|
|
})
|
|
}
|
|
|
|
body, err := io.ReadAll(c.Req.Body)
|
|
if err != nil {
|
|
return response.Error(500, "error reading bytes", err)
|
|
}
|
|
|
|
query := &DashboardQuery{}
|
|
err = json.Unmarshal(body, query)
|
|
if err != nil {
|
|
return response.Error(400, "error parsing body", err)
|
|
}
|
|
|
|
resp := s.search.doDashboardQuery(c.Req.Context(), c.SignedInUser, c.SignedInUser.GetOrgID(), *query)
|
|
|
|
if resp.Error != nil {
|
|
return response.Error(500, "error handling search request", resp.Error)
|
|
}
|
|
|
|
if len(resp.Frames) == 0 {
|
|
msg := "invalid search response"
|
|
return response.Error(500, msg, errors.New(msg))
|
|
}
|
|
|
|
bytes, err := resp.MarshalJSON()
|
|
if err != nil {
|
|
return response.Error(500, "error marshalling response", err)
|
|
}
|
|
|
|
return response.JSON(200, bytes)
|
|
}
|