add asserts

apache · Apr 14, 2021 · 111ef8b · 111ef8b
1 parent a0a7297
commit 111ef8b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/plans/logical/v2Commands.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/plans/logical/v2Commands.scala
@@ -1025,8 +1025,10 @@ case class CacheTableAsSelect(
     options: Map[String, String],
     isAnalyzed: Boolean = false) extends AnalysisOnlyCommand {
   override protected def withNewChildrenInternal(
-      newChildren: IndexedSeq[LogicalPlan]): CacheTableAsSelect =
+      newChildren: IndexedSeq[LogicalPlan]): CacheTableAsSelect = {
+    assert(!isAnalyzed)
     copy(plan = newChildren.head)
+  }
 
   override def childrenToAnalyze: Seq[LogicalPlan] = plan :: Nil
 

diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/command/views.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/command/views.scala
@@ -72,8 +72,10 @@ case class CreateViewCommand(
   import ViewHelper._
 
   override protected def withNewChildrenInternal(
-      newChildren: IndexedSeq[LogicalPlan]): CreateViewCommand =
+      newChildren: IndexedSeq[LogicalPlan]): CreateViewCommand = {
+    assert(!isAnalyzed)
     copy(plan = newChildren.head)
+  }
 
   override def innerChildren: Seq[QueryPlan[_]] = Seq(plan)
 
@@ -249,8 +251,10 @@ case class AlterViewAsCommand(
   import ViewHelper._
 
   override protected def withNewChildrenInternal(
-      newChildren: IndexedSeq[LogicalPlan]): AlterViewAsCommand =
+      newChildren: IndexedSeq[LogicalPlan]): AlterViewAsCommand = {
+    assert(!isAnalyzed)
     copy(query = newChildren.head)
+  }
 
   override def innerChildren: Seq[QueryPlan[_]] = Seq(query)