mirror of
https://github.com/grafana/grafana.git
synced 2024-11-25 10:20:29 -06:00
aa03b8f8a7
This PR has two steps that together create a functional dry-run capability for the migration.
By enabling the feature flag alertingPreviewUpgrade when on legacy alerting it will:
a. Allow all Grafana Alerting background services except for the scheduler to start (multiorg alertmanager, state manager, routes, …).
b. Allow the UI to show Grafana Alerting pages alongside legacy ones (with appropriate in-app warnings that UA is not actually running).
c. Show a new “Alerting Upgrade” page and register associated /api/v1/upgrade endpoints that will allow the user to upgrade their organization live without restart and present a summary of the upgrade in a table.
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
package ngalert
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
"github.com/grafana/grafana/pkg/services/quota"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
)
|
|
|
|
type RuleUsageReader interface {
|
|
Count(ctx context.Context, orgID int64) (int64, error)
|
|
}
|
|
|
|
func RegisterQuotas(cfg *setting.Cfg, qs quota.Service, rules RuleUsageReader) error {
|
|
defaultLimits, err := readQuotaConfig(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return qs.RegisterQuotaReporter("a.NewUsageReporter{
|
|
TargetSrv: models.QuotaTargetSrv,
|
|
DefaultLimits: defaultLimits,
|
|
Reporter: UsageReporter(rules),
|
|
})
|
|
}
|
|
|
|
func UsageReporter(rules RuleUsageReader) quota.UsageReporterFunc {
|
|
return func(ctx context.Context, scopeParams *quota.ScopeParameters) (*quota.Map, error) {
|
|
u := "a.Map{}
|
|
|
|
var orgID int64 = 0
|
|
if scopeParams != nil {
|
|
orgID = scopeParams.OrgID
|
|
}
|
|
|
|
if orgUsage, err := rules.Count(ctx, orgID); err != nil {
|
|
return u, err
|
|
} else {
|
|
tag, err := quota.NewTag(models.QuotaTargetSrv, models.QuotaTarget, quota.OrgScope)
|
|
if err != nil {
|
|
return u, err
|
|
}
|
|
u.Set(tag, orgUsage)
|
|
}
|
|
|
|
if globalUsage, err := rules.Count(ctx, 0); err != nil {
|
|
return u, err
|
|
} else {
|
|
tag, err := quota.NewTag(models.QuotaTargetSrv, models.QuotaTarget, quota.GlobalScope)
|
|
if err != nil {
|
|
return u, err
|
|
}
|
|
u.Set(tag, globalUsage)
|
|
}
|
|
|
|
return u, nil
|
|
}
|
|
}
|
|
|
|
func readQuotaConfig(cfg *setting.Cfg) (*quota.Map, error) {
|
|
limits := "a.Map{}
|
|
|
|
if cfg == nil {
|
|
return limits, nil
|
|
}
|
|
|
|
globalQuotaTag, err := quota.NewTag(models.QuotaTargetSrv, models.QuotaTarget, quota.GlobalScope)
|
|
if err != nil {
|
|
return limits, err
|
|
}
|
|
orgQuotaTag, err := quota.NewTag(models.QuotaTargetSrv, models.QuotaTarget, quota.OrgScope)
|
|
if err != nil {
|
|
return limits, err
|
|
}
|
|
|
|
limits.Set(globalQuotaTag, cfg.Quota.Global.AlertRule)
|
|
limits.Set(orgQuotaTag, cfg.Quota.Org.AlertRule)
|
|
return limits, nil
|
|
}
|