refactor: create separate reconciler for control flow Stage (#2848)

Signed-off-by: Hidde Beydals <hidde@hhh.computer>
akuity · Oct 28, 2024 · b2be266 · b2be266
1 parent a25dc52
commit b2be266
Show file tree

Hide file tree

Showing 14 changed files with 4,938 additions and 872 deletions.
diff --git a/cmd/controlplane/controller.go b/cmd/controlplane/controller.go
@@ -27,6 +27,7 @@ import (
 	"github.com/akuity/kargo/internal/credentials"
 	credsdb "github.com/akuity/kargo/internal/credentials/kubernetes"
 	"github.com/akuity/kargo/internal/directives"
+	"github.com/akuity/kargo/internal/indexer"
 	"github.com/akuity/kargo/internal/logging"
 	"github.com/akuity/kargo/internal/os"
 	"github.com/akuity/kargo/internal/types"
@@ -291,6 +292,7 @@ func (o *controllerOptions) setupReconcilers(
 	if argocdMgr != nil {
 		argoCDClient = argocdMgr.GetClient()
 	}
+	sharedIndexer := indexer.NewSharedFieldIndexer(kargoMgr.GetFieldIndexer())
 
 	directivesEngine := directives.NewSimpleEngine(credentialsDB, kargoMgr.GetClient(), argoCDClient)
 
@@ -310,10 +312,19 @@ func (o *controllerOptions) setupReconcilers(
 		argocdMgr,
 		directivesEngine,
 		stagesReconcilerCfg,
+		sharedIndexer,
 	); err != nil {
 		return fmt.Errorf("error setting up Stages reconciler: %w", err)
 	}
 
+	if err := stages.NewControlFlowStageReconciler(stagesReconcilerCfg).SetupWithManager(
+		ctx,
+		kargoMgr,
+		sharedIndexer,
+	); err != nil {
+		return fmt.Errorf("error setting up control flow Stages reconciler: %w", err)
+	}
+
 	if err := warehouses.SetupReconcilerWithManager(
 		kargoMgr,
 		credentialsDB,