mirror of
https://github.com/grafana/grafana.git
synced 2024-11-25 18:30:41 -06:00
06d5850396
* add method CanReadAllRules to rule authorization service
* add alias type Namespace for Folder in ngalert's models package. It implements the Namespacer interface that is used by authz logic
* update state history's backends to authorize access to rules.
* update Loki to add folders UIDs to query.
* Update BuildLogQuery to drop filter by folders if it's too long and fall back to in-memory filtering.
119 lines
3.4 KiB
Go
119 lines
3.4 KiB
Go
package state_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/stretchr/testify/mock"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/infra/tracing"
|
|
"github.com/grafana/grafana/pkg/services/annotations"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/accesscontrol/fakes"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/eval"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/metrics"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/state"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/state/historian"
|
|
)
|
|
|
|
func BenchmarkProcessEvalResults(b *testing.B) {
|
|
as := annotations.FakeAnnotationsRepo{}
|
|
as.On("SaveMany", mock.Anything, mock.Anything).Return(nil)
|
|
metrics := metrics.NewHistorianMetrics(prometheus.NewRegistry(), metrics.Subsystem)
|
|
store := historian.NewAnnotationStore(&as, nil, metrics)
|
|
annotationBackendLogger := log.New("ngalert.state.historian", "backend", "annotations")
|
|
ac := &fakes.FakeRuleService{}
|
|
hist := historian.NewAnnotationBackend(annotationBackendLogger, store, nil, metrics, ac)
|
|
cfg := state.ManagerCfg{
|
|
Historian: hist,
|
|
Tracer: tracing.InitializeTracerForTest(),
|
|
Log: log.New("ngalert.state.manager"),
|
|
}
|
|
sut := state.NewManager(cfg, state.NewNoopPersister())
|
|
now := time.Now().UTC()
|
|
rule := makeBenchRule()
|
|
results := makeBenchResults(100)
|
|
labels := map[string]string{}
|
|
|
|
var ans []state.StateTransition
|
|
for i := 0; i < b.N; i++ {
|
|
ans = sut.ProcessEvalResults(context.Background(), now, &rule, results, labels, nil)
|
|
}
|
|
|
|
b.StopTimer()
|
|
|
|
_ = fmt.Sprintf("%v", len(ans))
|
|
}
|
|
|
|
func makeBenchRule() models.AlertRule {
|
|
dashUID := "my-dash"
|
|
panelID := int64(14)
|
|
return models.AlertRule{
|
|
ID: 5,
|
|
OrgID: 1,
|
|
Title: "some rule",
|
|
Condition: "A",
|
|
Data: []models.AlertQuery{},
|
|
Updated: time.Now().UTC(),
|
|
IntervalSeconds: 60,
|
|
Version: 2,
|
|
UID: "abcd-efg",
|
|
NamespaceUID: "my-folder",
|
|
DashboardUID: &dashUID,
|
|
PanelID: &panelID,
|
|
RuleGroup: "my-group",
|
|
RuleGroupIndex: 2,
|
|
NoDataState: models.NoData,
|
|
ExecErrState: models.ErrorErrState,
|
|
For: 5 * time.Minute,
|
|
Annotations: map[string]string{
|
|
"text": "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.",
|
|
"url": "https://grafana.com",
|
|
},
|
|
Labels: map[string]string{
|
|
"alertname": "some rule",
|
|
"a": "b",
|
|
"cluster": "prod-eu-west-123",
|
|
"namespace": "coolthings",
|
|
},
|
|
}
|
|
}
|
|
|
|
func makeBenchResults(count int) eval.Results {
|
|
labels := map[string]string{
|
|
"alertname": "some rule",
|
|
"a": "b",
|
|
"cluster": "prod-eu-west-123",
|
|
"namespace": "coolthings",
|
|
}
|
|
one := 1.0
|
|
results := make([]eval.Result, 0, count)
|
|
for i := 0; i < count; i++ {
|
|
results = append(results, eval.Result{
|
|
Instance: labels,
|
|
State: eval.Alerting,
|
|
EvaluatedAt: time.Now().UTC(),
|
|
EvaluationDuration: 5 * time.Second,
|
|
Values: map[string]eval.NumberValueCapture{
|
|
"A": {
|
|
Var: "A",
|
|
Value: &one,
|
|
},
|
|
"B": {
|
|
Var: "B",
|
|
Value: &one,
|
|
},
|
|
"C": {
|
|
Var: "C",
|
|
Value: &one,
|
|
},
|
|
},
|
|
})
|
|
}
|
|
return results
|
|
}
|