change spark context

642514a4 · 高雅喆 · 10d6fed4 · 642514a4
Commit 642514a4 authored Oct 04, 2018 by 高雅喆
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

Main.scala eda/node2vec/src/main/scala/com/gmei/Main.scala +3 -2

No files found.
--- a/eda/node2vec/src/main/scala/com/gmei/Main.scala
+++ b/eda/node2vec/src/main/scala/com/gmei/Main.scala
@@ -87,8 +87,9 @@ object Main {

      //1. get the input and node2vec
      GmeiConfig.setup(param)
-      val context = GmeiConfig.getSparkSession()._1
-      val sc = GmeiConfig.getSparkSession()._2
+      val spark_env = GmeiConfig.getSparkSession()
+      val context = spark_env._1
+      val sc = spark_env._2

      val ti = new TiContext(sc)
      ti.tidbMapTable(dbName = GmeiConfig.config.getString("tidb.database"),tableName = "nd_data_meigou_cid")