Add new Execution suite

miland-db · Nov 25, 2024 · 33c6f52 · 33c6f52
1 parent aa2047f
commit 33c6f52
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 137 deletions.
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/scripting/SqlScriptingExecutionNode.scala b/sql/core/src/main/scala/org/apache/spark/sql/scripting/SqlScriptingExecutionNode.scala
@@ -39,7 +39,7 @@ sealed trait CompoundStatementExec extends Logging {
   /**
    * Whether the statement originates from the SQL statement that returns the result.
    */
-  val isResult: Boolean = false
+  def isResult: Boolean = false
 
   /**
    * Reset execution of the current node.
@@ -127,7 +127,7 @@ class SingleStatementExec(
    */
   var isExecuted = false
 
-  override val isResult: Boolean = parsedPlan.isInstanceOf[Project]
+  override def isResult: Boolean = parsedPlan.isInstanceOf[Project] && !isExecuted
 
   /**
    * Get the SQL query text corresponding to this statement.