sidero/app/caps-controller-manager/controllers/metalcluster_controller.go
Andrey Smirnov fbcd02a45d
feat: update for Talos 1.3.0
Rename to siderolabs, bump dependencies, controller-runtime, get rid of
netaddr, new SideroLink API, etc.

Use bootstrap cluster with a control plane + worker to avoid nasty
restarts when host-mode SideroLink IP pops up.

Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
2022-12-16 16:35:30 +04:00

111 lines
3.3 KiB
Go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
package controllers
import (
"context"
"fmt"
"github.com/go-logr/logr"
apierrors "k8s.io/apimachinery/pkg/api/errors"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/client-go/tools/record"
"sigs.k8s.io/cluster-api/util"
"sigs.k8s.io/cluster-api/util/patch"
ctrl "sigs.k8s.io/controller-runtime"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/controller"
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
"sigs.k8s.io/controller-runtime/pkg/reconcile"
infrav1 "github.com/siderolabs/sidero/app/caps-controller-manager/api/v1alpha3"
"github.com/siderolabs/sidero/app/caps-controller-manager/pkg/constants"
)
// MetalClusterReconciler reconciles a MetalCluster object.
type MetalClusterReconciler struct {
client.Client
Log logr.Logger
Scheme *runtime.Scheme
Recorder record.EventRecorder
}
// +kubebuilder:rbac:groups=infrastructure.cluster.x-k8s.io,resources=metalclusters,verbs=get;list;watch;create;update;patch;delete
// +kubebuilder:rbac:groups=infrastructure.cluster.x-k8s.io,resources=metalclusters/status,verbs=get;update;patch
// +kubebuilder:rbac:groups=cluster.x-k8s.io,resources=clusters;clusters/status,verbs=get;list;watch
func (r *MetalClusterReconciler) Reconcile(ctx context.Context, req ctrl.Request) (_ ctrl.Result, err error) {
log := r.Log.WithValues("metalcluster", req.NamespacedName)
// Fetch the metalCluster instance
metalCluster := &infrav1.MetalCluster{}
err = r.Get(ctx, req.NamespacedName, metalCluster)
if apierrors.IsNotFound(err) {
return reconcile.Result{}, nil
}
if err != nil {
return ctrl.Result{}, err
}
log = log.WithName(metalCluster.APIVersion)
// Fetch the Cluster
cluster, err := util.GetOwnerCluster(ctx, r.Client, metalCluster.ObjectMeta)
if err != nil {
return ctrl.Result{}, err
}
if cluster == nil {
log.Info("Cluster Controller has not yet set OwnerRef")
return ctrl.Result{RequeueAfter: constants.DefaultRequeueAfter}, nil
}
log = log.WithName(fmt.Sprintf("cluster=%s", cluster.Name))
// Initialize the patch helper
patchHelper, err := patch.NewHelper(metalCluster, r.Client)
if err != nil {
return ctrl.Result{}, err
}
// Always attempt to Patch the metalCluster object and status after each reconciliation.
defer func() {
if e := patchHelper.Patch(ctx, metalCluster); err != nil {
log.Error(err, "failed to patch metalCluster")
if e == nil {
err = e
}
}
}()
// If the MetalCluster doesn't have our finalizer, add it.
controllerutil.AddFinalizer(metalCluster, infrav1.ClusterFinalizer)
// Handle deleted machines
if !metalCluster.ObjectMeta.DeletionTimestamp.IsZero() {
log.Info("deleting cluster")
// Cluster is deleted so remove the finalizer.
controllerutil.RemoveFinalizer(metalCluster, infrav1.ClusterFinalizer)
return ctrl.Result{}, nil
}
metalCluster.Status.Ready = true
return ctrl.Result{}, nil
}
func (r *MetalClusterReconciler) SetupWithManager(ctx context.Context, mgr ctrl.Manager, options controller.Options) error {
return ctrl.NewControllerManagedBy(mgr).
WithOptions(options).
For(&infrav1.MetalCluster{}).
Complete(r)
}