3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-07-28 05:59:10 +02:00
Code Issues Projects Releases Packages Wiki Activity

merge main, wire with rest of funcs

Browse source
This commit is contained in:
Denis Mishin 2024-06-12 22:34:45 -04:00
parent 74c5bfbbd8
commit c945b08bcf
4 changed files with 83 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

50
internal/zero/controller/telemetry.go
View file

@ -3,35 +3,83 @@ package controller
import (
"context"
"fmt"
"net"
"net/url"
"time"
"github.com/rs/zerolog"
"go.opentelemetry.io/otel/sdk/instrumentation"
"github.com/pomerium/pomerium/internal/log"
"github.com/pomerium/pomerium/internal/telemetry/prometheus"
connect_mux "github.com/pomerium/pomerium/internal/zero/connect-mux"
"github.com/pomerium/pomerium/internal/zero/healthcheck"
"github.com/pomerium/pomerium/internal/zero/telemetry/reporter"
"github.com/pomerium/pomerium/internal/zero/telemetry/sessions"
"github.com/pomerium/pomerium/pkg/grpc/databroker"
"github.com/pomerium/pomerium/pkg/health"
"github.com/pomerium/pomerium/pkg/zero/connect"
)
const (
producerSessionAnalytics = "session-analytics"
producerEnvoy = "envoy"
)
func (c *controller) initTelemetry(ctx context.Context, clientProvider func() (databroker.DataBrokerServiceClient, error)) error {
sessionMetricProducer := sessions.NewProducer(instrumentation.Scope{}, clientProvider)
startTime := time.Now()
sessionMetricProducer := sessions.NewProducer(instrumentation.Scope{Name: "cluster"}, clientProvider)
envoyMetricProducer, err := prometheus.NewProducer(c.buildEnvoyMetricProducerOptions(nil, nil, startTime)...)
if err != nil {
return fmt.Errorf("error creating envoy metric producer: %w", err)
}
r, err := reporter.New(ctx, c.api.GetTelemetryConn(),
reporter.WithProducer(producerSessionAnalytics, sessionMetricProducer),
reporter.WithProducer(producerEnvoy, envoyMetricProducer),
)
if err != nil {
return fmt.Errorf("error creating telemetry metrics reporter: %w", err)
}
err = c.api.Watch(ctx, connect_mux.WithOnTelemetryRequested(func(ctx context.Context, req *connect.TelemetryRequest) {
sessionMetricProducer.SetEnabled(req.GetSessionAnalytics() != nil)
if envoyMetricRequest := req.GetEnvoyMetrics(); envoyMetricRequest != nil {
opts := c.buildEnvoyMetricProducerOptions(envoyMetricRequest.GetMetrics(), envoyMetricRequest.GetLabels(), startTime)
err := envoyMetricProducer.SetConfig(opts...)
if err != nil {
log.Warn(ctx).Err(err).Msg("failed to set envoy metric producer options")
}
} else {
_ = envoyMetricProducer.SetConfig(c.buildEnvoyMetricProducerOptions(nil, nil, startTime)...)
}
c.telemetryReporter.CollectAndExportMetrics(ctx)
}))
if err != nil {
return fmt.Errorf("watch telemetry: %w", err)
}
c.telemetryReporter = r
return nil
}
func (c *controller) buildEnvoyMetricProducerOptions(metrics, labels []string, startTime time.Time) []prometheus.ProducerOption {
return []prometheus.ProducerOption{
prometheus.WithIncludeMetrics(metrics...),
prometheus.WithIncludeLabels(labels...),
prometheus.WithScope(instrumentation.Scope{Name: "envoy"}),
prometheus.WithScrapeURL((&url.URL{
Scheme: "http",
Host: net.JoinHostPort("localhost", c.bootstrapConfig.GetConfig().OutboundPort),
Path: "/envoy/stats/prometheus",
}).String()),
prometheus.WithStartTime(startTime),
}
}
func (c *controller) shutdownTelemetry(ctx context.Context) {
ctx, cancel := context.WithTimeout(context.WithoutCancel(ctx), c.cfg.shutdownTimeout)
defer cancel()