odpstable-fix-AbsaOSS#820

BerlinJava · Sep 2, 2024 · df527ce · df527ce
1 parent a6f63b1
commit df527ce
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
@@ -164,6 +164,8 @@ class DataSourceV2Plugin
       SourceIdentifier(Some(format), pathUris: _*)
 
     case `_: TableV2`(tv2) => extractSourceIdFromDeltaTableV2(tv2)
+
+    case `_: OdpsTable`(ot) => SourceIdentifier(Some("odpstable"),s"odpstable:${extractValue[String](ot,"name")}")
   }
 
   private def extractSourceIdFromDeltaTableV2(table: AnyRef): SourceIdentifier = {
@@ -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")
+
 }