otel-collector/pkg/metering/meter.go

package metering

import (
	"github.com/hanzoai/otel-collector/pkg/schema/traces"
	"go.opentelemetry.io/collector/pdata/pcommon"
	"go.opentelemetry.io/collector/pdata/plog"
	"go.opentelemetry.io/collector/pdata/pmetric"
	"go.opentelemetry.io/collector/pdata/ptrace"
)

// Meter is an interface that receives telemetry data and
// calculates billable metrics.
type Meter[T ptrace.Traces | pmetric.Metrics | plog.Logs, V ptrace.ResourceSpans | pmetric.ResourceMetrics | plog.ResourceLogs] interface {
	// Size calculates size of the telemetry data in bytes.
	Size(T) int
	SizePerResource(V) int
	// Count calculates count of the telemetry data.
	Count(T) int
	CountPerResource(V) int
}

// Sizer is an interface that calculates the size of different of map[string]any
type Sizer interface {
	SizeOfMapStringAny(map[string]any) int
	SizeOfFlatPcommonMapInMapStringString(pcommon.Map) int
	SizeOfInt(int) int
	SizeOfFloat64(float64) int
	SizeOfTraceID(pcommon.TraceID) int
	SizeOfSpanID(pcommon.SpanID) int
	SizeOfFlatPcommonMapInNumberStringBool(pcommon.Map) (int, int, int)
	SizeOfEvents([]string) int
	SizeOfOtelSpanRefs([]traces.OtelSpanRef) int
	TotalSizeIfKeyExists(int, int, int) int
	TotalSizeIfKeyExistsAndValueIsMapOrSlice(int, int, int) int
}

// Logs calculates billable metrics for logs.
type Logs interface {
	Meter[plog.Logs, plog.ResourceLogs]
}

// Traces calculates billable metrics for traces.
type Traces interface {
	Meter[ptrace.Traces, ptrace.ResourceSpans]
}

// Metrics calculates billable metrics for metrics.
type Metrics interface {
	Meter[pmetric.Metrics, pmetric.ResourceMetrics]
}