44 lines
1.6 KiB
Go
44 lines
1.6 KiB
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
databasev1alpha1 "git.ddupan.top/panxiao81/ayatori/api/database/v1alpha1"
|
|
"git.ddupan.top/panxiao81/ayatori/internal/database/application"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/handler"
|
|
)
|
|
|
|
type ProvisioningReconciler struct {
|
|
Client client.Client
|
|
Service *application.DatabaseReconciliation
|
|
}
|
|
|
|
func NewProvisioningReconciler(cache client.Client, service *application.DatabaseReconciliation) *ProvisioningReconciler {
|
|
return &ProvisioningReconciler{Client: cache, Service: service}
|
|
}
|
|
|
|
func (r *ProvisioningReconciler) Reconcile(ctx context.Context, request ctrl.Request) (ctrl.Result, error) {
|
|
operationContext, cancel := context.WithTimeout(ctx, 30*time.Second)
|
|
defer cancel()
|
|
if err := r.Service.Reconcile(operationContext, request.Name); err != nil {
|
|
return ctrl.Result{}, err
|
|
}
|
|
return ctrl.Result{RequeueAfter: dependencyRetry}, nil
|
|
}
|
|
|
|
func (r *ProvisioningReconciler) SetupWithManager(manager ctrl.Manager) error {
|
|
if r.Client == nil || r.Service == nil {
|
|
return errors.New("provisioning controller requires injected client and use case")
|
|
}
|
|
return ctrl.NewControllerManagedBy(manager).Named("database-provisioning").
|
|
For(&databasev1alpha1.PostgreSQLDatabase{}).
|
|
Watches(&databasev1alpha1.PostgreSQLTenant{}, handler.EnqueueRequestsFromMapFunc(databaseRequestsForTenant)).
|
|
Watches(&databasev1alpha1.PostgreSQLInstance{}, handler.EnqueueRequestsFromMapFunc(func(ctx context.Context, object client.Object) []ctrl.Request {
|
|
return databaseRequestsForInstance(ctx, r.Client, object)
|
|
})).Complete(r)
|
|
}
|