simplify code

Author: cjc7373

controllers/apps/component/transformer_component_status.go

@@ -31,6 +31,7 @@ import (
 	"k8s.io/apimachinery/pkg/api/meta"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/util/sets"
+	"k8s.io/client-go/tools/record"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 
 	appsv1 "github.com/apecloud/kubeblocks/apis/apps/v1"
@@ -304,169 +305,142 @@ func (t *componentStatusTransformer) updateComponentStatus(transCtx *componentTr
 }
 
 func (t *componentStatusTransformer) reconcileStatusCondition(transCtx *componentTransformContext) error {
-	t.reconcileAvailableCondition(transCtx)
-	err1 := t.reconcileHasFailureCondition(transCtx)
-	err2 := t.reconcileUpdatingCondition(transCtx)
-	t.reconcileWorkloadRunningCondition(transCtx)
-	return errors.Join(err1, err2)
+	err1 := t.reconcileAvailableCondition(transCtx)
+	err2 := t.reconcileHasFailureCondition(transCtx)
+	err3 := t.reconcileUpdatingCondition(transCtx)
+	err4 := t.reconcileWorkloadRunningCondition(transCtx)
+	return errors.Join(err1, err2, err3, err4)
 }
 
-func (t *componentStatusTransformer) reconcileHasFailureCondition(transCtx *componentTransformContext) error {
-	status := metav1.ConditionFalse
-	reason := "NoFailure"
-	message := ""
-
-	hasFailedPod, messages := t.hasFailedPod()
-
-	if hasFailedPod {
-		status = metav1.ConditionTrue
-		reason = "PodFailure"
-		message = "component has failed pod(s)"
-		for _, msg := range messages {
-			message += "; " + msg
-		}
-	}
-
-	_, hasFailedScaleOut, err := t.hasScaleOutRunning(transCtx)
+func (t *componentStatusTransformer) checkNSetCondition(
+	eventRecorder record.EventRecorder,
+	conditionType string,
+	checker func() (err error, status metav1.ConditionStatus, reason, message string),
+) error {
+	err, status, reason, message := checker()
 	if err != nil {
 		return err
 	}
-
-	if hasFailedScaleOut {
-		status = metav1.ConditionTrue
-		reason = "ScaleOutFailure"
-		message = "component scale out has failure"
-	}
-
 	cond := metav1.Condition{
-		Type:               appsv1.ConditionTypeHasFailure,
+		Type:               conditionType,
 		Status:             status,
 		ObservedGeneration: t.comp.Generation,
 		Reason:             reason,
 		Message:            message,
 	}
 	if meta.SetStatusCondition(&t.comp.Status.Conditions, cond) {
-		transCtx.EventRecorder.Event(t.comp, corev1.EventTypeNormal, reason, message)
+		eventRecorder.Event(t.comp, corev1.EventTypeNormal, reason, message)
 	}
 	return nil
 }
 
-func (t *componentStatusTransformer) reconcileUpdatingCondition(transCtx *componentTransformContext) error {
-	status := metav1.ConditionFalse
-	reason := "NotUpdating"
-	message := ""
-
-	// check if the component scale out failed
-	hasRunningScaleOut, _, err := t.hasScaleOutRunning(transCtx)
-	if err != nil {
-		return err
-	}
-	if hasRunningScaleOut {
-		status = metav1.ConditionTrue
-		reason = "ScaleOutRunning"
-		message = "component scale out is running"
-	}
+func (t *componentStatusTransformer) reconcileHasFailureCondition(transCtx *componentTransformContext) error {
+	return t.checkNSetCondition(
+		transCtx.EventRecorder,
+		appsv1.ConditionTypeHasFailure,
+		func() (err error, status metav1.ConditionStatus, reason string, message string) {
+			hasFailedPod, messages := t.hasFailedPod()
+			if hasFailedPod {
+				message = "component has failed pod(s)"
+				for _, msg := range messages {
+					message += "; " + msg
+				}
+				return nil, metav1.ConditionTrue, "PodFailure", message
+			}
 
-	// check if the volume expansion is running
-	hasRunningVolumeExpansion := t.hasVolumeExpansionRunning()
-	if hasRunningVolumeExpansion {
-		status = metav1.ConditionTrue
-		reason = "VolumeExpansionRunning"
-		message = "component volume expansion is running"
-	}
+			_, hasFailedScaleOut, err := t.hasScaleOutRunning(transCtx)
+			if err != nil {
+				return err, "", "", ""
+			}
+			if hasFailedScaleOut {
+				return nil, metav1.ConditionTrue, "ScaleOutFailure", "component scale out has failure"
+			}
 
-	cond := metav1.Condition{
-		Type:               appsv1.ConditionTypeUpdating,
-		Status:             status,
-		ObservedGeneration: t.comp.Generation,
-		Reason:             reason,
-		Message:            message,
-	}
-	if meta.SetStatusCondition(&t.comp.Status.Conditions, cond) {
-		transCtx.EventRecorder.Event(t.comp, corev1.EventTypeNormal, reason, message)
-	}
-	return nil
+			return nil, metav1.ConditionFalse, "NoFailure", ""
+		},
+	)
 }
 
-func (t *componentStatusTransformer) reconcileWorkloadRunningCondition(transCtx *componentTransformContext) {
-	status := metav1.ConditionTrue
-	reason := "WorkloadRunning"
-	message := ""
-	comp := t.comp
+func (t *componentStatusTransformer) reconcileUpdatingCondition(transCtx *componentTransformContext) error {
+	return t.checkNSetCondition(
+		transCtx.EventRecorder,
+		appsv1.ConditionTypeUpdating,
+		func() (err error, status metav1.ConditionStatus, reason string, message string) {
+			hasRunningScaleOut, _, err := t.hasScaleOutRunning(transCtx)
+			if err != nil {
+				return err, "", "", ""
+			}
+			if hasRunningScaleOut {
+				status = metav1.ConditionTrue
+				reason = "ScaleOutRunning"
+				message = "component scale out is running"
+				return nil, status, reason, message
+			}
 
-	switch {
-	case t.runningITS == nil:
-		status = metav1.ConditionFalse
-		reason = "WorkloadNotExist"
-		message = "waiting for workload to be created"
-	case !t.isWorkloadUpdated():
-		status = metav1.ConditionFalse
-		reason = "WorkloadNotUpdated"
-		message = "observed workload's generation not matching component's"
-	case !t.runningITS.IsInstanceSetReady():
-		status = metav1.ConditionFalse
-		reason = "WorkloadNotReady"
-		message = "workload not ready"
-	}
+			hasRunningVolumeExpansion := t.hasVolumeExpansionRunning()
+			if hasRunningVolumeExpansion {
+				status = metav1.ConditionTrue
+				reason = "VolumeExpansionRunning"
+				message = "component volume expansion is running"
+				return nil, status, reason, message
+			}
 
-	cond := metav1.Condition{
-		Type:               appsv1.ConditionTypeWorkloadRunning,
-		Status:             status,
-		ObservedGeneration: comp.Generation,
-		Reason:             reason,
-		Message:            message,
-	}
-	if meta.SetStatusCondition(&comp.Status.Conditions, cond) {
-		transCtx.EventRecorder.Event(comp, corev1.EventTypeNormal, reason, message)
-	}
+			return nil, metav1.ConditionFalse, "NotUpdating", ""
+		},
+	)
 }
 
-func (t *componentStatusTransformer) reconcileAvailableCondition(transCtx *componentTransformContext) {
+func (t *componentStatusTransformer) reconcileWorkloadRunningCondition(transCtx *componentTransformContext) error {
+	return t.checkNSetCondition(
+		transCtx.EventRecorder,
+		appsv1.ConditionTypeWorkloadRunning,
+		func() (err error, status metav1.ConditionStatus, reason string, message string) {
+			status = metav1.ConditionTrue
+			reason = "WorkloadRunning"
+			message = ""
+
+			switch {
+			case t.runningITS == nil:
+				status = metav1.ConditionFalse
+				reason = "WorkloadNotExist"
+				message = "waiting for workload to be created"
+			case !t.isWorkloadUpdated():
+				status = metav1.ConditionFalse
+				reason = "WorkloadNotUpdated"
+				message = "observed workload's generation not matching component's"
+			case !t.runningITS.IsInstanceSetReady():
+				status = metav1.ConditionFalse
+				reason = "WorkloadNotReady"
+				message = "workload not ready"
+			}
+
+			return nil, status, reason, message
+		},
+	)
+}
+
+func (t *componentStatusTransformer) reconcileAvailableCondition(transCtx *componentTransformContext) error {
 	policy := component.GetComponentAvailablePolicy(transCtx.CompDef)
 	if policy.WithPhases == nil && policy.WithRole == nil {
-		return
+		return nil
 	}
 
-	var (
-		comp    = transCtx.Component
-		reason  string
-		message string
-	)
-	status := metav1.ConditionTrue
-	if policy.WithPhases != nil {
-		status1, message1 := t.availableWithPhases(transCtx, comp, policy)
-		if status1 != metav1.ConditionTrue {
-			status = status1
-			reason = "PhaseCheckFail"
-		}
-		message += fmt.Sprintf("phase check: %s. ", message1)
-	}
-	if policy.WithRole != nil {
-		status2, message2 := t.availableWithRole(transCtx, comp, policy)
-		if status2 != metav1.ConditionTrue {
-			status = status2
-			if reason != "" {
-				reason = "PhaseAndRoleCheckFail"
-			} else {
-				reason = "RoleCheckFail"
+	return t.checkNSetCondition(
+		transCtx.EventRecorder,
+		appsv1.ConditionTypeAvailable,
+		func() (err error, status metav1.ConditionStatus, reason string, message string) {
+			if policy.WithPhases != nil {
+				status, message := t.availableWithPhases(transCtx, transCtx.Component, policy)
+				return nil, status, "PhaseCheckFail", message
+			}
+			if policy.WithRole != nil {
+				status, message := t.availableWithRole(transCtx, transCtx.Component, policy)
+				return nil, status, "RoleCheckFail", message
 			}
-		}
-		message += fmt.Sprintf("role check: %s. ", message2)
-	}
-	if reason == "" {
-		reason = "Available"
-	}
 
-	cond := metav1.Condition{
-		Type:               appsv1.ConditionTypeAvailable,
-		Status:             status,
-		ObservedGeneration: comp.Generation,
-		Reason:             reason,
-		Message:            message,
-	}
-	if meta.SetStatusCondition(&comp.Status.Conditions, cond) {
-		transCtx.EventRecorder.Event(comp, corev1.EventTypeNormal, reason, message)
-	}
+			return nil, metav1.ConditionTrue, "Available", ""
+		},
+	)
 }
 
 func (t *componentStatusTransformer) availableWithPhases(_ *componentTransformContext,