fix: lookup rebalancing scheduled job by schedule ID if deleted or re…

…created

castai/resource_rebalancing_job.go

@@ -83,12 +83,31 @@ func resourceRebalancingJobCreate(ctx context.Context, d *schema.ResourceData, m
 }
 
 func resourceRebalancingJobRead(ctx context.Context, d *schema.ResourceData, meta any) diag.Diagnostics {
+	client := meta.(*ProviderConfig).api
 	clusterID := d.Get("cluster_id").(string)
-	job, found, err := getRebalancingJobById(ctx, meta.(*ProviderConfig).api, clusterID, d.Id())
+	job, found, err := getRebalancingJobById(ctx, client, clusterID, d.Id())
 	if err != nil {
 		return diag.FromErr(err)
 	}
 	if !d.IsNewResource() && !found {
+		rebalancingScheduleID := d.Get("rebalancing_schedule_id").(string)
+		params := &sdk.ScheduledRebalancingAPIListRebalancingJobsParams{
+			RebalancingScheduleId: lo.ToPtr(rebalancingScheduleID),
+		}
+		listResp, err := client.ScheduledRebalancingAPIListRebalancingJobsWithResponse(ctx, clusterID, params)
+		if checkErr := sdk.CheckOKResponse(listResp, err); checkErr != nil {
+			return diag.FromErr(checkErr)
+		}
+		for _, j := range *listResp.JSON200.Jobs {
+			if *j.RebalancingScheduleId == rebalancingScheduleID {
+				if err := rebalancingJobToState(&j, d); err != nil {
+					return diag.FromErr(err)
+				}
+
+				return nil
+			}
+		}
+
 		tflog.Warn(ctx, "Rebalancing job not found, removing from state", map[string]any{"id": d.Id()})
 		d.SetId("")
 		return nil
@@ -189,7 +208,10 @@ func getRebalancingJobByScheduleName(ctx context.Context, client *sdk.ClientWith
 		return nil, fmt.Errorf("getting schedule: %w", err)
 	}
 
-	resp, err := client.ScheduledRebalancingAPIListRebalancingJobsWithResponse(ctx, clusterID)
+	params := sdk.ScheduledRebalancingAPIListRebalancingJobsParams{
+		RebalancingScheduleId: schedule.Id,
+	}
+	resp, err := client.ScheduledRebalancingAPIListRebalancingJobsWithResponse(ctx, clusterID, &params)
 	if checkErr := sdk.CheckOKResponse(resp, err); checkErr != nil {
 		return nil, checkErr
 	}