Chore: Move annotations cleanup to the annotations service (#55618)

pkg/services/annotations/annotationsimpl/cleanup.go

@@ -0,0 +1,62 @@
+package annotationsimpl
+
+import (
+	"context"
+
+	"github.com/grafana/grafana/pkg/infra/log"
+	"github.com/grafana/grafana/pkg/services/sqlstore/db"
+	"github.com/grafana/grafana/pkg/setting"
+)
+
+// CleanupServiceImpl is responsible for cleaning old annotations.
+type CleanupServiceImpl struct {
+	store store
+}
+
+func ProvideCleanupService(db db.DB, cfg *setting.Cfg) *CleanupServiceImpl {
+	return &CleanupServiceImpl{
+		store: &xormRepositoryImpl{
+			cfg: cfg,
+			db:  db,
+			log: log.New("annotations"),
+		},
+	}
+}
+
+const (
+	alertAnnotationType     = "alert_id <> 0"
+	dashboardAnnotationType = "dashboard_id <> 0 AND alert_id = 0"
+	apiAnnotationType       = "alert_id = 0 AND dashboard_id = 0"
+)
+
+// Run deletes old annotations created by alert rules, API
+// requests and human made in the UI. It subsequently deletes orphaned rows
+// from the annotation_tag table. Cleanup actions are performed in batches
+// so that no query takes too long to complete.
+//
+// Returns the number of annotation and annotation_tag rows deleted. If an
+// error occurs, it returns the number of rows affected so far.
+func (cs *CleanupServiceImpl) Run(ctx context.Context, cfg *setting.Cfg) (int64, int64, error) {
+	var totalCleanedAnnotations int64
+	affected, err := cs.store.CleanAnnotations(ctx, cfg.AlertingAnnotationCleanupSetting, alertAnnotationType)
+	totalCleanedAnnotations += affected
+	if err != nil {
+		return totalCleanedAnnotations, 0, err
+	}
+
+	affected, err = cs.store.CleanAnnotations(ctx, cfg.APIAnnotationCleanupSettings, apiAnnotationType)
+	totalCleanedAnnotations += affected
+	if err != nil {
+		return totalCleanedAnnotations, 0, err
+	}
+
+	affected, err = cs.store.CleanAnnotations(ctx, cfg.DashboardAnnotationCleanupSettings, dashboardAnnotationType)
+	totalCleanedAnnotations += affected
+	if err != nil {
+		return totalCleanedAnnotations, 0, err
+	}
+	if totalCleanedAnnotations > 0 {
+		affected, err = cs.store.CleanOrphanedAnnotationTags(ctx)
+	}
+	return totalCleanedAnnotations, affected, err
+}