Implement max_recursionfor hierarchical namespace file info selector

Tom-Newton · Oct 4, 2024 · b9d6ed2 · b9d6ed2
1 parent 0e7e966
commit b9d6ed2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cpp/src/arrow/filesystem/azurefs.cc b/cpp/src/arrow/filesystem/azurefs.cc
@@ -1955,7 +1955,11 @@ class AzureFileSystem::Impl {
           if (path.Name == base_location.path && !path.IsDirectory) {
             return NotADir(base_location);
           }
-          acc_results->push_back(FileInfoFromPath(base_location.container, path));
+          if (internal::GetAbstractPathDepth(path.Name) -
+                  internal::GetAbstractPathDepth(base_location.path) - 1 <=
+              select.max_recursion) {
+            acc_results->push_back(FileInfoFromPath(base_location.container, path));
+          }
         }
       }
     } catch (const Storage::StorageException& exception) {