mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
7bf7308ea5
Removes request/response connection/hop headers for call resource in similar manner as Go's reverse proxy functions. Also removes Prometheus datasource custom call resource header manipulation in regards to hop-by-hop headers. Fixes #60076 Ref #58646 Co-authored-by: Will Browne <wbrowne@users.noreply.github.com>
88 lines
2.0 KiB
Go
88 lines
2.0 KiB
Go
package resource
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/tsdb/prometheus/client"
|
|
"github.com/grafana/grafana/pkg/tsdb/prometheus/utils"
|
|
"github.com/grafana/grafana/pkg/util/maputil"
|
|
)
|
|
|
|
type Resource struct {
|
|
promClient *client.Client
|
|
log log.Logger
|
|
}
|
|
|
|
func New(
|
|
httpClient *http.Client,
|
|
settings backend.DataSourceInstanceSettings,
|
|
plog log.Logger,
|
|
) (*Resource, error) {
|
|
jsonData, err := utils.GetJsonData(settings)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
httpMethod, _ := maputil.GetStringOptional(jsonData, "httpMethod")
|
|
|
|
return &Resource{
|
|
log: plog,
|
|
promClient: client.NewClient(httpClient, httpMethod, settings.URL),
|
|
}, nil
|
|
}
|
|
|
|
func (r *Resource) Execute(ctx context.Context, req *backend.CallResourceRequest) (*backend.CallResourceResponse, error) {
|
|
r.log.FromContext(ctx).Debug("Sending resource query", "URL", req.URL)
|
|
resp, err := r.promClient.QueryResource(ctx, req)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error querying resource: %v", err)
|
|
}
|
|
|
|
defer func() {
|
|
tmpErr := resp.Body.Close()
|
|
if tmpErr != nil && err == nil {
|
|
err = tmpErr
|
|
}
|
|
}()
|
|
|
|
var buf bytes.Buffer
|
|
// Should be more efficient than ReadAll. See https://github.com/prometheus/client_golang/pull/976
|
|
_, err = buf.ReadFrom(resp.Body)
|
|
body := buf.Bytes()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
callResponse := &backend.CallResourceResponse{
|
|
Status: resp.StatusCode,
|
|
Headers: resp.Header,
|
|
Body: body,
|
|
}
|
|
|
|
return callResponse, err
|
|
}
|
|
|
|
func (r *Resource) DetectVersion(ctx context.Context, req *backend.CallResourceRequest) (*backend.CallResourceResponse, error) {
|
|
newReq := &backend.CallResourceRequest{
|
|
PluginContext: req.PluginContext,
|
|
Path: "/api/v1/status/buildinfo",
|
|
}
|
|
|
|
resp, err := r.Execute(ctx, newReq)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
callResponse := &backend.CallResourceResponse{
|
|
Status: 200,
|
|
Body: resp.Body,
|
|
}
|
|
|
|
return callResponse, nil
|
|
}
|