lanyuanxiaoyao/hudi
1
0
Fork 0
Code Activity

Support referencing subquery with column aliases by table alias in merge into (#3380)

Browse Source
This commit is contained in:
董可伦
2021-08-21 21:53:16 +08:00
committed by GitHub
parent ab3fbb8895
commit be8c1e499f
2 changed files with 59 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
hudi-spark-datasource/hudi-spark2/src/main/scala/org/apache/spark/sql/hudi/parser/HoodieSpark2ExtendedSqlAstBuilder.scala
View File

@@ -185,12 +185,12 @@ class HoodieSpark2ExtendedSqlAstBuilder(conf: SQLConf, delegate: ParserInterface
*/
protected def mayApplyAliasPlan(tableAlias: TableAliasContext, plan: LogicalPlan): LogicalPlan = {
if (tableAlias.strictIdentifier != null) {
val subquery = SubqueryAlias(tableAlias.strictIdentifier.getText, plan)
val alias = tableAlias.strictIdentifier.getText
if (tableAlias.identifierList != null) {
val columnNames = visitIdentifierList(tableAlias.identifierList)
UnresolvedSubqueryColumnAliases(columnNames, subquery)
SubqueryAlias(alias, UnresolvedSubqueryColumnAliases(columnNames, plan))
} else {
subquery
SubqueryAlias(alias, plan)
}
} else {
plan