[CARMEL-1463][FOLLOWUP] Fix a bug of delete command type casting (del…

…ta-io#3)
LantaoJin · Nov 6, 2019 · 56f4c3b · 56f4c3b
1 parent 2f49adf
commit 56f4c3b
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/main/scala/io/delta/tables/execution/DeltaTableOperations.scala b/src/main/scala/io/delta/tables/execution/DeltaTableOperations.scala
@@ -43,8 +43,13 @@ trait DeltaTableOperations extends AnalysisHelper { self: DeltaTable =>
     subqueryNotSupportedCheck(condition, "DELETE")
 
     val qe = sparkSession.sessionState.executePlan(delete)
-    val resolvedDelete = qe.analyzed.asInstanceOf[Delete]
-    val deleteCommand = DeleteCommand(resolvedDelete)
+    val deleteCommand = qe.analyzed match {
+      case command: DeleteCommand =>
+        command
+      case _ =>
+        val resolvedDelete = qe.analyzed.asInstanceOf[Delete]
+        DeleteCommand(resolvedDelete)
+    }
     deleteCommand.run(sparkSession)
   }