odpstable-fix-#820 (#828)

* odpstable-fix-#820 * odpstable-fix-#820 : adjust the cases order for odps table
AbsaOSS · Sep 5, 2024 · 1a910c2 · 1a910c2
1 parent fd58cd0
commit 1a910c2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/core/src/main/scala/za/co/absa/spline/harvester/plugin/embedded/DataSourceV2Plugin.scala b/core/src/main/scala/za/co/absa/spline/harvester/plugin/embedded/DataSourceV2Plugin.scala
@@ -163,6 +163,8 @@ class DataSourceV2Plugin
       val pathUris = paths.map(prependFileSchemaIfMissing)
       SourceIdentifier(Some(format), pathUris: _*)
 
+    case `_: OdpsTable`(ot) => SourceIdentifier(Some("odpstable"),s"odpstable:${extractValue[String](ot,"name")}")
+
     case `_: TableV2`(tv2) => extractSourceIdFromDeltaTableV2(tv2)
   }
 
@@ -231,4 +233,7 @@ object DataSourceV2Plugin {
   private object `_: TableV2` extends SafeTypeMatchingExtractor[AnyRef](
     "org.apache.spark.sql.connector.catalog.Table")
 
+  private object `_: OdpsTable` extends SafeTypeMatchingExtractor[AnyRef](
+    "org.apache.spark.sql.execution.datasources.v2.odps.OdpsTable")
+
 }