cozystack/internal/controller/workload_controller.go

package controller

import (
	"context"
	"strings"
	"time"

	corev1 "k8s.io/api/core/v1"
	apierrors "k8s.io/apimachinery/pkg/api/errors"
	"k8s.io/apimachinery/pkg/runtime"
	"k8s.io/apimachinery/pkg/types"
	ctrl "sigs.k8s.io/controller-runtime"
	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/log"

	cozyv1alpha1 "github.com/cozystack/cozystack/api/v1alpha1"
)

const (
	deletionRequeueDelay = 30 * time.Second
)

// WorkloadMonitorReconciler reconciles a WorkloadMonitor object
type WorkloadReconciler struct {
	client.Client
	Scheme *runtime.Scheme
}

// workload_controller.go
func (r *WorkloadReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
	log := log.FromContext(ctx)

	w := &cozyv1alpha1.Workload{}
	if err := r.Get(ctx, req.NamespacedName, w); err != nil {
		if apierrors.IsNotFound(err) {
			return ctrl.Result{}, nil
		}
		return ctrl.Result{}, err
	}

	// If my monitor is gone, delete me.
	monName, has := w.Labels["workloadmonitor.cozystack.io/name"]
	if !has {
		return ctrl.Result{}, r.Delete(ctx, w)
	}
	monitor := &cozyv1alpha1.WorkloadMonitor{}
	if err := r.Get(ctx, types.NamespacedName{Namespace: w.Namespace, Name: monName}, monitor); apierrors.IsNotFound(err) {
		return ctrl.Result{}, r.Delete(ctx, w)
	} else if err != nil {
		log.Error(err, "failed to get WorkloadMonitor", "monitor", monName)
		return ctrl.Result{}, err
	}

	return ctrl.Result{}, nil
}

// SetupWithManager registers our controller with the Manager and sets up watches.
func (r *WorkloadReconciler) SetupWithManager(mgr ctrl.Manager) error {
	return ctrl.NewControllerManagedBy(mgr).
		// Watch Workload objects
		For(&cozyv1alpha1.Workload{}).
		Complete(r)
}

func getMonitoredObject(w *cozyv1alpha1.Workload) client.Object {
	switch {
	case strings.HasPrefix(w.Name, "pvc-"):
		obj := &corev1.PersistentVolumeClaim{}
		obj.Name = strings.TrimPrefix(w.Name, "pvc-")
		obj.Namespace = w.Namespace
		return obj
	case strings.HasPrefix(w.Name, "svc-"):
		obj := &corev1.Service{}
		obj.Name = strings.TrimPrefix(w.Name, "svc-")
		obj.Namespace = w.Namespace
		return obj
	case strings.HasPrefix(w.Name, "pod-"):
		obj := &corev1.Pod{}
		obj.Name = strings.TrimPrefix(w.Name, "pod-")
		obj.Namespace = w.Namespace
		return obj
	}
	var obj client.Object
	return obj
}