vrg_status_pvcs.go

// SPDX-FileCopyrightText: The RamenDR authors
// SPDX-License-Identifier: Apache-2.0

package controllers

import (
	"github.com/go-logr/logr"
	ramen "github.com/ramendr/ramen/api/v1alpha1"
)

// findProtectedPVC returns the &VRG.Status.ProtectedPVC[x] for the given pvcName
func (v *VRGInstance) findProtectedPVC(pvcNamespaceName, pvcName string) *ramen.ProtectedPVC {
	return FindProtectedPVC(v.instance, pvcNamespaceName, pvcName)
}

func FindProtectedPVC(vrg *ramen.VolumeReplicationGroup, pvcNamespaceName, pvcName string) *ramen.ProtectedPVC {
	protectedPvc, _ := FindProtectedPvcAndIndex(vrg, pvcNamespaceName, pvcName)

	return protectedPvc
}

func (v *VRGInstance) pvcStatusDeleteIfPresent(pvcNamespaceName, pvcName string, log logr.Logger) {
	pvcStatus, i := FindProtectedPvcAndIndex(v.instance, pvcNamespaceName, pvcName)
	if pvcStatus == nil {
		log.Info("PVC status absent already")

		return
	}

	log.Info("PVC status delete", "index", i)
	v.instance.Status.ProtectedPVCs = sliceUnorderedElementDelete(v.instance.Status.ProtectedPVCs, i)
}

func sliceUnorderedElementDelete[T any](s []T, i int) []T {
	s[i] = s[len(s)-1]

	return s[:len(s)-1]
}

func FindProtectedPvcAndIndex(
	vrg *ramen.VolumeReplicationGroup, pvcNamespaceName, pvcName string,
) (*ramen.ProtectedPVC, int) {
	for index := range vrg.Status.ProtectedPVCs {
		protectedPVC := &vrg.Status.ProtectedPVCs[index]
		if protectedPVC.Namespace == pvcNamespaceName && protectedPVC.Name == pvcName {
			return protectedPVC, index
		}
	}

	return nil, len(vrg.Status.ProtectedPVCs)
}

func (v *VRGInstance) findFirstProtectedPVCWithName(pvcName string) *ramen.ProtectedPVC {
	for index := range v.instance.Status.ProtectedPVCs {
		protectedPVC := &v.instance.Status.ProtectedPVCs[index]
		if protectedPVC.Name == pvcName {
			return protectedPVC
		}
	}

	return nil
}

func (v *VRGInstance) vrgStatusPvcNamespacesSetIfUnset() {
	vrg := v.instance

	for i := range vrg.Status.ProtectedPVCs {
		pvc := &vrg.Status.ProtectedPVCs[i]
		log := v.log.WithValues("PVC", pvc.Name)

		if pvc.Namespace != "" {
			log.V(1).Info("VRG status PVC namespace set already", "namespace", pvc.Namespace)

			continue
		}

		log.Info("VRG status PVC namespace unset; setting")

		pvc.Namespace = vrg.GetNamespace()
	}
}