fix: get running its in load resource

Author: kizuna-lek

controllers/apps/component/transformer_component_load_resources.go

@@ -27,6 +27,7 @@ import (
 	"sigs.k8s.io/controller-runtime/pkg/client"
 
 	appsv1 "github.com/apecloud/kubeblocks/apis/apps/v1"
+	workloads "github.com/apecloud/kubeblocks/apis/workloads/v1"
 	appsutil "github.com/apecloud/kubeblocks/controllers/apps/util"
 	"github.com/apecloud/kubeblocks/pkg/controller/component"
 	"github.com/apecloud/kubeblocks/pkg/controller/graph"
@@ -72,6 +73,12 @@ func (t *componentLoadResourcesTransformer) transformForNativeComponent(transCtx
 	}
 	transCtx.SynthesizeComponent = synthesizedComp
 
+	runningITS, err := t.runningInstanceSetObject(transCtx, synthesizedComp)
+	if err != nil {
+		return err
+	}
+	transCtx.RunningWorkload = runningITS
+
 	return nil
 }
 
@@ -91,3 +98,16 @@ func getNCheckCompDefinition(ctx context.Context, cli client.Reader, name string
 	}
 	return compDef, nil
 }
+
+func (t *componentLoadResourcesTransformer) runningInstanceSetObject(ctx *componentTransformContext,
+	synthesizeComp *component.SynthesizedComponent) (*workloads.InstanceSet, error) {
+	objs, err := component.ListOwnedWorkloads(ctx.GetContext(), ctx.GetClient(),
+		synthesizeComp.Namespace, synthesizeComp.ClusterName, synthesizeComp.Name)
+	if err != nil {
+		return nil, err
+	}
+	if len(objs) == 0 {
+		return nil, nil
+	}
+	return objs[0], nil
+}

controllers/apps/component/transformer_component_workload.go

@@ -65,11 +65,10 @@ func (t *componentWorkloadTransformer) Transform(ctx graph.TransformContext, dag
 	comp := transCtx.Component
 	synthesizeComp := transCtx.SynthesizeComponent
 
-	runningITS, err := t.runningInstanceSetObject(ctx, synthesizeComp)
-	if err != nil {
-		return err
+	var runningITS *workloads.InstanceSet
+	if transCtx.RunningWorkload != nil {
+		runningITS = transCtx.RunningWorkload.(*workloads.InstanceSet)
 	}
-	transCtx.RunningWorkload = runningITS
 
 	protoITS, err := factory.BuildInstanceSet(synthesizeComp, compDef)
 	if err != nil {
@@ -100,19 +99,6 @@ func (t *componentWorkloadTransformer) Transform(ctx graph.TransformContext, dag
 	return err
 }
 
-func (t *componentWorkloadTransformer) runningInstanceSetObject(ctx graph.TransformContext,
-	synthesizeComp *component.SynthesizedComponent) (*workloads.InstanceSet, error) {
-	objs, err := component.ListOwnedWorkloads(ctx.GetContext(), ctx.GetClient(),
-		synthesizeComp.Namespace, synthesizeComp.ClusterName, synthesizeComp.Name)
-	if err != nil {
-		return nil, err
-	}
-	if len(objs) == 0 {
-		return nil, nil
-	}
-	return objs[0], nil
-}
-
 func (t *componentWorkloadTransformer) reconcileWorkload(ctx context.Context, cli client.Reader,
 	synthesizedComp *component.SynthesizedComponent, comp *appsv1.Component, runningITS, protoITS *workloads.InstanceSet) error {
 	// if runningITS already exists, the image changes in protoITS will be