add DCHECK for path containing selector

kszucs · Nov 4, 2019 · f2f9689 · f2f9689
1 parent 1bae5ff
commit f2f9689
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cpp/src/arrow/dataset/partition.cc b/cpp/src/arrow/dataset/partition.cc
@@ -91,6 +91,9 @@ Result<std::shared_ptr<Expression>> HivePartitionScheme::Parse(
 Status ApplyPartitionScheme(const PartitionScheme& scheme, const fs::Selector& selector,
                             std::vector<fs::FileStats> files, PathPartitions* out) {
   for (const auto& file : files) {
+    // XXX is this the right way to drop the base dir?
+    DCHECK(std::equal(selector.base_dir.begin(), selector.base_dir.end(),
+                      file.path().begin()));
     const auto& path = file.path().substr(selector.base_dir.size());
     std::shared_ptr<Expression> partition;
     RETURN_NOT_OK(scheme.Parse(path, &partition));