Skip to content

Commit

Permalink
fix breaks from API change
Browse files Browse the repository at this point in the history
Signed-off-by: Yuri Shkuro <github@ysh.us>
  • Loading branch information
yurishkuro committed Oct 23, 2024
1 parent 58fcd44 commit 313e1be
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 33 deletions.
10 changes: 2 additions & 8 deletions cmd/collector/app/handler/otlp_receiver.go
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,6 @@ import (

"github.com/jaegertracing/jaeger/cmd/collector/app/flags"
"github.com/jaegertracing/jaeger/cmd/collector/app/processor"
"github.com/jaegertracing/jaeger/model"
"github.com/jaegertracing/jaeger/pkg/config/tlscfg"
"github.com/jaegertracing/jaeger/pkg/tenancy"
)
Expand Down Expand Up @@ -157,20 +156,15 @@ func newConsumerDelegate(logger *zap.Logger, spanProcessor processor.SpanProcess
processor.UnknownTransport, // could be gRPC or HTTP
processor.OTLPSpanFormat,
tm),
protoFromTraces: otlp2jaeger.ProtoFromTraces,
}
}

type consumerDelegate struct {
batchConsumer batchConsumer
protoFromTraces func(td ptrace.Traces) ([]*model.Batch, error)
batchConsumer batchConsumer
}

func (c *consumerDelegate) consume(ctx context.Context, td ptrace.Traces) error {
batches, err := c.protoFromTraces(td)
if err != nil {
return err
}
batches := otlp2jaeger.ProtoFromTraces(td)
for _, batch := range batches {
err := c.batchConsumer.consume(ctx, batch)
if err != nil {
Expand Down
12 changes: 0 additions & 12 deletions cmd/collector/app/handler/otlp_receiver_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,6 @@ import (
"go.opentelemetry.io/collector/receiver/otlpreceiver"

"github.com/jaegertracing/jaeger/cmd/collector/app/flags"
"github.com/jaegertracing/jaeger/model"
"github.com/jaegertracing/jaeger/pkg/config/corscfg"
"github.com/jaegertracing/jaeger/pkg/config/tlscfg"
"github.com/jaegertracing/jaeger/pkg/tenancy"
Expand Down Expand Up @@ -114,17 +113,6 @@ func TestStartOtlpReceiver_Error(t *testing.T) {
assert.Contains(t, err.Error(), "could not create the OTLP receiver")
}

func TestProtoFromTracesError(t *testing.T) {
mockErr := errors.New("mock error")
c := &consumerDelegate{
protoFromTraces: func(_ ptrace.Traces) ([]*model.Batch, error) {
return nil, mockErr
},
}
err := c.consume(context.Background(), ptrace.Traces{})
assert.Equal(t, mockErr, err)
}

func TestOtelHost_ReportFatalError(t *testing.T) {
logger, buf := testutils.NewLogger()
host := &otelHost{logger: logger}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@ package factoryadapter
import (
"context"
"errors"
"fmt"

otlp2jaeger "github.com/open-telemetry/opentelemetry-collector-contrib/pkg/translator/jaeger"
"go.opentelemetry.io/collector/pdata/ptrace"
Expand All @@ -27,10 +26,7 @@ func NewTraceWriter(spanWriter spanstore_v1.Writer) spanstore.Writer {

// WriteTraces implements spanstore.Writer.
func (t *TraceWriter) WriteTraces(ctx context.Context, td ptrace.Traces) error {
batches, err := otlp2jaeger.ProtoFromTraces(td)
if err != nil {
return fmt.Errorf("cannot transform OTLP traces to Jaeger format: %w", err)
}
batches := otlp2jaeger.ProtoFromTraces(td)
var errs []error
for _, batch := range batches {
for _, span := range batch.Spans {
Expand Down
6 changes: 1 addition & 5 deletions cmd/jaeger/internal/processors/adaptivesampling/processor.go
Original file line number Diff line number Diff line change
Expand Up @@ -65,11 +65,7 @@ func (tp *traceProcessor) close(context.Context) error {
}

func (tp *traceProcessor) processTraces(_ context.Context, td ptrace.Traces) (ptrace.Traces, error) {
batches, err := otlp2jaeger.ProtoFromTraces(td)
if err != nil {
return td, fmt.Errorf("cannot transform OTLP traces to Jaeger format: %w", err)
}

batches := otlp2jaeger.ProtoFromTraces(td)
for _, batch := range batches {
for _, span := range batch.Spans {
if span.Process == nil {
Expand Down
4 changes: 1 addition & 3 deletions cmd/query/app/otlp_translator.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,7 @@ func otlp2traces(otlpSpans []byte) ([]*model.Trace, error) {
if err != nil {
return nil, fmt.Errorf("cannot unmarshal OTLP : %w", err)
}
jaegerBatches, _ := model2otel.ProtoFromTraces(otlpTraces)
// ProtoFromTraces will not give an error

jaegerBatches := model2otel.ProtoFromTraces(otlpTraces)
var traces []*model.Trace
traceMap := make(map[model.TraceID]*model.Trace)
for _, batch := range jaegerBatches {
Expand Down

0 comments on commit 313e1be

Please sign in to comment.