// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0

package pref // import "go.opentelemetry.io/collector/pdata/xpdata/pref"

import (
	"reflect"

	"go.opentelemetry.io/collector/pdata/internal"
	pmetadata "go.opentelemetry.io/collector/pdata/internal/metadata"
	"go.opentelemetry.io/collector/pdata/ptrace"
	"go.opentelemetry.io/collector/pdata/xpdata/internal/metadata"
)

// MarkPipelineOwnedTraces marks the ptrace.Traces data as owned by the pipeline, returns true if the data were
// previously not owned by the pipeline, otherwise false.
func MarkPipelineOwnedTraces(td ptrace.Traces) bool {
	return internal.GetTracesState(internal.TracesWrapper(td)).MarkPipelineOwned()
}

func RefTraces(td ptrace.Traces) {
	internal.GetTracesState(internal.TracesWrapper(td)).Ref()
}

func UnrefTraces(td ptrace.Traces) {
	if metadata.PdataEnableRefCountingFeatureGate.IsEnabled() {
		if !internal.GetTracesState(internal.TracesWrapper(td)).Unref() {
			return
		}
		// Don't call DeleteExportLogsServiceRequest without the gate because we reset the data and that may still cause issues.
		if pmetadata.PdataUseProtoPoolingFeatureGate.IsEnabled() {
			internal.DeleteExportTraceServiceRequest(internal.GetTracesOrig(internal.TracesWrapper(td)), true)
		}
	}
}

// TODO: Generate this in pdata.

func EqualTraces(td1, td2 ptrace.Traces) bool {
	return reflect.DeepEqual(internal.GetTracesOrig(internal.TracesWrapper(td1)), internal.GetTracesOrig(internal.TracesWrapper(td2)))
}
