FMWK-399-remove-artifacts-flag

- ShouldClearTarget()
aerospike · Sep 19, 2024 · b5c3cb9 · b5c3cb9
1 parent f18f8c4
commit b5c3cb9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/cmd/internal/app/writers.go b/cmd/internal/app/writers.go
@@ -38,7 +38,7 @@ func newLocalWriter(ctx context.Context, b *models.Backup, c *models.Common) (ba
 		opts = append(opts, local.WithFile(b.OutputFile))
 	}
 
-	if b.RemoveFiles || b.RemoveArtifacts {
+	if b.ShouldClearTarget() {
 		opts = append(opts, local.WithRemoveFiles())
 	}
 
@@ -72,7 +72,7 @@ func newS3Writer(
 		opts = append(opts, s3.WithFile(path))
 	}
 
-	if b.RemoveFiles || b.RemoveArtifacts {
+	if b.ShouldClearTarget() {
 		opts = append(opts, s3.WithRemoveFiles())
 	}
 
@@ -102,7 +102,7 @@ func newGcpWriter(
 		opts = append(opts, storage.WithFile(b.OutputFile))
 	}
 
-	if b.RemoveFiles || b.RemoveArtifacts {
+	if b.ShouldClearTarget() {
 		opts = append(opts, storage.WithRemoveFiles())
 	}
 
@@ -132,7 +132,7 @@ func newAzureWriter(
 		opts = append(opts, blob.WithFile(b.OutputFile))
 	}
 
-	if b.RemoveFiles || b.RemoveArtifacts {
+	if b.ShouldClearTarget() {
 		opts = append(opts, blob.WithRemoveFiles())
 	}
 

diff --git a/cmd/internal/models/backup.go b/cmd/internal/models/backup.go
@@ -28,3 +28,8 @@ type Backup struct {
 	ParallelNodes       bool
 	RemoveArtifacts     bool
 }
+
+// ShouldClearTarget check if we should clean target directory.
+func (b *Backup) ShouldClearTarget() bool {
+	return b.RemoveFiles || b.RemoveArtifacts
+}