feat: 接通 PostgreSQL 角色与数据库创建闭环
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
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)
|
||||
}
|
||||
Reference in New Issue
Block a user