Add --allow-partly-done option for cli

Author: boris-ilijic

cmd/pbm/main.go

@@ -812,6 +812,11 @@ func (app *pbmApp) buildRestoreCmd() *cobra.Command {
 		&restoreOptions.fallback, "fallback-enabled", true,
 		"Disables fallback sync feature when doing physical restore. Enabled by default.",
 	)
+	restoreCmd.Flags().BoolVar(
+		&restoreOptions.allowPartlyDone, "allow-partly-done", true,
+		"Allows parly done state of the cluster after physical restore. "+
+			"If disabled, fallback will be applied when cluster is partly-done. Enabled by default.",
+	)
 
 	restoreCmd.Flags().StringVar(&restoreOptions.rsMap, RSMappingFlag, "", RSMappingDoc)
 	_ = viper.BindPFlag(RSMappingFlag, restoreCmd.Flags().Lookup(RSMappingFlag))

cmd/pbm/restore.go

@@ -39,20 +39,21 @@ var (
 )
 
 type restoreOpts struct {
-	bcp           string
-	pitr          string
-	pitrBase      string
-	wait          bool
-	waitTime      time.Duration
-	extern        bool
-	ns            string
-	nsFrom        string
-	nsTo          string
-	usersAndRoles bool
-	rsMap         string
-	conf          string
-	ts            string
-	fallback      bool
+	bcp             string
+	pitr            string
+	pitrBase        string
+	wait            bool
+	waitTime        time.Duration
+	extern          bool
+	ns              string
+	nsFrom          string
+	nsTo            string
+	usersAndRoles   bool
+	rsMap           string
+	conf            string
+	ts              string
+	fallback        bool
+	allowPartlyDone bool
 
 	numParallelColls    int32
 	numInsertionWorkers int32