Merge pull request #98 from Codelone/master-dev

use MySQLReader executor sentence sql
vesoft-inc · Jun 16, 2021 · 14c9d4f · 14c9d4f
2 parents 052daf9 + a3aa765
commit 14c9d4f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nebula-exchange/src/main/scala/com/vesoft/nebula/exchange/reader/ServerBaseReader.scala b/nebula-exchange/src/main/scala/com/vesoft/nebula/exchange/reader/ServerBaseReader.scala
@@ -82,13 +82,15 @@ class MySQLReader(override val session: SparkSession, mysqlConfig: MySQLSourceCo
   override def read(): DataFrame = {
     val url =
       s"jdbc:mysql://${mysqlConfig.host}:${mysqlConfig.port}/${mysqlConfig.database}?useUnicode=true&characterEncoding=utf-8"
-    session.read
+    val df = session.read
       .format("jdbc")
       .option("url", url)
       .option("dbtable", mysqlConfig.table)
       .option("user", mysqlConfig.user)
       .option("password", mysqlConfig.password)
       .load()
+    df.createOrReplaceTempView(mysqlConfig.table)
+    session.sql(sentence)
   }
 }