feat(RedisCluster): add aws redis boilerplate

kyma-project · Feb 3, 2025 · 6c965bd · 6c965bd
1 parent 94bf79a
commit 6c965bd
Show file tree

Hide file tree

Showing 4 changed files with 91 additions and 0 deletions.
diff --git a/internal/controller/cloud-control/rediscluster_controller.go b/internal/controller/cloud-control/rediscluster_controller.go
@@ -23,6 +23,7 @@ import (
 	"github.com/kyma-project/cloud-manager/pkg/common/abstractions"
 	"github.com/kyma-project/cloud-manager/pkg/common/actions/focal"
 	"github.com/kyma-project/cloud-manager/pkg/composed"
+	awsrediscluster "github.com/kyma-project/cloud-manager/pkg/kcp/provider/aws/rediscluster"
 	gcprediscluster "github.com/kyma-project/cloud-manager/pkg/kcp/provider/gcp/rediscluster"
 	"github.com/kyma-project/cloud-manager/pkg/kcp/rediscluster"
 	ctrl "sigs.k8s.io/controller-runtime"
@@ -43,6 +44,7 @@ func SetupRedisClusterReconciler(
 			composed.NewStateFactory(composed.NewStateClusterFromCluster(kcpManager)),
 			focal.NewStateFactory(),
 			gcprediscluster.NewStateFactory(env),
+			awsrediscluster.NewStateFactory(),
 		),
 	).SetupWithManager(kcpManager)
 }

diff --git a/pkg/kcp/provider/aws/rediscluster/new.go b/pkg/kcp/provider/aws/rediscluster/new.go
@@ -0,0 +1,51 @@
+package rediscluster
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/kyma-project/cloud-manager/pkg/common/actions"
+	"github.com/kyma-project/cloud-manager/pkg/kcp/rediscluster/types"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"github.com/kyma-project/cloud-manager/api/cloud-control/v1beta1"
+	cloudcontrolv1beta1 "github.com/kyma-project/cloud-manager/api/cloud-control/v1beta1"
+	"github.com/kyma-project/cloud-manager/pkg/composed"
+)
+
+func New(stateFactory StateFactory) composed.Action {
+	return func(ctx context.Context, st composed.State) (error, context.Context) {
+
+		state, err := stateFactory.NewState(ctx, st.(types.State))
+		if err != nil {
+			composed.LoggerFromCtx(ctx).Error(err, "Failed to bootstrap AWS RedisCluster state")
+			redisCluster := st.Obj().(*v1beta1.RedisCluster)
+			redisCluster.Status.State = cloudcontrolv1beta1.StateError
+			return composed.UpdateStatus(redisCluster).
+				SetExclusiveConditions(metav1.Condition{
+					Type:    v1beta1.ConditionTypeError,
+					Status:  metav1.ConditionTrue,
+					Reason:  v1beta1.ReasonCloudProviderError,
+					Message: "Failed to create RedisCluster state",
+				}).
+				SuccessError(composed.StopAndForget).
+				SuccessLogMsg(fmt.Sprintf("Error creating new AWS RedisCluster state: %s", err)).
+				Run(ctx, st)
+		}
+
+		return composed.ComposeActions(
+			"redisCluster",
+			actions.AddFinalizer,
+			composed.IfElse(composed.Not(composed.MarkedForDeletionPredicate),
+				composed.ComposeActions(
+					"redisCluster-create",
+				),
+				composed.ComposeActions(
+					"redisCluster-delete",
+					actions.RemoveFinalizer,
+				),
+			),
+			composed.StopAndForgetAction,
+		)(ctx, state)
+	}
+}
diff --git a/pkg/kcp/provider/aws/rediscluster/state.go b/pkg/kcp/provider/aws/rediscluster/state.go
@@ -0,0 +1,32 @@
+package rediscluster
+
+import (
+	"context"
+
+	"github.com/kyma-project/cloud-manager/pkg/kcp/rediscluster/types"
+)
+
+type State struct {
+	types.State
+}
+
+type StateFactory interface {
+	NewState(ctx context.Context, redisClusterState types.State) (*State, error)
+}
+
+type stateFactory struct{}
+
+func NewStateFactory() StateFactory {
+	return &stateFactory{}
+}
+
+func (statefactory *stateFactory) NewState(ctx context.Context, redisClusterState types.State) (*State, error) {
+
+	return newState(redisClusterState), nil
+}
+
+func newState(redisClusterState types.State) *State {
+	return &State{
+		State: redisClusterState,
+	}
+}
diff --git a/pkg/kcp/rediscluster/reconciler.go b/pkg/kcp/rediscluster/reconciler.go
@@ -8,7 +8,9 @@ import (
 	cloudcontrolv1beta1 "github.com/kyma-project/cloud-manager/api/cloud-control/v1beta1"
 	"github.com/kyma-project/cloud-manager/pkg/common/actions/focal"
 	"github.com/kyma-project/cloud-manager/pkg/composed"
+	awsrediscluster "github.com/kyma-project/cloud-manager/pkg/kcp/provider/aws/rediscluster"
 	gcprediscluster "github.com/kyma-project/cloud-manager/pkg/kcp/provider/gcp/rediscluster"
+
 	"k8s.io/apimachinery/pkg/types"
 	ctrl "sigs.k8s.io/controller-runtime"
 )
@@ -22,17 +24,20 @@ type redisClusterReconciler struct {
 	focalStateFactory    focal.StateFactory
 
 	gcpStateFactory gcprediscluster.StateFactory
+	awsStateFactory awsrediscluster.StateFactory
 }
 
 func NewRedisClusterReconciler(
 	composedStateFactory composed.StateFactory,
 	focalStateFactory focal.StateFactory,
 	gcpStateFactory gcprediscluster.StateFactory,
+	awsStateFactory awsrediscluster.StateFactory,
 ) RedisClusterReconciler {
 	return &redisClusterReconciler{
 		composedStateFactory: composedStateFactory,
 		focalStateFactory:    focalStateFactory,
 		gcpStateFactory:      gcpStateFactory,
+		awsStateFactory:      awsStateFactory,
 	}
 }
 
@@ -59,6 +64,7 @@ func (r *redisClusterReconciler) newAction() composed.Action {
 					"providerSwitch",
 					nil,
 					composed.NewCase(focal.GcpProviderPredicate, gcprediscluster.New(r.gcpStateFactory)),
+					composed.NewCase(focal.AwsProviderPredicate, awsrediscluster.New(r.awsStateFactory)),
 				),
 			)(ctx, newState(st.(focal.State)))
 		},