IntenseWebs/grafana
3
0
Fork 0
mirror of https://github.com/grafana/grafana.git synced 2025-02-25 18:55:37 -06:00
Code Issues Packages Projects Releases Wiki Activity

Elasticsearch: Replace error source http client with a new error source methods (#96570)

Browse Source 
* Elasticsearch: Replace error source http client with methods

* Update

* Update

* Update tests

* Update

* Update

* Update pkg/tsdb/elasticsearch/data_query.go

Co-authored-by: Nathan Vērzemnieks <njvrzm@gmail.com>

* Trigger build

---------

Co-authored-by: Nathan Vērzemnieks <njvrzm@gmail.com>
This commit is contained in:
Ivana Huckova 2024-11-18 14:39:21 +01:00 committed by GitHub
parent 1366197522
commit d00c4fd35a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/tsdb/elasticsearch/data_query.go
View File

@ -77,10 +77,17 @@ func (e *elasticsearchDataQuery) execute() (*backend.QueryDataResponse, error) {
e.logger.Info("Prepared request", "queriesLength", len(queries), "duration", time.Since(start), "stage", es.StagePrepareRequest)
res, err := e.client.ExecuteMultisearch(req)
if err != nil {
// We are returning error containing the source that was added through errorsource.Middleware
if backend.IsDownstreamHTTPError(err) {
err = errorsource.DownstreamError(err, false)
}
return errorsource.AddErrorToResponse(e.dataQueries[0].RefID, response, err), nil
}
if res.Status >= 400 {
errWithSource := errorsource.SourceError(backend.ErrorSourceFromHTTPStatus(res.Status), fmt.Errorf("unexpected status code: %d", res.Status), false)
return errorsource.AddErrorToResponse(e.dataQueries[0].RefID, response, errWithSource), nil
}
return parseResponse(e.ctx, res.Responses, queries, e.client.GetConfiguredFields(), e.keepLabelsInResponse, e.logger)
}

6
pkg/tsdb/elasticsearch/elasticsearch.go
View File

@ -20,7 +20,6 @@ import (
"github.com/grafana/grafana-plugin-sdk-go/backend/instancemgmt"
"github.com/grafana/grafana-plugin-sdk-go/backend/log"
exp "github.com/grafana/grafana-plugin-sdk-go/experimental/errorsource"
exphttpclient "github.com/grafana/grafana-plugin-sdk-go/experimental/errorsource/httpclient"
es "github.com/grafana/grafana/pkg/tsdb/elasticsearch/client"
)
@ -90,10 +89,7 @@ func newInstanceSettings(httpClientProvider *httpclient.Provider) datasource.Ins
httpCliOpts.SigV4.Service = "es"
}
// set the default middlewars from the httpClientProvider
httpCliOpts.Middlewares = httpClientProvider.Opts.Middlewares
// enable experimental http client to support errors with source
httpCli, err := exphttpclient.New(httpCliOpts)
httpCli, err := httpClientProvider.New(httpCliOpts)
if err != nil {
return nil, err
}

6
pkg/tsdb/elasticsearch/error_handling_test.go
View File

@ -4,6 +4,7 @@ import (
"errors"
"testing"
"github.com/grafana/grafana-plugin-sdk-go/backend"
"github.com/grafana/grafana-plugin-sdk-go/experimental/errorsource"
"github.com/stretchr/testify/require"
)
@ -43,7 +44,8 @@ func TestErrorAvgMissingField(t *testing.T) {
require.NoError(t, err)
// FIXME: we should return the received error message
require.Len(t, result.response.Responses, 0)
require.Equal(t, "unexpected status code: 400", result.response.Responses["A"].Error.Error())
require.Equal(t, backend.ErrorSourceDownstream, result.response.Responses["A"].ErrorSource)
}
func TestErrorAvgMissingFieldNoDetailedErrors(t *testing.T) {
@ -71,7 +73,7 @@ func TestErrorAvgMissingFieldNoDetailedErrors(t *testing.T) {
require.NoError(t, err)
// FIXME: we should return the received error message
require.Len(t, result.response.Responses, 0)
require.Equal(t, "unexpected status code: 400", result.response.Responses["A"].Error.Error())
}
func TestErrorTooManyDateHistogramBuckets(t *testing.T) {