wip: implement restore logic

Signed-off-by: Mayank Shah <mayank.shah@percona.com>

Author: mayankshah1607

pkg/apis/psmdb/v1/psmdb_types.go

@@ -1744,3 +1744,11 @@ type LogRotateSpec struct {
 func (cr *PerconaServerMongoDB) IsLogCollectorEnabled() bool {
 	return cr.Spec.LogCollector != nil && cr.Spec.LogCollector.Enabled
 }
+
+func (cr *PerconaServerMongoDB) GetAllReplsets() []*ReplsetSpec {
+	replsets := cr.Spec.Replsets
+	if cr.Spec.Sharding.Enabled {
+		replsets = append(replsets, cr.Spec.Sharding.ConfigsvrReplSet)
+	}
+	return replsets
+}

pkg/controller/perconaservermongodbrestore/physical.go

@@ -50,10 +50,7 @@ func (r *ReconcilePerconaServerMongoDBRestore) reconcilePhysicalRestore(
 
 	status := cr.Status
 
-	replsets := cluster.Spec.Replsets
-	if cluster.Spec.Sharding.Enabled {
-		replsets = append(replsets, cluster.Spec.Sharding.ConfigsvrReplSet)
-	}
+	replsets := cluster.GetAllReplsets()
 
 	if cr.Status.State == psmdbv1.RestoreStateNew {
 		pod := corev1.Pod{}
@@ -536,11 +533,7 @@ func (r *ReconcilePerconaServerMongoDBRestore) updateStatefulSetForPhysicalResto
 func (r *ReconcilePerconaServerMongoDBRestore) prepareStatefulSetsForPhysicalRestore(ctx context.Context, cluster *psmdbv1.PerconaServerMongoDB) error {
 	log := logf.FromContext(ctx)
 
-	replsets := cluster.Spec.Replsets
-	if cluster.Spec.Sharding.Enabled {
-		replsets = append(replsets, cluster.Spec.Sharding.ConfigsvrReplSet)
-	}
-
+	replsets := cluster.GetAllReplsets()
 	for _, rs := range replsets {
 		stsName := naming.MongodStatefulSetName(cluster, rs)