2deced7d40
* Introduce new models RoutingTree, RouteDefaults and Route and api-server to serve them that is backed by provisioning notification policy service. * update method UpdatePolicyTree of notification policy service to return route and new version * declare new actions alert.notifications.routes:read and alert.notifications.routes:write and two corresponding fixed roles. --------- Co-authored-by: Tom Ratcliffe <tom.ratcliffe@grafana.com> Co-authored-by: Matthew Jacobson <matthew.jacobson@grafana.com>
56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package alerting
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/provisioning"
|
|
)
|
|
|
|
type NotificationPolicyProvisioner interface {
|
|
Provision(ctx context.Context, files []*AlertingFile) error
|
|
Unprovision(ctx context.Context, files []*AlertingFile) error
|
|
}
|
|
|
|
type defaultNotificationPolicyProvisioner struct {
|
|
logger log.Logger
|
|
notificationPolicyService provisioning.NotificationPolicyService
|
|
}
|
|
|
|
func NewNotificationPolicyProvisoner(logger log.Logger,
|
|
notificationPolicyService provisioning.NotificationPolicyService) NotificationPolicyProvisioner {
|
|
return &defaultNotificationPolicyProvisioner{
|
|
logger: logger,
|
|
notificationPolicyService: notificationPolicyService,
|
|
}
|
|
}
|
|
|
|
func (c *defaultNotificationPolicyProvisioner) Provision(ctx context.Context,
|
|
files []*AlertingFile) error {
|
|
for _, file := range files {
|
|
for _, np := range file.Policies {
|
|
_, _, err := c.notificationPolicyService.UpdatePolicyTree(ctx, np.OrgID,
|
|
np.Policy, models.ProvenanceFile, "")
|
|
if err != nil {
|
|
return fmt.Errorf("%s: %w", file.Filename, err)
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *defaultNotificationPolicyProvisioner) Unprovision(ctx context.Context,
|
|
files []*AlertingFile) error {
|
|
for _, file := range files {
|
|
for _, orgID := range file.ResetPolicies {
|
|
_, err := c.notificationPolicyService.ResetPolicyTree(ctx, int64(orgID), models.ProvenanceFile)
|
|
if err != nil {
|
|
return fmt.Errorf("%s: %w", file.Filename, err)
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|