mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-29 02:16:28 +02:00
fcb33966e2
* config: add enable_google_cloud_serverless_authentication to config protobuf
* use dependency injection for embedded envoy provider
* Revert "use dependency injection for embedded envoy provider"
This reverts commit 5c08990501.
* config: attach envoy version to Config to avoid metrics depending on envoy/files
108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package config
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"sync"
|
|
|
|
"github.com/pomerium/pomerium/internal/log"
|
|
"github.com/pomerium/pomerium/internal/middleware"
|
|
"github.com/pomerium/pomerium/internal/telemetry"
|
|
"github.com/pomerium/pomerium/internal/telemetry/metrics"
|
|
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
// A MetricsManager manages metrics for a given configuration.
|
|
type MetricsManager struct {
|
|
mu sync.RWMutex
|
|
installationID string
|
|
serviceName string
|
|
addr string
|
|
basicAuth string
|
|
handler http.Handler
|
|
}
|
|
|
|
// NewMetricsManager creates a new MetricsManager.
|
|
func NewMetricsManager(ctx context.Context, src Source) *MetricsManager {
|
|
ctx = log.WithContext(ctx, func(c zerolog.Context) zerolog.Context {
|
|
return c.Str("service", "metrics_manager")
|
|
})
|
|
mgr := &MetricsManager{}
|
|
metrics.RegisterInfoMetrics()
|
|
src.OnConfigChange(ctx, mgr.OnConfigChange)
|
|
mgr.OnConfigChange(ctx, src.GetConfig())
|
|
return mgr
|
|
}
|
|
|
|
// Close closes any underlying http server.
|
|
func (mgr *MetricsManager) Close() error {
|
|
return nil
|
|
}
|
|
|
|
// OnConfigChange updates the metrics manager when configuration is changed.
|
|
func (mgr *MetricsManager) OnConfigChange(ctx context.Context, cfg *Config) {
|
|
mgr.mu.Lock()
|
|
defer mgr.mu.Unlock()
|
|
|
|
mgr.updateInfo(cfg)
|
|
mgr.updateServer(cfg)
|
|
}
|
|
|
|
func (mgr *MetricsManager) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
mgr.mu.RLock()
|
|
defer mgr.mu.RUnlock()
|
|
|
|
if mgr.handler == nil {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
mgr.handler.ServeHTTP(w, r)
|
|
}
|
|
|
|
func (mgr *MetricsManager) updateInfo(cfg *Config) {
|
|
serviceName := telemetry.ServiceName(cfg.Options.Services)
|
|
if serviceName == mgr.serviceName {
|
|
return
|
|
}
|
|
|
|
hostname, err := os.Hostname()
|
|
if err != nil {
|
|
log.Error(context.TODO()).Err(err).Msg("telemetry/metrics: failed to get OS hostname")
|
|
hostname = "__unknown__"
|
|
}
|
|
|
|
metrics.SetBuildInfo(serviceName, hostname, cfg.EnvoyVersion)
|
|
mgr.serviceName = serviceName
|
|
}
|
|
|
|
func (mgr *MetricsManager) updateServer(cfg *Config) {
|
|
if cfg.Options.MetricsAddr == mgr.addr &&
|
|
cfg.Options.MetricsBasicAuth == mgr.basicAuth &&
|
|
cfg.Options.InstallationID == mgr.installationID {
|
|
return
|
|
}
|
|
|
|
mgr.addr = cfg.Options.MetricsAddr
|
|
mgr.basicAuth = cfg.Options.MetricsBasicAuth
|
|
mgr.installationID = cfg.Options.InstallationID
|
|
mgr.handler = nil
|
|
|
|
if mgr.addr == "" {
|
|
log.Info(context.TODO()).Msg("metrics: http server disabled")
|
|
return
|
|
}
|
|
|
|
handler, err := metrics.PrometheusHandler(EnvoyAdminURL, mgr.installationID)
|
|
if err != nil {
|
|
log.Error(context.TODO()).Err(err).Msg("metrics: failed to create prometheus handler")
|
|
return
|
|
}
|
|
|
|
if username, password, ok := cfg.Options.GetMetricsBasicAuth(); ok {
|
|
handler = middleware.RequireBasicAuth(username, password)(handler)
|
|
}
|
|
|
|
mgr.handler = handler
|
|
}
|