diff --git a/core/src/main/scala/org/apache/spark/ErrorInfo.scala b/core/src/main/scala/org/apache/spark/ErrorInfo.scala index 41c0e83f917b4..99e081ebb45ea 100644 --- a/core/src/main/scala/org/apache/spark/ErrorInfo.scala +++ b/core/src/main/scala/org/apache/spark/ErrorInfo.scala @@ -61,8 +61,13 @@ private[spark] object SparkThrowableHelper { queryContext: String = ""): String = { val errorInfo = errorClassToInfoMap.getOrElse(errorClass, throw new IllegalArgumentException(s"Cannot find error class '$errorClass'")) + val displayQueryContext = if (queryContext.isEmpty) { + "" + } else { + s"\n$queryContext" + } String.format(errorInfo.messageFormat.replaceAll("<[a-zA-Z0-9_-]+>", "%s"), - messageParameters: _*) + queryContext + messageParameters: _*) + displayQueryContext } def getSqlState(errorClass: String): String = { diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/trees/TreeNode.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/trees/TreeNode.scala index 079abd3f2e03c..09d24c5f81181 100644 --- a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/trees/TreeNode.scala +++ b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/trees/TreeNode.scala @@ -89,7 +89,7 @@ case class Origin( "" } val builder = new StringBuilder - builder ++= s"\n== SQL$objectContext$positionContext ==\n" + builder ++= s"== SQL$objectContext$positionContext ==\n" val text = sqlText.get val start = math.max(startIndex.get, 0) diff --git a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/trees/TreeNodeSuite.scala b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/trees/TreeNodeSuite.scala index ffbc5d89bdb61..899a740bdae86 100644 --- a/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/trees/TreeNodeSuite.scala +++ b/sql/catalyst/src/test/scala/org/apache/spark/sql/catalyst/trees/TreeNodeSuite.scala @@ -876,8 +876,7 @@ class TreeNodeSuite extends SparkFunSuite with SQLHelper { objectType = Some("VIEW"), objectName = Some("some_view")) val expected = - """ - |== SQL of VIEW some_view(line 3, position 38) == + """== SQL of VIEW some_view(line 3, position 38) == |...7890 + 1234567890 + 1234567890, cast('a' | ^^^^^^^^ |as /* comment */