103 lines
3.6 KiB
Go
103 lines
3.6 KiB
Go
/*
|
|
Copyright 2026.
|
|
|
|
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 controller
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"reflect"
|
|
"time"
|
|
|
|
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
logf "sigs.k8s.io/controller-runtime/pkg/log"
|
|
|
|
databasev1alpha1 "git.ddupan.top/panxiao81/postgresql-tenant-operator/api/v1alpha1"
|
|
)
|
|
|
|
// PostgreSQLInstanceReconciler reconciles a PostgreSQLInstance object
|
|
type PostgreSQLInstanceReconciler struct {
|
|
client.Client
|
|
Scheme *runtime.Scheme
|
|
Instances PostgreSQLInstances
|
|
Timeout time.Duration
|
|
}
|
|
|
|
// PostgreSQLInstances is the external dependency boundary used by the Instance state machine.
|
|
type PostgreSQLInstances interface {
|
|
Validate(context.Context, *databasev1alpha1.PostgreSQLInstance) (string, error)
|
|
InitializeRegistry(context.Context, *databasev1alpha1.PostgreSQLInstance) (string, error)
|
|
Forget(string)
|
|
}
|
|
|
|
// +kubebuilder:rbac:groups=database.ddupan.top,resources=postgresqlinstances,verbs=get;list;watch;create;update;patch;delete
|
|
// +kubebuilder:rbac:groups=database.ddupan.top,resources=postgresqlinstances/status,verbs=get;update;patch
|
|
// +kubebuilder:rbac:groups=database.ddupan.top,resources=postgresqlinstances/finalizers,verbs=update
|
|
|
|
// Reconcile is part of the main kubernetes reconciliation loop which aims to
|
|
// move the current state of the cluster closer to the desired state.
|
|
// TODO(user): Modify the Reconcile function to compare the state specified by
|
|
// the PostgreSQLInstance object against the actual cluster state, and then
|
|
// perform operations to make the cluster state reflect the state specified by
|
|
// the user.
|
|
//
|
|
// For more details, check Reconcile and its Result here:
|
|
// - https://pkg.go.dev/sigs.k8s.io/[email protected]/pkg/reconcile
|
|
func (r *PostgreSQLInstanceReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
|
|
logger := logf.FromContext(ctx)
|
|
instance := &databasev1alpha1.PostgreSQLInstance{}
|
|
if err := r.Get(ctx, req.NamespacedName, instance); err != nil {
|
|
if apierrors.IsNotFound(err) {
|
|
r.Instances.Forget(req.Name)
|
|
}
|
|
return ctrl.Result{}, client.IgnoreNotFound(err)
|
|
}
|
|
|
|
before := instance.DeepCopy()
|
|
operationCtx := ctx
|
|
if r.Timeout > 0 {
|
|
var cancel context.CancelFunc
|
|
operationCtx, cancel = context.WithTimeout(ctx, r.Timeout)
|
|
defer cancel()
|
|
}
|
|
result, reconcileErr := newInstanceStateMachine(r.Instances).reconcile(operationCtx, instance)
|
|
|
|
if !reflect.DeepEqual(before.Status, instance.Status) {
|
|
if err := r.Status().Patch(ctx, instance, client.MergeFrom(before)); err != nil {
|
|
if apierrors.IsConflict(err) {
|
|
logger.V(1).Info("instance status changed concurrently; retrying")
|
|
}
|
|
return ctrl.Result{}, err
|
|
}
|
|
}
|
|
|
|
return result, reconcileErr
|
|
}
|
|
|
|
// SetupWithManager sets up the controller with the Manager.
|
|
func (r *PostgreSQLInstanceReconciler) SetupWithManager(mgr ctrl.Manager) error {
|
|
if r.Instances == nil {
|
|
return fmt.Errorf("instance service must be injected before controller setup")
|
|
}
|
|
return ctrl.NewControllerManagedBy(mgr).
|
|
For(&databasev1alpha1.PostgreSQLInstance{}).
|
|
Named("postgresqlinstance").
|
|
Complete(r)
|
|
}
|