特征验证

recommend-feature-produce/src/main/java/com/tzld/piaoquan/recommend/feature/produce/FeatureDiff.java

@@ -60,11 +60,7 @@ public class FeatureDiff {
 //        tableToCol.put("alg_vid_feature_feed_province_root_return", "b19_feature");
     }
 
-    public static void main(String[] args) {
-
-
-        Map<String, String> tableToCol = new HashMap<>();
-
+    private static void addUser(Map<String, String> tableToCol) {
         tableToCol.put("alg_mid_feature_play", "c1_feature");
         tableToCol.put("alg_mid_feature_share_and_return", "c2_feature");
         tableToCol.put("alg_mid_feature_play_tags", "c3_feature");
@@ -74,6 +70,9 @@ public class FeatureDiff {
         tableToCol.put("alg_mid_feature_feed_exp_return_tags", "c7_feature");
         tableToCol.put("alg_mid_feature_sharecf", "c8_feature");
         tableToCol.put("alg_mid_feature_returncf", "c9_feature");
+    }
+
+    private static void addVideo(Map<String, String> tableToCol) {
         tableToCol.put("alg_vid_feature_all_exp", "b1_feature");
         tableToCol.put("alg_vid_feature_all_share", "b2_feature");
         tableToCol.put("alg_vid_feature_all_return", "b3_feature");
@@ -93,6 +92,21 @@ public class FeatureDiff {
         tableToCol.put("alg_vid_feature_feed_province_exp", "b17_feature");
         tableToCol.put("alg_vid_feature_feed_province_root_share", "b18_feature");
         tableToCol.put("alg_vid_feature_feed_province_root_return", "b19_feature");
+    }
+
+    public static void main(String[] args) {
+
+
+        Map<String, String> tableToCol = new HashMap<>();
+        if (args[3].equals("0")) {
+            addVideo(tableToCol);
+        } else if (args[3].equals("1")) {
+            addUser(tableToCol);
+        } else {
+            addVideo(tableToCol);
+            addUser(tableToCol);
+        }
+
 
         SparkConf sparkConf = new SparkConf()
                 // .setMaster("local")