feat: add new cluster phase ClusterRecovering (#2320)

Co-authored-by: wtaozzhang <wtaozzhang@tencent.com>
tkestack · Jun 6, 2024 · 3adc0f4 · 3adc0f4
1 parent e0e2b8b
commit 3adc0f4
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/api/platform/types.go b/api/platform/types.go
@@ -330,6 +330,8 @@ const (
 	ClusterUpscaling ClusterPhase = "Upscaling"
 	// ClusterDownscaling means the cluster is undergoing graceful down scaling.
 	ClusterDownscaling ClusterPhase = "Downscaling"
+	// ClusterRecovering means the cluster is recovering form confined.
+	ClusterRecovering ClusterPhase = "Recovering"
 )
 
 // ComponentPhase defines the phase of anywhere cluster component

diff --git a/api/platform/v1/types.go b/api/platform/v1/types.go
@@ -345,6 +345,8 @@ const (
 	ClusterUpscaling ClusterPhase = "Upscaling"
 	// ClusterDownscaling means the cluster is undergoing graceful down scaling.
 	ClusterDownscaling ClusterPhase = "Downscaling"
+	// ClusterRecovering means the cluster is recovering form confined.
+	ClusterRecovering ClusterPhase = "Recovering"
 )
 
 // ComponentPhase defines the phase of anywhere cluster component

diff --git a/pkg/platform/controller/cluster/cluster_controller.go b/pkg/platform/controller/cluster/cluster_controller.go
@@ -358,7 +358,7 @@ func (c *Controller) reconcile(ctx context.Context, key string, cluster *platfor
 		err = c.onUpdate(ctx, cluster)
 	case platformv1.ClusterUpscaling, platformv1.ClusterDownscaling:
 		err = c.onUpdate(ctx, cluster)
-	case platformv1.ClusterIdling, platformv1.ClusterConfined:
+	case platformv1.ClusterIdling, platformv1.ClusterConfined, platformv1.ClusterRecovering:
 		err = c.onUpdate(ctx, cluster)
 	case platformv1.ClusterTerminating:
 		log.FromContext(ctx).Info("Cluster has been terminated. Attempting to cleanup resources")
@@ -444,7 +444,8 @@ func (c *Controller) onUpdate(ctx context.Context, cluster *platformv1.Cluster)
 	if clusterWrapper.Status.Phase == platformv1.ClusterRunning ||
 		clusterWrapper.Status.Phase == platformv1.ClusterFailed ||
 		clusterWrapper.Status.Phase == platformv1.ClusterIdling ||
-		clusterWrapper.Status.Phase == platformv1.ClusterConfined {
+		clusterWrapper.Status.Phase == platformv1.ClusterConfined ||
+		clusterWrapper.Status.Phase == platformv1.ClusterRecovering {
 		err = provider.OnUpdate(ctx, clusterWrapper)
 		clusterWrapper = c.checkHealth(ctx, clusterWrapper)
 		if err != nil {