mirror of https://github.com/grafana/grafana
Tempo: convert to backend data source (#31618)
* Tempo: Support opentelemetry response * Tempo: convert Tempo to backend data source * Update data source test * Fix lint issues * Apply suggestions from code review Co-authored-by: Arve Knudsen <arve.knudsen@gmail.com> * Return error when trace not found Co-authored-by: Arve Knudsen <arve.knudsen@gmail.com>pull/31698/head
Zoltán Bedi
4 years ago
committed by
GitHub
parent
57f3de74c6
commit
862f09376f
@ -0,0 +1,127 @@ |
||||
package tempo |
||||
|
||||
import ( |
||||
"context" |
||||
"encoding/json" |
||||
"fmt" |
||||
"io/ioutil" |
||||
"net/http" |
||||
|
||||
"github.com/grafana/grafana-plugin-sdk-go/data" |
||||
"github.com/grafana/grafana/pkg/infra/log" |
||||
"github.com/grafana/grafana/pkg/models" |
||||
"github.com/grafana/grafana/pkg/tsdb" |
||||
|
||||
jaeger "github.com/jaegertracing/jaeger/model" |
||||
jaeger_json "github.com/jaegertracing/jaeger/model/converter/json" |
||||
|
||||
ot_pdata "go.opentelemetry.io/collector/consumer/pdata" |
||||
ot_jaeger "go.opentelemetry.io/collector/translator/trace/jaeger" |
||||
) |
||||
|
||||
type tempoExecutor struct { |
||||
httpClient *http.Client |
||||
} |
||||
|
||||
func newTempoExecutor(dsInfo *models.DataSource) (tsdb.TsdbQueryEndpoint, error) { |
||||
httpClient, err := dsInfo.GetHttpClient() |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
return &tempoExecutor{ |
||||
httpClient: httpClient, |
||||
}, nil |
||||
} |
||||
|
||||
var ( |
||||
plog log.Logger |
||||
) |
||||
|
||||
func init() { |
||||
plog = log.New("tsdb.tempo") |
||||
tsdb.RegisterTsdbQueryEndpoint("tempo", newTempoExecutor) |
||||
} |
||||
|
||||
func (e *tempoExecutor) Query(ctx context.Context, dsInfo *models.DataSource, tsdbQuery *tsdb.TsdbQuery) (*tsdb.Response, error) { |
||||
result := &tsdb.Response{ |
||||
Results: map[string]*tsdb.QueryResult{}, |
||||
} |
||||
refID := tsdbQuery.Queries[0].RefId |
||||
queryResult := &tsdb.QueryResult{} |
||||
result.Results[refID] = queryResult |
||||
|
||||
traceID := tsdbQuery.Queries[0].Model.Get("query").MustString("") |
||||
|
||||
plog.Debug("Querying tempo with traceID", "traceID", traceID) |
||||
|
||||
req, err := http.NewRequestWithContext(ctx, "GET", dsInfo.Url+"/api/traces/"+traceID, nil) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
req.Header.Set("Accept", "application/protobuf") |
||||
|
||||
resp, err := e.httpClient.Do(req) |
||||
if err != nil { |
||||
return nil, fmt.Errorf("failed get to tempo: %w", err) |
||||
} |
||||
|
||||
defer func() { |
||||
if err := resp.Body.Close(); err != nil { |
||||
plog.Warn("failed to close response body", "err", err) |
||||
} |
||||
}() |
||||
|
||||
if resp.StatusCode == http.StatusNotFound { |
||||
queryResult.Error = fmt.Errorf("failed to get trace: %s", traceID) |
||||
return result, nil |
||||
} |
||||
|
||||
body, err := ioutil.ReadAll(resp.Body) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
otTrace := ot_pdata.NewTraces() |
||||
err = otTrace.FromOtlpProtoBytes(body) |
||||
if err != nil { |
||||
return nil, fmt.Errorf("failed to convert tempo response to Otlp: %w", err) |
||||
} |
||||
|
||||
jaegerBatches, err := ot_jaeger.InternalTracesToJaegerProto(otTrace) |
||||
if err != nil { |
||||
return nil, fmt.Errorf("failed to translate to jaegerBatches %v: %w", traceID, err) |
||||
} |
||||
|
||||
jaegerTrace := &jaeger.Trace{ |
||||
Spans: []*jaeger.Span{}, |
||||
ProcessMap: []jaeger.Trace_ProcessMapping{}, |
||||
} |
||||
|
||||
// otel proto conversion doesn't set jaeger processes
|
||||
for _, batch := range jaegerBatches { |
||||
for _, s := range batch.Spans { |
||||
s.Process = batch.Process |
||||
} |
||||
|
||||
jaegerTrace.Spans = append(jaegerTrace.Spans, batch.Spans...) |
||||
jaegerTrace.ProcessMap = append(jaegerTrace.ProcessMap, jaeger.Trace_ProcessMapping{ |
||||
Process: *batch.Process, |
||||
ProcessID: batch.Process.ServiceName, |
||||
}) |
||||
} |
||||
jsonTrace := jaeger_json.FromDomain(jaegerTrace) |
||||
|
||||
traceBytes, err := json.Marshal(jsonTrace) |
||||
if err != nil { |
||||
return nil, fmt.Errorf("failed to json.Marshal trace \"%s\" :%w", traceID, err) |
||||
} |
||||
|
||||
frames := []*data.Frame{ |
||||
{Name: "Traces", RefID: refID, Fields: []*data.Field{data.NewField("trace", nil, []string{string(traceBytes)})}}, |
||||
} |
||||
queryResult.Dataframes = tsdb.NewDecodedDataFrames(frames) |
||||
|
||||
return result, nil |
||||
} |
||||
Loading…
Reference in new issue