From 0efb84617e2896af30d1edcacaf1f8b2cb3089a8 Mon Sep 17 00:00:00 2001
From: Alexander Weaver <weaver.alex.d@gmail.com>
Date: Fri, 3 Feb 2023 15:38:08 -0600
Subject: [PATCH] Alerting: Create benchmarking test for
 state.ProcessEvalResults (#62041)

* Create benchmark for ProcessEvalResults

* Simplify the test
---
 .../ngalert/state/manager_bench_test.go       | 106 ++++++++++++++++++
 1 file changed, 106 insertions(+)
 create mode 100644 pkg/services/ngalert/state/manager_bench_test.go

diff --git a/pkg/services/ngalert/state/manager_bench_test.go b/pkg/services/ngalert/state/manager_bench_test.go
new file mode 100644
index 00000000000..0690933a72c
--- /dev/null
+++ b/pkg/services/ngalert/state/manager_bench_test.go
@@ -0,0 +1,106 @@
+package state_test
+
+import (
+	"context"
+	"fmt"
+	"testing"
+	"time"
+
+	"github.com/grafana/grafana/pkg/services/annotations"
+	"github.com/grafana/grafana/pkg/services/ngalert/eval"
+	"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"
+	"github.com/stretchr/testify/mock"
+)
+
+func BenchmarkProcessEvalResults(b *testing.B) {
+	as := annotations.FakeAnnotationsRepo{}
+	as.On("SaveMany", mock.Anything, mock.Anything).Return(nil)
+	hist := historian.NewAnnotationBackend(&as, nil, nil)
+	cfg := state.ManagerCfg{
+		Historian: hist,
+	}
+	sut := state.NewManager(cfg)
+	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": eval.NumberValueCapture{
+					Var:   "A",
+					Value: &one,
+				},
+				"B": eval.NumberValueCapture{
+					Var:   "B",
+					Value: &one,
+				},
+				"C": eval.NumberValueCapture{
+					Var:   "C",
+					Value: &one,
+				},
+			},
+		})
+	}
+	return results
+}