improve disableRestore

percona · Dec 20, 2024 · 37af2e4 · 37af2e4
1 parent 0e32ee7
commit 37af2e4
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/percona/controller/pgrestore/controller.go b/percona/controller/pgrestore/controller.go
@@ -245,10 +245,25 @@ func disableRestore(ctx context.Context, c client.Client, pg *v2.PerconaPGCluste
 	fvar := false
 	pg.Spec.Backups.PGBackRest.Restore.Enabled = &fvar
 
+	delete(pg.Annotations, naming.LabelPGBackRestRestore)
+
 	if err := c.Patch(ctx, pg, client.MergeFrom(orig)); err != nil {
 		return errors.Wrap(err, "patch PGCluster")
 	}
 
+	postgresCluster := new(v1beta1.PostgresCluster)
+	if err := c.Get(ctx, client.ObjectKeyFromObject(pg), postgresCluster); err != nil {
+		return errors.Wrap(err, "get PostgresCluster")
+	}
+
+	origPostgres := postgresCluster.DeepCopy()
+
+	postgresCluster.Status.PGBackRest.Restore = new(v1beta1.PGBackRestJobStatus)
+
+	if err := c.Status().Patch(ctx, postgresCluster, client.MergeFrom(origPostgres)); err != nil {
+		return errors.Wrap(err, "patch PGCluster")
+	}
+
 	return nil
 }