viirya commented on a change in pull request #28988:
URL: https://github.com/apache/spark/pull/28988#discussion_r449903754
##########
File path:
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/optimizer/NestedColumnAliasing.scala
##########
@@ -174,9 +176,12 @@ object NestedColumnAliasing {
// If all nested fields of `attr` are used, we don't need to introduce
new aliases.
// By default, ColumnPruning rule uses `attr` already.
+ // Note that we need to remove cosmetic variations first, so we only
count a
+ // nested field once.
if (nestedFieldToAlias.nonEmpty &&
- nestedFieldToAlias
- .map { case (nestedField, _) =>
totalFieldNum(nestedField.dataType) }
+ dedupNestedFields.map(_.canonicalized.asInstanceOf[ExtractValue])
Review comment:
For the duplicated aliases, we can avoid it. We can work on
canonicalized extractors when generating aliases.
But we also need to convert coming extractors to canonicalized versions when
we look up into the alias map.
Currently the code looks clear. And it seems not a big deal, and I think it
is rare case that there are multiple extractors with cosmetic difference. So
currently I don't try to do that.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
