K8SPG-560: Create bootstrap restore object (#888)

* Create bootstrap restore object.

* Update demand-backup test.

* Cleanup

* Update tests.

* Update rbac.

* Improve

* Fix

Author: inelpandzic

config/rbac/cluster/role.yaml

@@ -163,6 +163,7 @@ rules:
   resources:
   - perconapgrestores
   verbs:
+  - create
   - get
   - list
   - watch

config/rbac/namespace/role.yaml

@@ -163,6 +163,7 @@ rules:
   resources:
   - perconapgrestores
   verbs:
+  - create
   - get
   - list
   - watch

deploy/bundle.yaml

@@ -45581,6 +45581,7 @@ rules:
   resources:
   - perconapgrestores
   verbs:
+  - create
   - get
   - list
   - watch

deploy/cw-bundle.yaml

@@ -45581,6 +45581,7 @@ rules:
   resources:
   - perconapgrestores
   verbs:
+  - create
   - get
   - list
   - watch

deploy/cw-rbac.yaml

@@ -167,6 +167,7 @@ rules:
   resources:
   - perconapgrestores
   verbs:
+  - create
   - get
   - list
   - watch

deploy/rbac.yaml

@@ -167,6 +167,7 @@ rules:
   resources:
   - perconapgrestores
   verbs:
+  - create
   - get
   - list
   - watch

e2e-tests/tests/start-from-backup/01-assert.yaml

@@ -107,3 +107,14 @@ status:
     ready: 3
     size: 3
   state: ready
+---
+apiVersion: pgv2.percona.com/v2
+kind: PerconaPGRestore
+metadata:
+  name: start-from-backup-bootstrap
+spec:
+  pgCluster: start-from-backup
+  repoName: repo1
+status:
+  state: Succeeded
+

percona/controller/pgcluster/controller.go

@@ -225,6 +225,10 @@ func (r *PGClusterReconciler) Reconcile(ctx context.Context, request reconcile.R
 		return reconcile.Result{}, errors.Wrap(err, "reconcile backups")
 	}
 
+	if err := r.createBootstrapRestoreObject(ctx, cr); err != nil {
+		return reconcile.Result{}, errors.Wrap(err, "reconcile restore")
+	}
+
 	if err := r.reconcilePMM(ctx, cr); err != nil {
 		return reconcile.Result{}, errors.Wrap(err, "failed to add pmm sidecar")
 	}

percona/controller/pgcluster/restore.go

@@ -0,0 +1,51 @@
+package pgcluster
+
+import (
+	"context"
+
+	k8serrors "k8s.io/apimachinery/pkg/api/errors"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	pNaming "github.com/percona/percona-postgresql-operator/percona/naming"
+	v2 "github.com/percona/percona-postgresql-operator/pkg/apis/pgv2.percona.com/v2"
+)
+
+// createBootstrapRestoreObject creates a PerconaPGRestore object for the bootstrap restore
+func (r *PGClusterReconciler) createBootstrapRestoreObject(ctx context.Context, cr *v2.PerconaPGCluster) error {
+	if cr.Spec.DataSource == nil || (cr.Spec.DataSource.PGBackRest == nil &&
+		cr.Spec.DataSource.PostgresCluster == nil &&
+		cr.Spec.DataSource.Volumes == nil) {
+		return nil
+	}
+
+	pgrName := cr.Name + "-bootstrap"
+	repoName := ""
+
+	if cr.Spec.DataSource.PGBackRest != nil {
+		repoName = cr.Spec.DataSource.PGBackRest.Repo.Name
+	}
+	if cr.Spec.DataSource.PostgresCluster != nil {
+		repoName = cr.Spec.DataSource.PostgresCluster.RepoName
+	}
+
+	pgr := &v2.PerconaPGRestore{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      pgrName,
+			Namespace: cr.Namespace,
+			Annotations: map[string]string{
+				pNaming.AnnotationClusterBootstrapRestore: "true",
+			},
+		},
+		Spec: v2.PerconaPGRestoreSpec{
+			PGCluster: cr.Name,
+			RepoName:  repoName,
+		},
+	}
+
+	err := r.Client.Create(ctx, pgr)
+	if err != nil && !k8serrors.IsAlreadyExists(err) {
+		return err
+	}
+
+	return nil
+}

percona/controller/pgrestore/controller.go

@@ -18,6 +18,7 @@ import (
 	"github.com/percona/percona-postgresql-operator/internal/logging"
 	"github.com/percona/percona-postgresql-operator/internal/naming"
 	"github.com/percona/percona-postgresql-operator/percona/controller"
+	pNaming "github.com/percona/percona-postgresql-operator/percona/naming"
 	v2 "github.com/percona/percona-postgresql-operator/pkg/apis/pgv2.percona.com/v2"
 	"github.com/percona/percona-postgresql-operator/pkg/apis/postgres-operator.crunchydata.com/v1beta1"
 )
@@ -40,7 +41,7 @@ func (r *PGRestoreReconciler) SetupWithManager(mgr manager.Manager) error {
 	return builder.ControllerManagedBy(mgr).For(&v2.PerconaPGRestore{}).Complete(r)
 }
 
-// +kubebuilder:rbac:groups=pgv2.percona.com,resources=perconapgrestores,verbs=get;list;watch
+// +kubebuilder:rbac:groups=pgv2.percona.com,resources=perconapgrestores,verbs=get;list;watch;create
 // +kubebuilder:rbac:groups=pgv2.percona.com,resources=perconapgrestores/status,verbs=patch;update
 // +kubebuilder:rbac:groups=pgv2.percona.com,resources=perconapgclusters,verbs=get;list;create;update;patch;watch
 // +kubebuilder:rbac:groups=postgres-operator.crunchydata.com,resources=postgresclusters,verbs=get;list;create;update;patch;watch
@@ -82,11 +83,12 @@ func (r *PGRestoreReconciler) Reconcile(ctx context.Context, request reconcile.R
 			return reconcile.Result{}, errors.Wrap(err, "update PGRestore status")
 		}
 
-		if err := startRestore(ctx, r.Client, pgCluster, pgRestore); err != nil {
-			return reconcile.Result{}, errors.Wrap(err, "start restore")
+		if _, ok := pgRestore.Annotations[pNaming.AnnotationClusterBootstrapRestore]; !ok {
+			if err := startRestore(ctx, r.Client, pgCluster, pgRestore); err != nil {
+				return reconcile.Result{}, errors.Wrap(err, "start restore")
+			}
 		}
 
-		log.Info("Restore is starting")
 		return reconcile.Result{}, nil
 	case v2.RestoreStarting:
 		job := &batchv1.Job{}
@@ -124,8 +126,10 @@ func (r *PGRestoreReconciler) Reconcile(ctx context.Context, request reconcile.R
 			return reconcile.Result{RequeueAfter: time.Second * 5}, nil
 		}
 
-		if err := disableRestore(ctx, r.Client, pgCluster); err != nil {
-			return reconcile.Result{}, errors.Wrap(err, "disable restore")
+		if _, ok := pgRestore.Annotations[pNaming.AnnotationClusterBootstrapRestore]; !ok {
+			if err := disableRestore(ctx, r.Client, pgCluster); err != nil {
+				return reconcile.Result{}, errors.Wrap(err, "disable restore")
+			}
 		}
 
 		// Don't add code after the status update.