Correlations: Add legacy storage (#112038)

2025-10-16 21:13:39 +03:00
parent bb08b2deea
commit bea45a94f0
17 changed files with 738 additions and 77 deletions
@@ -0,0 +1,151 @@
+package correlations
+
+import (
+	"encoding/json"
+
+	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/utils/ptr"
+
+	authlib "github.com/grafana/authlib/types"
+	correlationsV0 "github.com/grafana/grafana/apps/correlations/pkg/apis/correlation/v0alpha1"
+	"github.com/grafana/grafana/pkg/apimachinery/utils"
+)
+
+func ToResource(orig Correlation, namespacer authlib.NamespaceFormatter) (*correlationsV0.Correlation, error) {
+	cfg, err := ToSpecConfig(orig.Config)
+	if err != nil {
+		return nil, err
+	}
+	obj := &correlationsV0.Correlation{
+		ObjectMeta: v1.ObjectMeta{
+			Name:      orig.UID,
+			Namespace: namespacer(orig.OrgID),
+		},
+		Spec: correlationsV0.CorrelationSpec{
+			Label: orig.Label,
+			Type:  correlationsV0.CorrelationCorrelationType(orig.Type),
+			Source: correlationsV0.CorrelationDataSourceRef{
+				Group: ptr.Deref(orig.SourceType, ""),
+				Name:  orig.SourceUID,
+			},
+			Config: *cfg,
+		},
+	}
+	if orig.TargetUID != nil {
+		obj.Spec.Target = &correlationsV0.CorrelationDataSourceRef{
+			Group: ptr.Deref(orig.TargetType, ""),
+			Name:  *orig.TargetUID,
+		}
+	}
+	if orig.Description != "" {
+		obj.Spec.Description = &orig.Description
+	}
+	if orig.Provisioned {
+		tmp, _ := utils.MetaAccessor(obj)
+		tmp.SetManagerProperties(utils.ManagerProperties{
+			Kind: utils.ManagerKindClassicFP, // nolint:staticcheck
+		})
+	}
+	return obj, nil
+}
+
+func ToCorrelation(obj *correlationsV0.Correlation) (*Correlation, error) {
+	ns, err := authlib.ParseNamespace(obj.Namespace)
+	if err != nil {
+		return nil, err
+	}
+	cfg, err := ToConfig(obj.Spec.Config)
+	if err != nil {
+		return nil, err
+	}
+	result := &Correlation{
+		UID:         obj.Name,
+		OrgID:       ns.OrgID,
+		Label:       obj.Spec.Label,
+		Description: ptr.Deref(obj.Spec.Description, ""),
+		SourceUID:   obj.Spec.Source.Name,
+		SourceType:  ptr.To(obj.Spec.Source.Group),
+		Type:        CorrelationType(obj.Spec.Type),
+		Config:      *cfg,
+	}
+	if obj.Annotations[utils.AnnoKeyManagerKind] != "" {
+		result.Provisioned = true
+	}
+	if obj.Spec.Target != nil {
+		result.TargetUID = &obj.Spec.Target.Name
+		result.TargetType = ptr.To(obj.Spec.Target.Group)
+	}
+	return result, nil
+}
+
+func ToSpecConfig(orig CorrelationConfig) (*correlationsV0.CorrelationConfigSpec, error) {
+	out := &correlationsV0.CorrelationConfigSpec{}
+	raw, err := json.Marshal(orig)
+	if err != nil {
+		return nil, err
+	}
+	err = json.Unmarshal(raw, out)
+	if err != nil {
+		return nil, err
+	}
+	if len(out.Target) == 0 {
+		out.Target = nil
+	}
+	return out, err
+}
+
+func ToConfig(orig correlationsV0.CorrelationConfigSpec) (*CorrelationConfig, error) {
+	out := &CorrelationConfig{}
+	raw, err := json.Marshal(orig)
+	if err != nil {
+		return nil, err
+	}
+	err = json.Unmarshal(raw, out)
+	if err != nil {
+		return nil, err
+	}
+	if len(out.Target) == 0 {
+		out.Target = nil
+	}
+	return out, err
+}
+
+func ToUpdateCorrelationCommand(obj *correlationsV0.Correlation) (*UpdateCorrelationCommand, error) {
+	tmp, err := ToCorrelation(obj)
+	if err != nil {
+		return nil, err
+	}
+	if tmp.Config.Target == nil {
+		tmp.Config.Target = map[string]any{} // replace it
+	}
+	return &UpdateCorrelationCommand{
+		UID:         tmp.UID,
+		OrgId:       tmp.OrgID,
+		SourceUID:   tmp.SourceUID,
+		Label:       &tmp.Label,
+		Description: &tmp.Description,
+		Type:        &tmp.Type,
+		Config: &CorrelationConfigUpdateDTO{
+			Field:           &tmp.Config.Field,
+			Target:          &tmp.Config.Target,
+			Transformations: tmp.Config.Transformations,
+		},
+	}, nil
+}
+
+func ToCreateCorrelationCommand(obj *correlationsV0.Correlation) (*CreateCorrelationCommand, error) {
+	tmp, err := ToCorrelation(obj)
+	if err != nil {
+		return nil, err
+	}
+	return &CreateCorrelationCommand{
+		OrgId:       tmp.OrgID,
+		SourceUID:   tmp.SourceUID,
+		TargetUID:   tmp.TargetUID,
+		Label:       tmp.Label,
+		Description: tmp.Description,
+		Config:      tmp.Config,
+		Type:        tmp.Type,
+		Provisioned: tmp.Provisioned,
+	}, nil
+}