receiver/elasticapmreceiver/internal/mappers/intakeV2ToOtlpTopLevelFields.go

// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

package mappers // import "github.com/elastic/opentelemetry-collector-components/receiver/elasticapmreceiver/internal/mappers"

import (
	"strings"
	"time"

	"github.com/elastic/apm-data/model/modelpb"
	"go.opentelemetry.io/collector/pdata/pcommon"
	"go.opentelemetry.io/collector/pdata/plog"
	"go.opentelemetry.io/collector/pdata/ptrace"
	"go.uber.org/zap"
)

type TopLevelFieldSetter interface {
	SetTraceID(v pcommon.TraceID)
	SetSpanID(v pcommon.SpanID)
}

// Shared across LogRecord and Span
func SetTopLevelFieldsCommon(event *modelpb.APMEvent, t TopLevelFieldSetter, logger *zap.Logger) {

	if event.Trace != nil && event.Trace.Id != "" {
		traceId, err := TraceIDFromHex(event.Trace.Id)
		if err == nil {
			t.SetTraceID(traceId)
		} else {
			logger.Error("failed to parse trace ID", zap.String("trace_id", event.Trace.Id))
		}
	}

	if event.Span != nil && event.Span.Id != "" {
		spanId, err := SpanIdFromHex(event.Span.Id)
		if err == nil {
			t.SetSpanID(spanId)
		} else {
			logger.Error("failed to parse span ID", zap.String("span_id", event.Span.Id))
		}
	}

	if event.Transaction != nil && event.Transaction.Id != "" {
		transactionId, err := SpanIdFromHex(event.Transaction.Id)
		if err != nil {
			t.SetSpanID(transactionId)
		} else {
			logger.Error("failed to parse transaction ID", zap.String("transaction_id", (event.Transaction.Id)))
		}
	}
}

// Sets top level fields on ptrace.Span based on the APMEvent
func SetTopLevelFieldsSpan(event *modelpb.APMEvent, timestamp time.Time, s ptrace.Span, logger *zap.Logger) {
	SetTopLevelFieldsCommon(event, s, logger)

	if event.ParentId != "" {
		parentId, err := SpanIdFromHex(event.ParentId)
		if err == nil {
			s.SetParentSpanID(parentId)
		} else {
			logger.Error("failed to parse parent span ID", zap.String("parent_id", event.ParentId))
		}
	}

	if strings.EqualFold(event.Event.Outcome, "success") {
		s.Status().SetCode(ptrace.StatusCodeOk)
	} else if strings.EqualFold(event.Event.Outcome, "failure") {
		s.Status().SetCode(ptrace.StatusCodeError)
	}

	duration := time.Duration(event.GetEvent().GetDuration())
	s.SetStartTimestamp(pcommon.NewTimestampFromTime(timestamp))
	s.SetEndTimestamp(pcommon.NewTimestampFromTime(timestamp.Add(duration)))
}

// Sets top level fields on plog.LogRecord based on the APMEvent
func SetTopLevelFieldsLogRecord(event *modelpb.APMEvent, timestamp time.Time, l plog.LogRecord, logger *zap.Logger) {
	SetTopLevelFieldsCommon(event, l, logger)
	l.SetTimestamp(pcommon.NewTimestampFromTime(timestamp))
}