mirror of
https://github.com/grafana/grafana.git
synced 2025-02-12 00:25:46 -06:00
f83cd401e5
* Alerting: Send alerts to external Alertmanager(s) Within this PR we're adding support for registering or unregistering sending to a set of external alertmanagers. A few of the things that are going are: - Introduce a new table to hold "admin" (either org or global) configuration we can change at runtime. - A new periodic check that polls for this configuration and adjusts the "senders" accordingly. - Introduces a new concept of "senders" that are responsible for shipping the alerts to the external Alertmanager(s). In a nutshell, this is the Prometheus notifier (the one in charge of sending the alert) mapped to a multi-tenant map. There are a few code movements here and there but those are minor, I tried to keep things intact as much as possible so that we could have an easier diff.
93 lines
2.4 KiB
Go
93 lines
2.4 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
ngmodels "github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
"github.com/grafana/grafana/pkg/services/sqlstore"
|
|
)
|
|
|
|
var (
|
|
// ErrNoAdminConfiguration is an error for when no admin configuration is found.
|
|
ErrNoAdminConfiguration = fmt.Errorf("no admin configuration available")
|
|
)
|
|
|
|
type UpdateAdminConfigurationCmd struct {
|
|
AdminConfiguration *ngmodels.AdminConfiguration
|
|
}
|
|
|
|
type AdminConfigurationStore interface {
|
|
GetAdminConfiguration(orgID int64) (*ngmodels.AdminConfiguration, error)
|
|
GetAdminConfigurations() ([]*ngmodels.AdminConfiguration, error)
|
|
DeleteAdminConfiguration(orgID int64) error
|
|
UpdateAdminConfiguration(UpdateAdminConfigurationCmd) error
|
|
}
|
|
|
|
func (st *DBstore) GetAdminConfiguration(orgID int64) (*ngmodels.AdminConfiguration, error) {
|
|
cfg := &ngmodels.AdminConfiguration{}
|
|
err := st.SQLStore.WithDbSession(context.Background(), func(sess *sqlstore.DBSession) error {
|
|
ok, err := sess.Table("ngalert_configuration").Where("org_id = ?", orgID).Get(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !ok {
|
|
return ErrNoAdminConfiguration
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return cfg, nil
|
|
}
|
|
|
|
func (st DBstore) GetAdminConfigurations() ([]*ngmodels.AdminConfiguration, error) {
|
|
var cfg []*ngmodels.AdminConfiguration
|
|
err := st.SQLStore.WithDbSession(context.Background(), func(sess *sqlstore.DBSession) error {
|
|
if err := sess.Table("ngalert_configuration").Find(&cfg); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return cfg, nil
|
|
}
|
|
|
|
func (st DBstore) DeleteAdminConfiguration(orgID int64) error {
|
|
return st.SQLStore.WithDbSession(context.Background(), func(sess *sqlstore.DBSession) error {
|
|
_, err := sess.Exec("DELETE FROM ngalert_configuration WHERE org_id = ?", orgID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func (st DBstore) UpdateAdminConfiguration(cmd UpdateAdminConfigurationCmd) error {
|
|
return st.SQLStore.WithTransactionalDbSession(context.Background(), func(sess *sqlstore.DBSession) error {
|
|
has, err := sess.Table("ngalert_configuration").Where("org_id = ?", cmd.AdminConfiguration.OrgID).Exist()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !has {
|
|
_, err := sess.Table("ngalert_configuration").Insert(cmd.AdminConfiguration)
|
|
return err
|
|
}
|
|
|
|
_, err = sess.Table("ngalert_configuration").AllCols().Update(cmd.AdminConfiguration)
|
|
return err
|
|
})
|
|
}
|