parent
5db0804f1d
commit
2fd259c162
@ -0,0 +1,2 @@
|
|||||||
|
bin/
|
||||||
|
.vscode/
|
||||||
@ -0,0 +1,14 @@
|
|||||||
|
all: bindir
|
||||||
|
go build -o ./bin ./cmd/...
|
||||||
|
|
||||||
|
httpcat: bindir
|
||||||
|
go build -o ./bin ./cmd/httpcat/...
|
||||||
|
|
||||||
|
mdfmt: bindir
|
||||||
|
go build -o ./bin ./cmd/md-fmt/...
|
||||||
|
|
||||||
|
otelq: bindir
|
||||||
|
go build -o ./bin ./cmd/otelq/...
|
||||||
|
|
||||||
|
bindir:
|
||||||
|
mkdir ./bin; exit 0
|
||||||
@ -0,0 +1,3 @@
|
|||||||
|
# Tempo To OTLP
|
||||||
|
|
||||||
|
Converts a Tempo protobuf message to an OTLP protobuf message
|
||||||
@ -0,0 +1,181 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/golang/protobuf/proto" // NOTE: keep this, it's required to unmarshall the tempopb
|
||||||
|
"github.com/grafana/tempo/pkg/tempopb"
|
||||||
|
tempocommonv1 "github.com/grafana/tempo/pkg/tempopb/common/v1"
|
||||||
|
tempotracev1 "github.com/grafana/tempo/pkg/tempopb/trace/v1"
|
||||||
|
commonv1 "go.opentelemetry.io/proto/otlp/common/v1"
|
||||||
|
resourcev1 "go.opentelemetry.io/proto/otlp/resource/v1"
|
||||||
|
tracev1 "go.opentelemetry.io/proto/otlp/trace/v1"
|
||||||
|
newproto "google.golang.org/protobuf/proto"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
file = flag.String("file", "", "Tempo proto file. if set, will not use stdin.")
|
||||||
|
out = flag.String("out", "", "Output file destination. Defaults to stdout.")
|
||||||
|
// asJSON = flag.Bool("json", false, "Read input as jsonpb.") // TODO(wperron) implement this
|
||||||
|
)
|
||||||
|
|
||||||
|
// TODO(wperron) implement reading from stdin
|
||||||
|
func main() {
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
var output io.Writer
|
||||||
|
if *out == "" {
|
||||||
|
output = os.Stdout
|
||||||
|
} else {
|
||||||
|
output = must(os.OpenFile(*out, os.O_CREATE+os.O_RDWR, 0o664))
|
||||||
|
}
|
||||||
|
|
||||||
|
h := must(os.Open(*file))
|
||||||
|
bs := must(io.ReadAll(h))
|
||||||
|
var trace tempopb.Trace
|
||||||
|
if err := proto.Unmarshal(bs, &trace); err != nil {
|
||||||
|
log.Fatalln(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
td := must(convert(trace))
|
||||||
|
bs = must(newproto.Marshal(&td))
|
||||||
|
written := must(io.Copy(output, bytes.NewReader(bs)))
|
||||||
|
|
||||||
|
fmt.Fprintf(os.Stderr, "completed successfully, %d bytes written", written)
|
||||||
|
}
|
||||||
|
|
||||||
|
func convert(trace tempopb.Trace) (tracev1.TracesData, error) {
|
||||||
|
td := make([]*tracev1.ResourceSpans, 0, len(trace.Batches))
|
||||||
|
for _, resourceSpans := range trace.Batches {
|
||||||
|
rs := &tracev1.ResourceSpans{
|
||||||
|
Resource: &resourcev1.Resource{
|
||||||
|
Attributes: convertAttributes(resourceSpans.Resource.Attributes),
|
||||||
|
DroppedAttributesCount: resourceSpans.Resource.DroppedAttributesCount,
|
||||||
|
},
|
||||||
|
ScopeSpans: make([]*tracev1.ScopeSpans, 0),
|
||||||
|
}
|
||||||
|
for _, ils := range resourceSpans.InstrumentationLibrarySpans {
|
||||||
|
scope := &tracev1.ScopeSpans{
|
||||||
|
Spans: make([]*tracev1.Span, 0),
|
||||||
|
}
|
||||||
|
|
||||||
|
scope.Scope = &commonv1.InstrumentationScope{
|
||||||
|
Name: ils.InstrumentationLibrary.Name,
|
||||||
|
Version: ils.InstrumentationLibrary.Version,
|
||||||
|
Attributes: convertAttributes(resourceSpans.Resource.Attributes), // TODO(wperron) is this right?
|
||||||
|
DroppedAttributesCount: resourceSpans.Resource.DroppedAttributesCount, // TODO(wperron) is this right?
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, span := range ils.Spans {
|
||||||
|
otelSpan := &tracev1.Span{
|
||||||
|
TraceId: span.TraceId,
|
||||||
|
SpanId: span.SpanId,
|
||||||
|
TraceState: span.TraceState,
|
||||||
|
ParentSpanId: span.ParentSpanId,
|
||||||
|
Name: span.Name,
|
||||||
|
Kind: tracev1.Span_SpanKind(span.Kind),
|
||||||
|
StartTimeUnixNano: span.StartTimeUnixNano,
|
||||||
|
EndTimeUnixNano: span.EndTimeUnixNano,
|
||||||
|
Attributes: convertAttributes(span.Attributes),
|
||||||
|
DroppedAttributesCount: span.DroppedAttributesCount,
|
||||||
|
Events: convertEvents(span.Events),
|
||||||
|
DroppedEventsCount: span.DroppedEventsCount,
|
||||||
|
Links: convertLinks(span.Links),
|
||||||
|
DroppedLinksCount: span.DroppedLinksCount,
|
||||||
|
Status: convertStatus(span.Status),
|
||||||
|
}
|
||||||
|
scope.Spans = append(scope.Spans, otelSpan)
|
||||||
|
}
|
||||||
|
|
||||||
|
rs.ScopeSpans = append(rs.ScopeSpans, scope)
|
||||||
|
}
|
||||||
|
td = append(td, rs)
|
||||||
|
}
|
||||||
|
|
||||||
|
return tracev1.TracesData{
|
||||||
|
ResourceSpans: td,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertAttributes(attrs []*tempocommonv1.KeyValue) []*commonv1.KeyValue {
|
||||||
|
kvs := make([]*commonv1.KeyValue, 0, len(attrs))
|
||||||
|
for _, a := range attrs {
|
||||||
|
kvs = append(kvs, &commonv1.KeyValue{
|
||||||
|
Key: a.Key,
|
||||||
|
Value: convertAnyValue(a.Value),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return kvs
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertAnyValue(av *tempocommonv1.AnyValue) *commonv1.AnyValue {
|
||||||
|
inner := av.GetValue()
|
||||||
|
v := &commonv1.AnyValue{}
|
||||||
|
switch inner.(type) {
|
||||||
|
case *tempocommonv1.AnyValue_StringValue:
|
||||||
|
v.Value = &commonv1.AnyValue_StringValue{StringValue: av.GetStringValue()}
|
||||||
|
case *tempocommonv1.AnyValue_IntValue:
|
||||||
|
v.Value = &commonv1.AnyValue_IntValue{IntValue: av.GetIntValue()}
|
||||||
|
case *tempocommonv1.AnyValue_DoubleValue:
|
||||||
|
v.Value = &commonv1.AnyValue_DoubleValue{DoubleValue: av.GetDoubleValue()}
|
||||||
|
case *tempocommonv1.AnyValue_BoolValue:
|
||||||
|
v.Value = &commonv1.AnyValue_BoolValue{BoolValue: av.GetBoolValue()}
|
||||||
|
case *tempocommonv1.AnyValue_ArrayValue:
|
||||||
|
inner := av.GetArrayValue().Values
|
||||||
|
i := make([]*commonv1.AnyValue, 0, len(inner))
|
||||||
|
for _, val := range inner {
|
||||||
|
i = append(i, convertAnyValue(val))
|
||||||
|
}
|
||||||
|
v.Value = &commonv1.AnyValue_ArrayValue{ArrayValue: &commonv1.ArrayValue{Values: i}}
|
||||||
|
case *tempocommonv1.AnyValue_KvlistValue:
|
||||||
|
inner := av.GetKvlistValue().Values
|
||||||
|
v.Value = &commonv1.AnyValue_KvlistValue{KvlistValue: &commonv1.KeyValueList{Values: convertAttributes(inner)}}
|
||||||
|
}
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertEvents(events []*tempotracev1.Span_Event) []*tracev1.Span_Event {
|
||||||
|
ev := make([]*tracev1.Span_Event, 0, len(events))
|
||||||
|
for _, event := range events {
|
||||||
|
ev = append(ev, &tracev1.Span_Event{
|
||||||
|
TimeUnixNano: event.TimeUnixNano,
|
||||||
|
Name: event.Name,
|
||||||
|
DroppedAttributesCount: event.DroppedAttributesCount,
|
||||||
|
Attributes: convertAttributes(event.Attributes),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return ev
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertLinks(links []*tempotracev1.Span_Link) []*tracev1.Span_Link {
|
||||||
|
ls := make([]*tracev1.Span_Link, 0, len(links))
|
||||||
|
for _, link := range links {
|
||||||
|
ls = append(ls, &tracev1.Span_Link{
|
||||||
|
TraceId: link.TraceId,
|
||||||
|
SpanId: link.SpanId,
|
||||||
|
TraceState: link.TraceState,
|
||||||
|
DroppedAttributesCount: link.DroppedAttributesCount,
|
||||||
|
Attributes: convertAttributes(link.Attributes),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return ls
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertStatus(status *tempotracev1.Status) *tracev1.Status {
|
||||||
|
return &tracev1.Status{
|
||||||
|
Message: status.Message,
|
||||||
|
Code: tracev1.Status_StatusCode(status.Code),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func must[T any](v T, err error) T {
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return v
|
||||||
|
}
|
||||||
Loading…
Reference in new issue