Merge pull request #44 from tobegit3hub/fix/fix_register_table

4paradigm · May 12, 2023 · 5b2331a · 5b2331a
2 parents 677302e + 2d77827
commit 5b2331a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/core/pom.xml b/core/pom.xml
@@ -34,8 +34,8 @@
     <sbt.project.name>core</sbt.project.name>
 
     <!-- 4paradigm -->
-    <openmldb.batch.version>0.7.3</openmldb.batch.version>
-    <openmldb.native.version>0.7.3-allinone</openmldb.native.version>
+    <openmldb.batch.version>0.8.0</openmldb.batch.version>
+    <openmldb.native.version>0.8.0-allinone</openmldb.native.version>
   </properties>
 
   <dependencies>

diff --git a/sql/core/src/main/scala/org/apache/spark/sql/Dataset.scala b/sql/core/src/main/scala/org/apache/spark/sql/Dataset.scala
@@ -3273,9 +3273,6 @@ class Dataset[T] private[sql](
   @deprecated("Use createOrReplaceTempView(viewName) instead.", "2.0.0")
   def registerTempTable(tableName: String): Unit = {
     createOrReplaceTempView(tableName)
-
-    // Add by 4paradigm
-    sparkSession.openmldbSession.registerTable(tableName, this.asInstanceOf[DataFrame])
   }
 
   /**
@@ -3306,6 +3303,9 @@ class Dataset[T] private[sql](
    * @since 2.0.0
    */
   def createOrReplaceTempView(viewName: String): Unit = withPlan {
+    // Add by 4paradigm
+    sparkSession.openmldbSession.registerTableInOpenmldbSession("default_db", viewName, this.asInstanceOf[DataFrame])
+
     createTempViewCommand(viewName, replace = true, global = false)
   }