mirror of
https://github.com/hanzoai/traces.git
synced 2026-11-03 04:05:45 +00:00
MarioandGitHub
e0354d1bed
* Update Prometheus to v0.307.2 * Update otel collector * Update OTel collector
36 lines
972 B
Go
36 lines
972 B
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package refconsumer // import "go.opentelemetry.io/collector/service/internal/refconsumer"
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.opentelemetry.io/collector/consumer"
|
|
"go.opentelemetry.io/collector/consumer/xconsumer"
|
|
"go.opentelemetry.io/collector/pdata/pprofile"
|
|
"go.opentelemetry.io/collector/pdata/xpdata/pref"
|
|
)
|
|
|
|
func NewProfiles(cons xconsumer.Profiles) xconsumer.Profiles {
|
|
return refProfiles{
|
|
consumer: cons,
|
|
}
|
|
}
|
|
|
|
type refProfiles struct {
|
|
consumer xconsumer.Profiles
|
|
}
|
|
|
|
// ConsumeProfiles measures telemetry before calling ConsumeProfiles because the data may be mutated downstream
|
|
func (c refProfiles) ConsumeProfiles(ctx context.Context, ld pprofile.Profiles) error {
|
|
if pref.MarkPipelineOwnedProfiles(ld) {
|
|
defer pref.UnrefProfiles(ld)
|
|
}
|
|
return c.consumer.ConsumeProfiles(ctx, ld)
|
|
}
|
|
|
|
func (c refProfiles) Capabilities() consumer.Capabilities {
|
|
return c.consumer.Capabilities()
|
|
}
|