129 lines
3.0 KiB
Go
129 lines
3.0 KiB
Go
// Copyright 2015 Prometheus Team
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/prometheus/common/log"
|
|
"github.com/prometheus/common/route"
|
|
|
|
"github.com/prometheus/alertmanager/config"
|
|
"github.com/prometheus/alertmanager/notify"
|
|
"github.com/prometheus/alertmanager/provider"
|
|
"github.com/prometheus/alertmanager/types"
|
|
)
|
|
|
|
var (
|
|
configFile = flag.String("config.file", "config.yml", "The configuration file")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
data := provider.NewMemData()
|
|
|
|
alerts := provider.NewMemAlerts(data)
|
|
notifies := provider.NewMemNotifies(data)
|
|
silences := provider.NewMemSilences()
|
|
|
|
inhibitor := &Inhibitor{alerts: alerts}
|
|
|
|
routedNotifier := notify.NewRoutedNotifier(func(confs []*config.NotificationConfig) map[string]notify.Notifier {
|
|
res := notify.Build(confs)
|
|
for name, n := range res {
|
|
res[name] = ¬ify.LogNotifier{
|
|
Log: log.With("notifier", "dedup"),
|
|
Notifier: notify.NewDedupingNotifier(notifies, n),
|
|
}
|
|
}
|
|
return res
|
|
})
|
|
|
|
var notifier notify.Notifier
|
|
notifier = ¬ify.LogNotifier{
|
|
Log: log.With("notifier", "routed"),
|
|
Notifier: routedNotifier,
|
|
}
|
|
notifier = ¬ify.MutingNotifier{
|
|
Notifier: notifier,
|
|
Muter: inhibitor,
|
|
}
|
|
notifier = ¬ify.LogNotifier{
|
|
Log: log.With("notifier", "inhibit"),
|
|
Notifier: notifier,
|
|
}
|
|
notifier = ¬ify.MutingNotifier{
|
|
Notifier: notifier,
|
|
Muter: silences,
|
|
}
|
|
notifier = ¬ify.LogNotifier{
|
|
Log: log.With("notifier", "silencer"),
|
|
Notifier: notifier,
|
|
}
|
|
|
|
disp := NewDispatcher(alerts, notifier)
|
|
|
|
if !reloadConfig(*configFile, disp, routedNotifier, inhibitor) {
|
|
os.Exit(1)
|
|
}
|
|
|
|
go disp.Run()
|
|
defer disp.Stop()
|
|
|
|
router := route.New()
|
|
|
|
NewAPI(router.WithPrefix("/api"), alerts, silences)
|
|
|
|
go http.ListenAndServe(":9091", router)
|
|
|
|
var (
|
|
hup = make(chan os.Signal)
|
|
term = make(chan os.Signal)
|
|
)
|
|
signal.Notify(hup, syscall.SIGHUP)
|
|
signal.Notify(term, os.Interrupt, syscall.SIGTERM)
|
|
|
|
go func() {
|
|
for range hup {
|
|
reloadConfig(*configFile, disp, routedNotifier, inhibitor)
|
|
}
|
|
}()
|
|
|
|
<-term
|
|
|
|
log.Infoln("Received SIGTERM, exiting gracefully...")
|
|
os.Exit(0)
|
|
}
|
|
|
|
func reloadConfig(filename string, rls ...types.Reloadable) (success bool) {
|
|
log.Infof("Loading configuration file %s", filename)
|
|
|
|
conf, err := config.LoadFile(filename)
|
|
if err != nil {
|
|
log.Errorf("Couldn't load configuration (-config.file=%s): %v", filename, err)
|
|
return false
|
|
}
|
|
success = true
|
|
|
|
for _, rl := range rls {
|
|
success = success && rl.ApplyConfig(conf)
|
|
}
|
|
return success
|
|
}
|