trace_data.go

// Copyright 2024 tailsampling-go contributors
// Adapted from OpenTelemetry Collector Contrib
// Licensed under the Apache License, Version 2.0

package tailsampling

import (
	"container/list"

	"go.opentelemetry.io/otel/sdk/trace"
)

const MaxSpansPerTrace = 100

type traceData struct {
	spans        []trace.ReadOnlySpan
	spanCount    int
	queueElement *list.Element
}

func (td *traceData) addSpan(span trace.ReadOnlySpan) bool {
	if td.spanCount >= MaxSpansPerTrace {
		return false
	}
	td.spans = append(td.spans, span)
	td.spanCount++
	return true
}

func (td *traceData) getTraceData() *TraceData {
	return &TraceData{
		ReceivedBatches: td.spans,
		SpanCount:       td.spanCount,
	}
}

func (td *traceData) hasRootSpan() bool {
	for _, span := range td.spans {
		parent := span.Parent()
		if !parent.IsValid() || parent.IsRemote() {
			return true
		}
	}
	return false
}