Introduce cleanupCluster state

It will be used for cleaning databases/collections in the sharded
cluster before the logical restore.

Author: boris-ilijic

pbm/defs/defs.go

@@ -97,15 +97,16 @@ const (
 	// for phys restore, to indicate shards have been stopped
 	StatusDown Status = "down"
 
-	StatusStarting   Status = "starting"
-	StatusRunning    Status = "running"
-	StatusDumpDone   Status = "dumpDone"
-	StatusCopyReady  Status = "copyReady"
-	StatusCopyDone   Status = "copyDone"
-	StatusPartlyDone Status = "partlyDone"
-	StatusDone       Status = "done"
-	StatusCancelled  Status = "canceled"
-	StatusError      Status = "error"
+	StatusStarting       Status = "starting"
+	StatusCleanupCluster Status = "cleanupCluster"
+	StatusRunning        Status = "running"
+	StatusDumpDone       Status = "dumpDone"
+	StatusCopyReady      Status = "copyReady"
+	StatusCopyDone       Status = "copyDone"
+	StatusPartlyDone     Status = "partlyDone"
+	StatusDone           Status = "done"
+	StatusCancelled      Status = "canceled"
+	StatusError          Status = "error"
 
 	// status to communicate last op timestamp if it's not set
 	// during external restore

pbm/restore/logical.go

@@ -240,7 +240,9 @@ func (r *Restore) Snapshot(
 		return err
 	}
 
-	err = r.toState(ctx, defs.StatusRunning, &defs.WaitActionStart)
+	err = r.toState(ctx, defs.StatusCleanupCluster, &defs.WaitActionStart)
+
+	err = r.toState(ctx, defs.StatusRunning, nil)
 	if err != nil {
 		return err
 	}