revert code

apache · Dec 21, 2023 · 9e82fe2 · 9e82fe2
1 parent d5a1db3
commit 9e82fe2
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/...i-extension-spark-3-4/src/main/scala/org/apache/kyuubi/sql/watchdog/MaxScanStrategy.scala b/...i-extension-spark-3-4/src/main/scala/org/apache/kyuubi/sql/watchdog/MaxScanStrategy.scala
@@ -123,15 +123,10 @@ case class MaxScanStrategy(session: SparkSession)
         } else {
           lazy val scanFileSize = relation.tableMeta.stats.map(_.sizeInBytes).sum
           if (maxFileSizeOpt.exists(_ < scanFileSize)) {
-            throw new MaxFileSizeExceedException(
-              s"""
-                 |Your SQL job scan a whole huge table without any partition filter,
-                 |You should optimize your SQL logical according partition structure
-                 |or shorten query scope such as p_date, detail as below:
-                 |Table: ${relation.tableMeta.qualifiedName}
-                 |Owner: ${relation.tableMeta.owner}
-                 |Partition Structure: ${relation.partitionCols.map(_.name).mkString(", ")}
-                 |""".stripMargin)
+            throw nonPartTableMaxFileExceedError(
+              scanFileSize,
+              maxFileSizeOpt.get,
+              Some(relation.tableMeta))
           }
         }
       case ScanOperation(