mirror of
https://github.com/grafana/grafana.git
synced 2024-11-29 04:04:00 -06:00
003e3efce9
* add version to time-interval models * set time interval fingerprint as version * update to check provided version * delete to check if version is provided in query parameter 'version' * update integration tests * update specs
320 lines
10 KiB
Go
320 lines
10 KiB
Go
package provisioning
|
|
|
|
import (
|
|
"context"
|
|
"encoding/binary"
|
|
"fmt"
|
|
"hash/fnv"
|
|
"unsafe"
|
|
|
|
"github.com/prometheus/alertmanager/config"
|
|
"github.com/prometheus/alertmanager/timeinterval"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/api/tooling/definitions"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
)
|
|
|
|
type MuteTimingService struct {
|
|
configStore alertmanagerConfigStore
|
|
provenanceStore ProvisioningStore
|
|
xact TransactionManager
|
|
log log.Logger
|
|
validator ProvenanceStatusTransitionValidator
|
|
}
|
|
|
|
func NewMuteTimingService(config AMConfigStore, prov ProvisioningStore, xact TransactionManager, log log.Logger) *MuteTimingService {
|
|
return &MuteTimingService{
|
|
configStore: &alertmanagerConfigStoreImpl{store: config},
|
|
provenanceStore: prov,
|
|
xact: xact,
|
|
log: log,
|
|
validator: ValidateProvenanceRelaxed,
|
|
}
|
|
}
|
|
|
|
// GetMuteTimings returns a slice of all mute timings within the specified org.
|
|
func (svc *MuteTimingService) GetMuteTimings(ctx context.Context, orgID int64) ([]definitions.MuteTimeInterval, error) {
|
|
rev, err := svc.configStore.Get(ctx, orgID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if rev.cfg.AlertmanagerConfig.MuteTimeIntervals == nil {
|
|
return []definitions.MuteTimeInterval{}, nil
|
|
}
|
|
|
|
provenances, err := svc.provenanceStore.GetProvenances(ctx, orgID, (&definitions.MuteTimeInterval{}).ResourceType())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := make([]definitions.MuteTimeInterval, 0, len(rev.cfg.AlertmanagerConfig.MuteTimeIntervals))
|
|
for _, interval := range rev.cfg.AlertmanagerConfig.MuteTimeIntervals {
|
|
version := calculateMuteTimeIntervalFingerprint(interval)
|
|
def := definitions.MuteTimeInterval{MuteTimeInterval: interval, Version: version}
|
|
if prov, ok := provenances[def.ResourceID()]; ok {
|
|
def.Provenance = definitions.Provenance(prov)
|
|
}
|
|
result = append(result, def)
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
// GetMuteTiming returns a mute timing by name
|
|
func (svc *MuteTimingService) GetMuteTiming(ctx context.Context, name string, orgID int64) (definitions.MuteTimeInterval, error) {
|
|
rev, err := svc.configStore.Get(ctx, orgID)
|
|
if err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
|
|
mt, _, err := getMuteTiming(rev, name)
|
|
if err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
|
|
result := definitions.MuteTimeInterval{
|
|
MuteTimeInterval: mt,
|
|
Version: calculateMuteTimeIntervalFingerprint(mt),
|
|
}
|
|
|
|
prov, err := svc.provenanceStore.GetProvenance(ctx, &result, orgID)
|
|
if err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
result.Provenance = definitions.Provenance(prov)
|
|
return result, nil
|
|
}
|
|
|
|
// CreateMuteTiming adds a new mute timing within the specified org. The created mute timing is returned.
|
|
func (svc *MuteTimingService) CreateMuteTiming(ctx context.Context, mt definitions.MuteTimeInterval, orgID int64) (definitions.MuteTimeInterval, error) {
|
|
if err := mt.Validate(); err != nil {
|
|
return definitions.MuteTimeInterval{}, MakeErrTimeIntervalInvalid(err)
|
|
}
|
|
|
|
revision, err := svc.configStore.Get(ctx, orgID)
|
|
if err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
|
|
if revision.cfg.AlertmanagerConfig.MuteTimeIntervals == nil {
|
|
revision.cfg.AlertmanagerConfig.MuteTimeIntervals = []config.MuteTimeInterval{}
|
|
}
|
|
for _, existing := range revision.cfg.AlertmanagerConfig.MuteTimeIntervals {
|
|
if mt.Name == existing.Name {
|
|
return definitions.MuteTimeInterval{}, ErrTimeIntervalExists.Errorf("")
|
|
}
|
|
}
|
|
revision.cfg.AlertmanagerConfig.MuteTimeIntervals = append(revision.cfg.AlertmanagerConfig.MuteTimeIntervals, mt.MuteTimeInterval)
|
|
|
|
err = svc.xact.InTransaction(ctx, func(ctx context.Context) error {
|
|
if err := svc.configStore.Save(ctx, revision, orgID); err != nil {
|
|
return err
|
|
}
|
|
return svc.provenanceStore.SetProvenance(ctx, &mt, orgID, models.Provenance(mt.Provenance))
|
|
})
|
|
if err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
return definitions.MuteTimeInterval{
|
|
MuteTimeInterval: mt.MuteTimeInterval,
|
|
Version: calculateMuteTimeIntervalFingerprint(mt.MuteTimeInterval),
|
|
Provenance: mt.Provenance,
|
|
}, nil
|
|
}
|
|
|
|
// UpdateMuteTiming replaces an existing mute timing within the specified org. The replaced mute timing is returned. If the mute timing does not exist, ErrMuteTimingsNotFound is returned.
|
|
func (svc *MuteTimingService) UpdateMuteTiming(ctx context.Context, mt definitions.MuteTimeInterval, orgID int64) (definitions.MuteTimeInterval, error) {
|
|
if err := mt.Validate(); err != nil {
|
|
return definitions.MuteTimeInterval{}, MakeErrTimeIntervalInvalid(err)
|
|
}
|
|
|
|
// check that provenance is not changed in an invalid way
|
|
storedProvenance, err := svc.provenanceStore.GetProvenance(ctx, &mt, orgID)
|
|
if err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
if err := svc.validator(storedProvenance, models.Provenance(mt.Provenance)); err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
|
|
revision, err := svc.configStore.Get(ctx, orgID)
|
|
if err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
|
|
if revision.cfg.AlertmanagerConfig.MuteTimeIntervals == nil {
|
|
return definitions.MuteTimeInterval{}, nil
|
|
}
|
|
|
|
old, idx, err := getMuteTiming(revision, mt.Name)
|
|
if err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
|
|
err = svc.checkOptimisticConcurrency(old, models.Provenance(mt.Provenance), mt.Version, "update")
|
|
if err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
|
|
revision.cfg.AlertmanagerConfig.MuteTimeIntervals[idx] = mt.MuteTimeInterval
|
|
|
|
// TODO add diff and noop detection
|
|
err = svc.xact.InTransaction(ctx, func(ctx context.Context) error {
|
|
if err := svc.configStore.Save(ctx, revision, orgID); err != nil {
|
|
return err
|
|
}
|
|
return svc.provenanceStore.SetProvenance(ctx, &mt, orgID, models.Provenance(mt.Provenance))
|
|
})
|
|
if err != nil {
|
|
return definitions.MuteTimeInterval{}, err
|
|
}
|
|
return definitions.MuteTimeInterval{
|
|
MuteTimeInterval: mt.MuteTimeInterval,
|
|
Version: calculateMuteTimeIntervalFingerprint(mt.MuteTimeInterval),
|
|
Provenance: mt.Provenance,
|
|
}, err
|
|
}
|
|
|
|
// DeleteMuteTiming deletes the mute timing with the given name in the given org. If the mute timing does not exist, no error is returned.
|
|
func (svc *MuteTimingService) DeleteMuteTiming(ctx context.Context, name string, orgID int64, provenance definitions.Provenance, version string) error {
|
|
target := definitions.MuteTimeInterval{MuteTimeInterval: config.MuteTimeInterval{Name: name}, Provenance: provenance}
|
|
// check that provenance is not changed in an invalid way
|
|
storedProvenance, err := svc.provenanceStore.GetProvenance(ctx, &target, orgID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := svc.validator(storedProvenance, models.Provenance(provenance)); err != nil {
|
|
return err
|
|
}
|
|
|
|
revision, err := svc.configStore.Get(ctx, orgID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if revision.cfg.AlertmanagerConfig.MuteTimeIntervals == nil {
|
|
return nil
|
|
}
|
|
if isMuteTimeInUse(name, []*definitions.Route{revision.cfg.AlertmanagerConfig.Route}) {
|
|
return ErrTimeIntervalInUse.Errorf("")
|
|
}
|
|
for i, existing := range revision.cfg.AlertmanagerConfig.MuteTimeIntervals {
|
|
if name != existing.Name {
|
|
continue
|
|
}
|
|
err = svc.checkOptimisticConcurrency(existing, models.Provenance(provenance), version, "delete")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
intervals := revision.cfg.AlertmanagerConfig.MuteTimeIntervals
|
|
revision.cfg.AlertmanagerConfig.MuteTimeIntervals = append(intervals[:i], intervals[i+1:]...)
|
|
}
|
|
|
|
return svc.xact.InTransaction(ctx, func(ctx context.Context) error {
|
|
if err := svc.configStore.Save(ctx, revision, orgID); err != nil {
|
|
return err
|
|
}
|
|
return svc.provenanceStore.DeleteProvenance(ctx, &target, orgID)
|
|
})
|
|
}
|
|
|
|
func isMuteTimeInUse(name string, routes []*definitions.Route) bool {
|
|
if len(routes) == 0 {
|
|
return false
|
|
}
|
|
for _, route := range routes {
|
|
for _, mtName := range route.MuteTimeIntervals {
|
|
if mtName == name {
|
|
return true
|
|
}
|
|
}
|
|
if isMuteTimeInUse(name, route.Routes) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func getMuteTiming(rev *cfgRevision, name string) (config.MuteTimeInterval, int, error) {
|
|
if rev.cfg.AlertmanagerConfig.MuteTimeIntervals == nil {
|
|
return config.MuteTimeInterval{}, -1, ErrTimeIntervalNotFound.Errorf("")
|
|
}
|
|
for idx, mt := range rev.cfg.AlertmanagerConfig.MuteTimeIntervals {
|
|
if mt.Name == name {
|
|
return mt, idx, nil
|
|
}
|
|
}
|
|
return config.MuteTimeInterval{}, -1, ErrTimeIntervalNotFound.Errorf("")
|
|
}
|
|
|
|
func calculateMuteTimeIntervalFingerprint(interval config.MuteTimeInterval) string {
|
|
sum := fnv.New64()
|
|
|
|
writeBytes := func(b []byte) {
|
|
_, _ = sum.Write(b)
|
|
// add a byte sequence that cannot happen in UTF-8 strings.
|
|
_, _ = sum.Write([]byte{255})
|
|
}
|
|
writeString := func(s string) {
|
|
if len(s) == 0 {
|
|
writeBytes(nil)
|
|
return
|
|
}
|
|
// #nosec G103
|
|
// avoid allocation when converting string to byte slice
|
|
writeBytes(unsafe.Slice(unsafe.StringData(s), len(s)))
|
|
}
|
|
// this temp slice is used to convert ints to bytes.
|
|
tmp := make([]byte, 8)
|
|
writeInt := func(u int) {
|
|
binary.LittleEndian.PutUint64(tmp, uint64(u))
|
|
writeBytes(tmp)
|
|
}
|
|
|
|
writeRange := func(r timeinterval.InclusiveRange) {
|
|
writeInt(r.Begin)
|
|
writeInt(r.End)
|
|
}
|
|
|
|
// fields that determine the rule state
|
|
writeString(interval.Name)
|
|
for _, ti := range interval.TimeIntervals {
|
|
for _, time := range ti.Times {
|
|
writeInt(time.StartMinute)
|
|
writeInt(time.EndMinute)
|
|
}
|
|
for _, itm := range ti.Months {
|
|
writeRange(itm.InclusiveRange)
|
|
}
|
|
for _, itm := range ti.DaysOfMonth {
|
|
writeRange(itm.InclusiveRange)
|
|
}
|
|
for _, itm := range ti.Weekdays {
|
|
writeRange(itm.InclusiveRange)
|
|
}
|
|
for _, itm := range ti.Years {
|
|
writeRange(itm.InclusiveRange)
|
|
}
|
|
if ti.Location != nil {
|
|
writeString(ti.Location.String())
|
|
}
|
|
}
|
|
return fmt.Sprintf("%016x", sum.Sum64())
|
|
}
|
|
|
|
func (svc *MuteTimingService) checkOptimisticConcurrency(current config.MuteTimeInterval, provenance models.Provenance, desiredVersion string, action string) error {
|
|
if desiredVersion == "" {
|
|
if provenance != models.ProvenanceFile {
|
|
// if version is not specified and it's not a file provisioning, emit a log message to reflect that optimistic concurrency is disabled for this request
|
|
svc.log.Debug("ignoring optimistic concurrency check because version was not provided", "timeInterval", current.Name, "operation", action)
|
|
}
|
|
return nil
|
|
}
|
|
currentVersion := calculateMuteTimeIntervalFingerprint(current)
|
|
if currentVersion != desiredVersion {
|
|
return ErrVersionConflict.Errorf("provided version %s of time interval %s does not match current version %s", desiredVersion, current.Name, currentVersion)
|
|
}
|
|
return nil
|
|
}
|