unset kyuubi.operation.plan.only.mode if mode is invalid

apache · Aug 10, 2022 · a1df52f · a1df52f
1 parent a75de1b
commit a1df52f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...ne/src/main/scala/org/apache/kyuubi/engine/spark/operation/SparkSQLOperationManager.scala b/...ne/src/main/scala/org/apache/kyuubi/engine/spark/operation/SparkSQLOperationManager.scala
@@ -69,6 +69,11 @@ class SparkSQLOperationManager private (name: String) extends OperationManager(n
       OperationLanguages.withName(lang.toUpperCase(Locale.ROOT)) match {
         case OperationLanguages.SQL =>
           val mode = spark.conf.get(OPERATION_PLAN_ONLY_MODE.key, operationModeDefault)
+          if (!OperationModes.values.exists(_.toString.equalsIgnoreCase(mode))) {
+            spark.conf.unset(OPERATION_PLAN_ONLY_MODE.key)
+            throw new NoSuchElementException(s"Unsupported operation mode: $mode," +
+              s" unset kyuubi.operation.plan.only.mode")
+          }
           OperationModes.withName(mode.toUpperCase(Locale.ROOT)) match {
             case NONE =>
               val incrementalCollect = spark.conf.getOption(OPERATION_INCREMENTAL_COLLECT.key)