mirror of
https://github.com/hanzoai/telemetry.git
synced 2026-10-30 20:01:49 +00:00
Bogdan DrutuandGitHub
3203167e9a
This PR also solves one more issues with the new interface, which is the ability to record multiple observations for a single async metric using one callback. Signed-off-by: Bogdan Drutu <[email protected]>
64 lines
1.9 KiB
Go
64 lines
1.9 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package batchprocessor // import "go.opentelemetry.io/collector/processor/batchprocessor"
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/metric"
|
|
|
|
"go.opentelemetry.io/collector/processor"
|
|
"go.opentelemetry.io/collector/processor/batchprocessor/internal/metadata"
|
|
"go.opentelemetry.io/collector/processor/internal"
|
|
)
|
|
|
|
type trigger int
|
|
|
|
const (
|
|
triggerTimeout trigger = iota
|
|
triggerBatchSize
|
|
)
|
|
|
|
type batchProcessorTelemetry struct {
|
|
exportCtx context.Context
|
|
|
|
processorAttr metric.MeasurementOption
|
|
telemetryBuilder *metadata.TelemetryBuilder
|
|
}
|
|
|
|
func newBatchProcessorTelemetry(set processor.Settings, currentMetadataCardinality func() int) (*batchProcessorTelemetry, error) {
|
|
attrs := metric.WithAttributeSet(attribute.NewSet(attribute.String(internal.ProcessorKey, set.ID.String())))
|
|
|
|
telemetryBuilder, err := metadata.NewTelemetryBuilder(set.TelemetrySettings)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = telemetryBuilder.RegisterProcessorBatchMetadataCardinalityCallback(func(_ context.Context, observer metric.Int64Observer) error {
|
|
observer.Observe(int64(currentMetadataCardinality()), attrs)
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &batchProcessorTelemetry{
|
|
exportCtx: context.Background(),
|
|
telemetryBuilder: telemetryBuilder,
|
|
processorAttr: attrs,
|
|
}, nil
|
|
}
|
|
|
|
func (bpt *batchProcessorTelemetry) record(trigger trigger, sent, bytes int64) {
|
|
switch trigger {
|
|
case triggerBatchSize:
|
|
bpt.telemetryBuilder.ProcessorBatchBatchSizeTriggerSend.Add(bpt.exportCtx, 1, bpt.processorAttr)
|
|
case triggerTimeout:
|
|
bpt.telemetryBuilder.ProcessorBatchTimeoutTriggerSend.Add(bpt.exportCtx, 1, bpt.processorAttr)
|
|
}
|
|
|
|
bpt.telemetryBuilder.ProcessorBatchBatchSendSize.Record(bpt.exportCtx, sent, bpt.processorAttr)
|
|
bpt.telemetryBuilder.ProcessorBatchBatchSendSizeBytes.Record(bpt.exportCtx, bytes, bpt.processorAttr)
|
|
}
|