Blaze-627:Make ORC and Parquet format detection more generic (#628)

kwai · Oct 22, 2024 · 032dc7e · 032dc7e
1 parent 009d904
commit 032dc7e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spark-extension/src/main/scala/org/apache/spark/sql/blaze/BlazeConverters.scala b/spark-extension/src/main/scala/org/apache/spark/sql/blaze/BlazeConverters.scala
@@ -298,9 +298,9 @@ object BlazeConverters extends Logging {
     logDebug(s"  dataFilters: ${dataFilters}")
     logDebug(s"  tableIdentifier: ${tableIdentifier}")
     relation.fileFormat match {
-      case p if p.isInstanceOf[ParquetFileFormat] =>
+      case p if p.getClass().getName().endsWith("ParquetFileFormat") =>
         addRenameColumnsExec(Shims.get.createNativeParquetScanExec(exec))
-      case p if p.isInstanceOf[OrcFileFormat] =>
+      case p if p.getClass().getName().endsWith("OrcFileFormat") =>
         addRenameColumnsExec(Shims.get.createNativeOrcScanExec(exec))
       case _ => throw new NotImplementedError("Cannot convert non parquet/orc scan exec")
     }