small change

gaoyangxiaozhu · Apr 9, 2024 · 1ab09f9 · 1ab09f9
1 parent cef0020
commit 1ab09f9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/velox/connectors/hive/SplitReader.cpp b/velox/connectors/hive/SplitReader.cpp
@@ -216,7 +216,9 @@ void SplitReader::setRowIndexColumnInfoIfNeed(
   auto rowIndexMetaColIdx =
       readerOutputType_->getChildIdxIfExists(kSparkReservedTmpMetaRowIndex);
   if (rowIndexMetaColIdx.has_value() &&
-      !fileType->containsChild(kSparkReservedTmpMetaRowIndex)) {
+      !fileType->containsChild(kSparkReservedTmpMetaRowIndex) &&
+      hiveSplit_->partitionKeys.find(kSparkReservedTmpMetaRowIndex) ==
+          hiveSplit_->partitionKeys.end()) {
     dwio::common::RowNumberColumnInfo rowNumberColumnInfo;
     rowNumberColumnInfo.insertPosition = rowIndexMetaColIdx.value();
     rowNumberColumnInfo.name = kSparkReservedTmpMetaRowIndex;