mirror of
https://github.com/portainer/portainer.git
synced 2026-06-23 06:20:12 +00:00
RHCowan
43 lines
1.3 KiB
Go
43 lines
1.3 KiB
Go
package libprometheus
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
prometheusreg "github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/common/model"
|
|
"github.com/prometheus/prometheus/config"
|
|
"github.com/prometheus/prometheus/discovery"
|
|
"github.com/prometheus/prometheus/discovery/targetgroup"
|
|
"github.com/prometheus/prometheus/notifier"
|
|
)
|
|
|
|
// NewNotifierManagerFromConfig creates a notifier.Manager from a loaded
|
|
// Prometheus config. It also returns a pre-populated target-sets channel ready
|
|
// to pass to Manager.Run() using the config's static alertmanager targets.
|
|
func NewNotifierManagerFromConfig(
|
|
cfg *config.Config,
|
|
reg prometheusreg.Registerer,
|
|
) (*notifier.Manager, chan map[string][]*targetgroup.Group, error) {
|
|
mgr := notifier.NewManager(
|
|
¬ifier.Options{
|
|
QueueCapacity: 10000,
|
|
DrainOnShutdown: true,
|
|
Registerer: reg,
|
|
},
|
|
model.LegacyValidation,
|
|
NewZerologSlogger(),
|
|
)
|
|
|
|
if err := mgr.ApplyConfig(cfg); err != nil {
|
|
return nil, nil, fmt.Errorf("apply prometheus notifier config: %w", err)
|
|
}
|
|
|
|
alertmanagerConfigs := cfg.AlertingConfig.AlertmanagerConfigs.ToMap()
|
|
serviceDiscoveryConfigs := make(map[string]discovery.Configs, len(alertmanagerConfigs))
|
|
for key, amCfg := range alertmanagerConfigs {
|
|
serviceDiscoveryConfigs[key] = amCfg.ServiceDiscoveryConfigs
|
|
}
|
|
|
|
return mgr, newStaticTargetSetsChannel(serviceDiscoveryConfigs), nil
|
|
}
|