feat - migrate crd in etcd

Signed-off-by: Hélia Barroso <[email protected]>

Author: heliapb

cmd/migrate.go

@@ -0,0 +1,55 @@
+// Copyright 2024 The prometheus-operator Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//	http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/prometheus-operator/poctl/internal/etcdmigrate"
+	"github.com/prometheus-operator/poctl/internal/k8sutil"
+	"github.com/prometheus-operator/poctl/internal/log"
+	"github.com/spf13/cobra"
+)
+
+// migrateCmd represents the etcd store objects migration command.
+var migrateCmd = &cobra.Command{
+	Use:   "migrate",
+	Short: "Automatically update Custom Resources to the latest storage version.",
+	Long:  `The migrate command in poctl automates the process of updating Kubernetes Custom Resources to the latest storage version. This is essential when upgrading a CRD that supports multiple API versions.`,
+	RunE:  runMigration,
+}
+
+func init() {
+	rootCmd.AddCommand(migrateCmd)
+}
+
+func runMigration(cmd *cobra.Command, _ []string) error {
+	logger, err := log.NewLogger()
+	if err != nil {
+		return fmt.Errorf("error while creating logger: %v", err)
+	}
+	clientSets, err := k8sutil.GetClientSets(kubeconfig)
+	if err != nil {
+		logger.Error("error while getting client sets", "err", err)
+		return err
+	}
+
+	if err := etcdmigrate.MigrateCRDs(cmd.Context(), clientSets); err != nil {
+		logger.Error("error while updating etcd store", "err", err)
+	}
+
+	logger.Info("Prometheus Operator CRD were update in etcd store ")
+	return nil
+}

internal/etcdmigrate/etcdmigrate.go

@@ -0,0 +1,97 @@
+// Copyright 2024 The prometheus-operator Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//	http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package etcdmigrate
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+
+	"github.com/prometheus-operator/poctl/internal/k8sutil"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+)
+
+func MigrateCRDs(ctx context.Context, clientSets *k8sutil.ClientSets) error {
+	crds, err := clientSets.APIExtensionsClient.ApiextensionsV1().CustomResourceDefinitions().List(ctx, metav1.ListOptions{})
+	if err != nil {
+		return fmt.Errorf("failed to list CRDs: %w", err)
+	}
+
+	for _, crd := range crds.Items {
+		if crd.Spec.Group != "monitoring.coreos.com" {
+			continue
+		}
+
+		var storageVersion string
+		for _, version := range crd.Spec.Versions {
+			if version.Storage {
+				storageVersion = version.Name
+				break
+			}
+		}
+		if storageVersion == "" {
+			continue
+		}
+
+		crdResourceVersion := schema.GroupVersionResource{
+			Group:    crd.Spec.Group,
+			Version:  storageVersion,
+			Resource: crd.Spec.Names.Plural,
+		}
+
+		namespaces, err := clientSets.KClient.CoreV1().Namespaces().List(ctx, metav1.ListOptions{})
+		if err != nil {
+			return fmt.Errorf("failed to list Namespaces %v", err)
+		}
+		for _, namespace := range namespaces.Items {
+			ns := namespace.Name
+
+			customResourcesInstances, err := clientSets.DClient.Resource(crdResourceVersion).Namespace(ns).List(ctx, metav1.ListOptions{})
+			if err != nil {
+				continue
+			}
+
+			for _, cri := range customResourcesInstances.Items {
+				name := cri.GetName()
+				apiVersion := cri.GetAPIVersion()
+
+				expectedAPIVersion := fmt.Sprintf("%s/%s", crd.Spec.Group, storageVersion)
+				if apiVersion == expectedAPIVersion {
+					continue
+				}
+
+				crdJSON, err := json.Marshal(cri.Object)
+				if err != nil {
+					continue
+				}
+
+				updatedStorageObject := &unstructured.Unstructured{}
+				if err := json.Unmarshal(crdJSON, &updatedStorageObject.Object); err != nil {
+					continue
+				}
+
+				_, err = clientSets.DClient.Resource(crdResourceVersion).Namespace(ns).Update(ctx, updatedStorageObject, metav1.UpdateOptions{})
+				if err != nil {
+					return fmt.Errorf("failed to update %s %s: %v", ns, name, err)
+				}
+			}
+		}
+	}
+
+	fmt.Println("CRD migration completed.")
+	return nil
+}