mirror of
https://github.com/grafana/grafana.git
synced 2025-08-06 02:23:28 +08:00
93cdc94a94
* capitalise logs in observability logs * capitalise oss-bit-tent packages * capitalise logs in aws-datasources * capitalise logs for traces and profiling * capitalise logs for partner datasources * capitalise logs in plugins platform * capitalise logs for observability metrics
92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
package tempo
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/grafana/grafana-plugin-sdk-go/data"
|
|
"github.com/grafana/grafana/pkg/tsdb/tempo/kinds/dataquery"
|
|
"go.opentelemetry.io/collector/pdata/ptrace"
|
|
)
|
|
|
|
func (s *Service) getTrace(ctx context.Context, pCtx backend.PluginContext, query backend.DataQuery) (*backend.DataResponse, error) {
|
|
result := &backend.DataResponse{}
|
|
refID := query.RefID
|
|
|
|
model := &dataquery.TempoQuery{}
|
|
err := json.Unmarshal(query.JSON, model)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
|
|
dsInfo, err := s.getDSInfo(ctx, pCtx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
request, err := s.createRequest(ctx, dsInfo, model.Query, query.TimeRange.From.Unix(), query.TimeRange.To.Unix())
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
|
|
resp, err := dsInfo.HTTPClient.Do(request)
|
|
if err != nil {
|
|
return result, fmt.Errorf("failed get to tempo: %w", err)
|
|
}
|
|
|
|
defer func() {
|
|
if err := resp.Body.Close(); err != nil {
|
|
s.logger.FromContext(ctx).Warn("Failed to close response body", "err", err)
|
|
}
|
|
}()
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return &backend.DataResponse{}, err
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
result.Error = fmt.Errorf("failed to get trace with id: %s Status: %s Body: %s", model.Query, resp.Status, string(body))
|
|
return result, nil
|
|
}
|
|
|
|
pbUnmarshaler := ptrace.ProtoUnmarshaler{}
|
|
otTrace, err := pbUnmarshaler.UnmarshalTraces(body)
|
|
|
|
if err != nil {
|
|
return &backend.DataResponse{}, fmt.Errorf("failed to convert tempo response to Otlp: %w", err)
|
|
}
|
|
|
|
frame, err := TraceToFrame(otTrace)
|
|
if err != nil {
|
|
return &backend.DataResponse{}, fmt.Errorf("failed to transform trace %v to data frame: %w", model.Query, err)
|
|
}
|
|
frame.RefID = refID
|
|
frames := []*data.Frame{frame}
|
|
result.Frames = frames
|
|
return result, nil
|
|
}
|
|
|
|
func (s *Service) createRequest(ctx context.Context, dsInfo *Datasource, traceID string, start int64, end int64) (*http.Request, error) {
|
|
var tempoQuery string
|
|
if start == 0 || end == 0 {
|
|
tempoQuery = fmt.Sprintf("%s/api/traces/%s", dsInfo.URL, traceID)
|
|
} else {
|
|
tempoQuery = fmt.Sprintf("%s/api/traces/%s?start=%d&end=%d", dsInfo.URL, traceID, start, end)
|
|
}
|
|
|
|
req, err := http.NewRequestWithContext(ctx, "GET", tempoQuery, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
req.Header.Set("Accept", "application/protobuf")
|
|
|
|
s.logger.FromContext(ctx).Debug("Tempo request", "url", req.URL.String(), "headers", req.Header)
|
|
return req, nil
|
|
}
|