account-operator/internal/controller/accountinfo_controller.go at 1d0f739c3ff5ab20e55c3d55cbceb5900c560af0 · platform-mesh/account-operator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
package controller

import (
	"context"

	"github.com/platform-mesh/subroutines"
	"github.com/platform-mesh/subroutines/lifecycle"
	"sigs.k8s.io/controller-runtime/pkg/client"
	"sigs.k8s.io/controller-runtime/pkg/predicate"
	"sigs.k8s.io/controller-runtime/pkg/reconcile"
	mcbuilder "sigs.k8s.io/multicluster-runtime/pkg/builder"
	mcmanager "sigs.k8s.io/multicluster-runtime/pkg/manager"
	mcreconcile "sigs.k8s.io/multicluster-runtime/pkg/reconcile"

	"github.com/platform-mesh/account-operator/api/v1alpha1"
	"github.com/platform-mesh/account-operator/internal/config"
	"github.com/platform-mesh/account-operator/pkg/subroutines/finalizeaccountinfo"
)

const accountInfoReconcilerName = "AccountInfoReconciler"

// AccountInfoReconciler orchestrates AccountInfo resources across logical clusters.
type AccountInfoReconciler struct {
	cfg       config.OperatorConfig
	lifecycle *lifecycle.Lifecycle
}

func NewAccountInfoReconciler(mgr mcmanager.Manager, cfg config.OperatorConfig) *AccountInfoReconciler { // coverage-ignore
	subs := []subroutines.Subroutine{}

	if cfg.Controllers.AccountInfo.Enabled {
		subs = append(subs, finalizeaccountinfo.New(mgr))
	}

	lc := lifecycle.New(mgr, accountInfoReconcilerName, func() client.Object { return &v1alpha1.AccountInfo{} }, subs...)

	return &AccountInfoReconciler{
		cfg:       cfg,
		lifecycle: lc,
	}
}

func (r *AccountInfoReconciler) SetupWithManager(mgr mcmanager.Manager, eventPredicates ...predicate.Predicate) error { // coverage-ignore
	builder := mcbuilder.ControllerManagedBy(mgr).
		Named(accountInfoReconcilerName).
		For(&v1alpha1.AccountInfo{})

	for _, p := range eventPredicates {
		builder = builder.WithEventFilter(p)
	}

	return builder.Complete(r)
}

func (r *AccountInfoReconciler) Reconcile(ctx context.Context, req mcreconcile.Request) (reconcile.Result, error) { // coverage-ignore
	return r.lifecycle.Reconcile(ctx, req)
}