pomerium/internal/zero/telemetry/opencensus/opencensus.go

// Package opencensus is a provider of opencensus based telemetry metrics to the zero telemetry system.
package opencensus

import (
	"context"
	"sync/atomic"

	"github.com/hashicorp/go-set/v3"
	"go.opentelemetry.io/otel/bridge/opencensus"
	"go.opentelemetry.io/otel/sdk/metric"
	"go.opentelemetry.io/otel/sdk/metric/metricdata"
)

type Producer struct {
	producer metric.Producer
	filter   atomic.Pointer[set.Set[string]]
}

var _ metric.Producer = (*Producer)(nil)

func New() *Producer {
	p := &Producer{
		producer: opencensus.NewMetricProducer(),
	}
	p.SetFilter(nil)
	return p
}

func (p *Producer) Produce(ctx context.Context) ([]metricdata.ScopeMetrics, error) {
	filter := p.filter.Load()
	metrics, err := p.producer.Produce(ctx)
	if err != nil {
		return nil, err
	}
	var out []metricdata.ScopeMetrics
	for _, sm := range metrics {
		var m []metricdata.Metrics
		for _, metric := range sm.Metrics {
			if filter.Contains(metric.Name) {
				m = append(m, metric)
			}
		}
		if len(m) > 0 {
			out = append(out, metricdata.ScopeMetrics{
				Scope:   sm.Scope,
				Metrics: m,
			})
		}
	}
	return out, nil
}

func (p *Producer) SetFilter(names []string) {
	p.filter.Store(set.From(names))
}