opentelemetry-collector/service/internal/obsconsumer/option.go

54 lines
1.8 KiB
Go

// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0
package obsconsumer // import "go.opentelemetry.io/collector/service/internal/obsconsumer"
import (
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/metric"
)
const (
ComponentOutcome = "otelcol.component.outcome"
)
// Option modifies the consumer behavior.
type Option func(*options)
type options struct {
staticDataPointAttributes []attribute.KeyValue
}
// WithStaticDataPointAttribute returns an Option that adds a static attribute to data points.
func WithStaticDataPointAttribute(attr attribute.KeyValue) Option {
return func(opts *options) {
opts.staticDataPointAttributes = append(opts.staticDataPointAttributes, attr)
}
}
type compiledOptions struct {
withSuccessAttrs metric.AddOption
withFailureAttrs metric.AddOption
withRefusedAttrs metric.AddOption
}
func (o *options) compile() compiledOptions {
successAttrs := make([]attribute.KeyValue, 0, 1+len(o.staticDataPointAttributes))
successAttrs = append(successAttrs, attribute.String(ComponentOutcome, "success"))
successAttrs = append(successAttrs, o.staticDataPointAttributes...)
failureAttrs := make([]attribute.KeyValue, 0, 1+len(o.staticDataPointAttributes))
failureAttrs = append(failureAttrs, attribute.String(ComponentOutcome, "failure"))
failureAttrs = append(failureAttrs, o.staticDataPointAttributes...)
refusedAttrs := make([]attribute.KeyValue, 0, 1+len(o.staticDataPointAttributes))
refusedAttrs = append(refusedAttrs, attribute.String(ComponentOutcome, "refused"))
refusedAttrs = append(refusedAttrs, o.staticDataPointAttributes...)
return compiledOptions{
withSuccessAttrs: metric.WithAttributes(successAttrs...),
withFailureAttrs: metric.WithAttributes(failureAttrs...),
withRefusedAttrs: metric.WithAttributes(refusedAttrs...),
}
}