会员标签4.0

src/main/java/com/gic/spark/SparkTagProcessByEnterpriseOrId.java → src/main/java/com/gic/spark/SparkTagProcessByEnterpriseId.java

@@ -13,22 +13,20 @@ import java.util.List;
  * @author: wangxk
  * @date: 2020/4/15
  */
-public class SparkTagProcessByEnterpriseOrId {
+public class SparkTagProcessByEnterpriseId {
 
     public static void main(String[] args){
         if (args == null || args.length < 4) {
             System.err.println("invalidate input params:");
             System.err.println("1.is production: true/false");
             System.err.println("2.extract data: true/false");
-            System.err.println("3.type: enterprise/tagGroup");
-            System.err.println("4.id: enterpriseId/tagGroupId");
+            System.err.println("3.id: enterpriseId");
             System.exit(2);
         }
 
         boolean isProd = Boolean.parseBoolean(args[0]);
         boolean extractData = Boolean.parseBoolean(args[1]);
-        String type=args[2];
-        String[] arrs=args[3].split(",");
+        String[] arrs=args[2].split(",");
         List<Integer>params=new ArrayList();
         for(String s:arrs){
             params.add(Integer.parseInt(s));
@@ -37,13 +35,7 @@ public class SparkTagProcessByEnterpriseOrId {
         SparkEnvManager.getInstance().init("member_tag_4.0");
         SparkConfigManager.getInstance().init(SparkConfigManager.EnvType.PRO, null);
         TagProcessManager.getInstance().init(isProd);
-        if(type.equals("enterprise")){
-            TagProcessManager.getInstance().setEnterpriseId(params);
-        }else if(type.equals("tagGroup")){
-            TagProcessManager.getInstance().setTagGroupId(params);
-        }else{
-            return;
-        }
+        TagProcessManager.getInstance().setEnterpriseId(params);
         TagProcessManager.getInstance().process(extractData);
     }
 }

src/main/java/com/gic/spark/SparkTagProcessByTagGroupId.java

+package com.gic.spark;
+
+import com.gic.sharding.sdk.ShardingConfigManager;
+import com.gic.spark.application.SparkEnvManager;
+import com.gic.spark.config.SparkConfigManager;
+import com.gic.spark.tag.TagProcessManager;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * @description:
+ * @author: wangxk
+ * @date: 2020/4/15
+ */
+public class SparkTagProcessByTagGroupId {
+
+    public static void main(String[] args){
+        if (args == null || args.length < 4) {
+            System.err.println("invalidate input params:");
+            System.err.println("1.is production: true/false");
+            System.err.println("2.extract data: true/false");
+            System.err.println("3.id: tagGroupId");
+            System.exit(2);
+        }
+
+        boolean isProd = Boolean.parseBoolean(args[0]);
+        boolean extractData = Boolean.parseBoolean(args[1]);
+        String[] arrs=args[2].split(",");
+        List<Integer>params=new ArrayList();
+        for(String s:arrs){
+            params.add(Integer.parseInt(s));
+        }
+        ShardingConfigManager.initDefualt();
+        SparkEnvManager.getInstance().init("member_tag_4.0");
+        SparkConfigManager.getInstance().init(SparkConfigManager.EnvType.PRO, null);
+        TagProcessManager.getInstance().init(isProd);
+        TagProcessManager.getInstance().setTagGroupId(params);
+        TagProcessManager.getInstance().process(extractData);
+    }
+}