FMWK-399-remove-artifacts-flag

- moved dir check
aerospike · Sep 19, 2024 · 3e59af8 · 3e59af8
1 parent 00b040b
commit 3e59af8
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/io/local/reader.go b/io/local/reader.go
@@ -112,6 +112,11 @@ func (r *Reader) StreamFiles(
 	defer close(readersCh)
 	// If it is a folder, open and return.
 	if r.isDir {
+		err := r.checkRestoreDirectory()
+		if err != nil {
+			errorsCh <- err
+			return
+		}
 		r.streamDirectory(ctx, readersCh, errorsCh)
 		return
 	}
@@ -123,12 +128,6 @@ func (r *Reader) StreamFiles(
 func (r *Reader) streamDirectory(
 	ctx context.Context, readersCh chan<- io.ReadCloser, errorsCh chan<- error,
 ) {
-	err := r.checkRestoreDirectory()
-	if err != nil {
-		errorsCh <- err
-		return
-	}
-
 	fileInfo, err := os.ReadDir(r.path)
 	if err != nil {
 		errorsCh <- fmt.Errorf("failed to read path %s: %w", r.path, err)