[HUDI-4161] Make sure partition values are taken from partition path (#5699)
This commit is contained in:
@@ -54,6 +54,16 @@ class BaseFileOnlyRelation(sqlContext: SQLContext,
|
||||
|
||||
override type FileSplit = HoodieBaseFileSplit
|
||||
|
||||
// TODO(HUDI-3204) this is to override behavior (exclusively) for COW tables to always extract
|
||||
// partition values from partition path
|
||||
// For more details please check HUDI-4161
|
||||
// NOTE: This override has to mirror semantic of whenever this Relation is converted into [[HadoopFsRelation]],
|
||||
// which is currently done for all cases, except when Schema Evolution is enabled
|
||||
override protected val shouldExtractPartitionValuesFromPartitionPath: Boolean = {
|
||||
val enableSchemaOnRead = !internalSchema.isEmptySchema
|
||||
!enableSchemaOnRead
|
||||
}
|
||||
|
||||
override lazy val mandatoryFields: Seq[String] =
|
||||
// TODO reconcile, record's key shouldn't be mandatory for base-file only relation
|
||||
Seq(recordKeyField)
|
||||
|
||||
@@ -171,7 +171,7 @@ abstract class HoodieBaseRelation(val sqlContext: SQLContext,
|
||||
protected val shouldExtractPartitionValuesFromPartitionPath: Boolean = {
|
||||
// Controls whether partition columns (which are the source for the partition path values) should
|
||||
// be omitted from persistence in the data files. On the read path it affects whether partition values (values
|
||||
// of partition columns) will be read from the data file ot extracted from partition path
|
||||
// of partition columns) will be read from the data file or extracted from partition path
|
||||
val shouldOmitPartitionColumns = metaClient.getTableConfig.shouldDropPartitionColumns && partitionColumns.nonEmpty
|
||||
val shouldExtractPartitionValueFromPath =
|
||||
optParams.getOrElse(DataSourceReadOptions.EXTRACT_PARTITION_VALUES_FROM_PARTITION_PATH.key,
|
||||
@@ -419,7 +419,7 @@ abstract class HoodieBaseRelation(val sqlContext: SQLContext,
|
||||
}
|
||||
} catch {
|
||||
case NonFatal(e) =>
|
||||
logWarning(s"Failed to get the right partition InternalRow for file : ${file.toString}")
|
||||
logWarning(s"Failed to get the right partition InternalRow for file: ${file.toString}", e)
|
||||
InternalRow.empty
|
||||
}
|
||||
}
|
||||
|
||||
@@ -108,9 +108,6 @@ case class HoodieFileIndex(spark: SparkSession,
|
||||
* @return list of PartitionDirectory containing partition to base files mapping
|
||||
*/
|
||||
override def listFiles(partitionFilters: Seq[Expression], dataFilters: Seq[Expression]): Seq[PartitionDirectory] = {
|
||||
val convertedPartitionFilters =
|
||||
HoodieFileIndex.convertFilterForTimestampKeyGenerator(metaClient, partitionFilters)
|
||||
|
||||
// Look up candidate files names in the col-stats index, if all of the following conditions are true
|
||||
// - Data-skipping is enabled
|
||||
// - Col-Stats Index is present
|
||||
@@ -144,7 +141,7 @@ case class HoodieFileIndex(spark: SparkSession,
|
||||
Seq(PartitionDirectory(InternalRow.empty, candidateFiles))
|
||||
} else {
|
||||
// Prune the partition path by the partition filters
|
||||
val prunedPartitions = prunePartition(cachedAllInputFileSlices.keySet.asScala.toSeq, convertedPartitionFilters)
|
||||
val prunedPartitions = prunePartition(cachedAllInputFileSlices.keySet.asScala.toSeq, partitionFilters)
|
||||
var totalFileSize = 0
|
||||
var candidateFileSize = 0
|
||||
|
||||
|
||||
Reference in New Issue
Block a user