The open and composable observability and data visualization platform. Visualize metrics, logs, and traces from multiple sources like Prometheus, Loki, Elasticsearch, InfluxDB, Postgres and many more.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
grafana/pkg/services/ngalert/state/manager_bench_test.go

115 lines
3.2 KiB

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/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)
hist := historian.NewAnnotationBackend(store, nil, metrics)
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)
}
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
}