ソースを参照

fix: 简易注消通用程序修改

许家凯 4 年 前
コミット
541c211e97

+ 7 - 4
src/main/scala/com/winhc/bigdata/spark/utils/CompanyIncrForCidWithoutMD5Utils.scala

@@ -44,12 +44,15 @@ case class CompanyIncrForCidWithoutMD5Utils(s: SparkSession,
       val key = s"$rowkey AS rowkey"
 
       var phoenixCols: Seq[String] = null
-      if (adsTable.equalsIgnoreCase("winhc_eci_dev.inc_ads_company_mortgage_info")) {
-        phoenixCols = phoenixCols2
-      } else {
-        phoenixCols = phoenixCols2.filter(!_.equalsIgnoreCase("id"))
+      adsTable match {
+        case "winhc_eci_dev.inc_ads_company_mortgage_info" | "winhc_eci_dev.inc_ads_company_brief_cancel_announcement" =>
+          println("add id ...")
+          phoenixCols = phoenixCols2
+        case _ =>
+          phoenixCols = phoenixCols2.filter(!_.equalsIgnoreCase("id"))
       }
 
+
       val res = phoenixCols.map(s => {
         if ("NEW_CID".equals(s.toUpperCase())) {
           s"cast ($s as string) as CID"