[
https://issues.apache.org/jira/browse/SPARK-6444?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Cheng Lian updated SPARK-6444:
------------------------------
Summary: SQL functions (either built-in or UDF) should check for data types
of their arguments (was: Sum expression should remain unresolved if the data
type isn't a numeric type)
> SQL functions (either built-in or UDF) should check for data types of their
> arguments
> -------------------------------------------------------------------------------------
>
> Key: SPARK-6444
> URL: https://issues.apache.org/jira/browse/SPARK-6444
> Project: Spark
> Issue Type: Bug
> Components: SQL
> Affects Versions: 1.0.2, 1.1.1, 1.2.1, 1.3.0
> Reporter: Cheng Lian
> Assignee: Cheng Lian
>
> Spark shell session for reproducing this issue:
> {code}
> import sqlContext._
> sql("""
> CREATE TABLE IF NOT EXISTS ut (
> c1 STRING,
> c2 STRING
> )
> """)
> sql("""
> SELECT SUM(c3) FROM (
> SELECT SUM(c1) AS c3, 0 AS c4 FROM ut -- (1)
> UNION ALL
> SELECT 0 AS c3, COUNT(c2) AS c4 FROM ut -- (2)
> ) t
> """).queryExecution.optimizedPlan
> {code}
> Exception thrown:
> {noformat}
> java.util.NoSuchElementException: key not found: c3#10
> at scala.collection.MapLike$class.default(MapLike.scala:228)
> at
> org.apache.spark.sql.catalyst.expressions.AttributeMap.default(AttributeMap.scala:29)
> at scala.collection.MapLike$class.apply(MapLike.scala:141)
> at
> org.apache.spark.sql.catalyst.expressions.AttributeMap.apply(AttributeMap.scala:29)
> at
> org.apache.spark.sql.catalyst.optimizer.UnionPushdown$$anonfun$1.applyOrElse(Optimizer.scala:80)
> at
> org.apache.spark.sql.catalyst.optimizer.UnionPushdown$$anonfun$1.applyOrElse(Optimizer.scala:79)
> at
> org.apache.spark.sql.catalyst.trees.TreeNode$$anonfun$3.apply(TreeNode.scala:187)
> at
> org.apache.spark.sql.catalyst.trees.TreeNode$$anonfun$3.apply(TreeNode.scala:187)
> at
> org.apache.spark.sql.catalyst.trees.CurrentOrigin$.withOrigin(TreeNode.scala:50)
> at
> org.apache.spark.sql.catalyst.trees.TreeNode.transformDown(TreeNode.scala:186)
> at
> org.apache.spark.sql.catalyst.trees.TreeNode.transform(TreeNode.scala:177)
> at
> org.apache.spark.sql.catalyst.optimizer.UnionPushdown$.pushToRight(Optimizer.scala:79)
> at
> org.apache.spark.sql.catalyst.optimizer.UnionPushdown$$anonfun$apply$1$$anonfun$applyOrElse$6.apply(Optimizer.scala:101)
> at
> org.apache.spark.sql.catalyst.optimizer.UnionPushdown$$anonfun$apply$1$$anonfun$applyOrElse$6.apply(Optimizer.scala:101)
> ...
> {noformat}
> The analyzed plan of the query is:
> {noformat}
> == Analyzed Logical Plan ==
> !Aggregate [], [SUM(CAST(c3#153, DoubleType)) AS _c0#157]
> (c)
> Union
> Project [CAST(c3#153, StringType) AS c3#164,c4#163L]
> (d)
> Project [c3#153,CAST(c4#154, LongType) AS c4#163L]
> Aggregate [], [SUM(CAST(c1#158, DoubleType)) AS c3#153,0 AS c4#154]
> (b)
> MetastoreRelation default, ut, None
> Project [CAST(c3#155, StringType) AS c3#162,c4#156L]
> (a)
> Aggregate [], [0 AS c3#155,COUNT(c2#161) AS c4#156L]
> MetastoreRelation default, ut, None
> {noformat}
> This case is very interesting. It involves 2 analysis rules, {{WidenTypes}}
> and {{PropagateStrings}}, and 1 optimizer rule, {{UnionPushdown}}. To see the
> details, we can turn on TRACE level log and check detailed rule execution
> process. The TL;DR is:
> # Since {{c1}} is STRING, {{SUM(c1)}} is also STRING (which is the root cause
> of the whole issue).
> # {{c3}} in {{(1)}} is STRING, while the one in {{(2)}} is INT. Thus
> {{WidenTypes}} casts the latter to STRING to ensure both sides of the UNION
> have the same schema. See {{(a)}}.
> # {{PropagateStrings}} casts {{c1}} in {{SUM(c1)}} to DOUBLE, which
> consequently changes data type of {{SUM(c1)}} and {{c3}} to DOUBLE. See
> {{(b)}}.
> # {{c3}} in the top level {{Aggregate}} is resolved as DOUBLE (c)
> # Since schemas of the two sides of the UNION are different again,
> {{WidenTypes}} casts {{SUM(c1) AS c3}} to STRING. See {{(d)}}.
> # The top level {{Aggregate}} becomes unresolved since {{c3#153}} is hidden
> by {{(d)}} now.
> # In the optimizing phase, {{UnionPushdown}} throws because the top level
> {{Aggregate}} is unresolved.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
