mariadb-operator/pkg/controller/maxscale/controller.go

package maxscale

import (
	"context"
	"fmt"
	"time"

	mariadbv1alpha1 "github.com/mariadb-operator/mariadb-operator/api/v1alpha1"
	"github.com/mariadb-operator/mariadb-operator/pkg/builder"
	"github.com/mariadb-operator/mariadb-operator/pkg/environment"
	apierrors "k8s.io/apimachinery/pkg/api/errors"
	"k8s.io/utils/ptr"
	ctrl "sigs.k8s.io/controller-runtime"
	"sigs.k8s.io/controller-runtime/pkg/client"
)

type MaxScaleReconciler struct {
	client.Client
	builder     *builder.Builder
	environment *environment.Environment
}

func NewMaxScaleReconciler(client client.Client, builder *builder.Builder, env *environment.Environment) *MaxScaleReconciler {
	return &MaxScaleReconciler{
		Client:      client,
		builder:     builder,
		environment: env,
	}
}

func (r *MaxScaleReconciler) Reconcile(ctx context.Context, mdb *mariadbv1alpha1.MariaDB) (ctrl.Result, error) {
	if !ptr.Deref(mdb.Spec.MaxScale, mariadbv1alpha1.MariaDBMaxScaleSpec{}).Enabled {
		return ctrl.Result{}, nil
	}
	if !mdb.IsReady() {
		return ctrl.Result{RequeueAfter: 1 * time.Second}, nil
	}

	key := mdb.MaxScaleKey()
	desiredMxs, err := r.builder.BuildMaxScale(key, mdb, &mdb.Spec.MaxScale.MaxScaleBaseSpec)
	if err != nil {
		return ctrl.Result{}, fmt.Errorf("error building MaxScale: %v", err)
	}
	monitorModule := mariadbv1alpha1.MonitorModuleMariadb
	if mdb.Galera().Enabled {
		monitorModule = mariadbv1alpha1.MonitorModuleGalera
	}
	desiredMxs.Spec.Monitor.Module = monitorModule
	desiredMxs.Spec.Auth.Generate.Enabled = true
	desiredMxs.SetDefaults(r.environment)

	var existingMxs mariadbv1alpha1.MaxScale
	if err := r.Get(ctx, key, &existingMxs); err != nil {
		if !apierrors.IsNotFound(err) {
			return ctrl.Result{}, err
		}
		return ctrl.Result{}, r.Create(ctx, desiredMxs)
	}

	patch := client.MergeFrom(existingMxs.DeepCopy())
	existingMxs.Spec.MaxScaleBaseSpec = desiredMxs.Spec.MaxScaleBaseSpec
	return ctrl.Result{}, r.Patch(ctx, &existingMxs, patch)
}