K8SPSMDB-1339: Validate PiTR target before starting restore

Author: egegunes

pkg/controller/perconaservermongodbrestore/perconaservermongodbrestore_controller.go

@@ -188,7 +188,7 @@ func (r *ReconcilePerconaServerMongoDBRestore) Reconcile(ctx context.Context, re
 	}
 
 	if cr.Status.State == psmdbv1.RestoreStateNew {
-		err := r.validate(ctx, cr, cluster)
+		err = r.validate(ctx, cr, cluster)
 		if err != nil {
 			if errors.Is(err, errWaitingPBM) {
 				err = nil
@@ -209,7 +209,7 @@ func (r *ReconcilePerconaServerMongoDBRestore) Reconcile(ctx context.Context, re
 				return rr, nil
 			}
 
-			return rr, errors.Wrap(err, "failed to validate restore")
+			return reconcile.Result{}, errors.Wrap(err, "failed to validate restore")
 		}
 
 		if cluster.Spec.Sharding.Enabled {

pkg/controller/perconaservermongodbrestore/validate.go

@@ -49,5 +49,21 @@ func (r *ReconcilePerconaServerMongoDBRestore) validate(ctx context.Context, cr
 		return errors.Wrap(err, "failed to validate backup")
 	}
 
+	if pitr := cr.Spec.PITR; pitr != nil {
+		switch {
+		case pitr.Type == psmdbv1.PITRestoreTypeDate && pitr.Date != nil:
+			ts := pitr.Date.Unix()
+
+			if _, err := pbmc.GetPITRChunkContains(ctx, ts); err != nil {
+				return err
+			}
+		case pitr.Type == psmdbv1.PITRestoreTypeLatest:
+			_, err := pbmc.GetLatestTimelinePITR(ctx)
+			if err != nil {
+				return err
+			}
+		}
+	}
+
 	return nil
 }