xufei 2 years ago
parent
commit
dbfb9628e1

+ 17 - 0
src/main/java/com/winhc/task/common/SummaryArgs.java

@@ -29,6 +29,23 @@ public class SummaryArgs {
 
     private static Map<String, SummaryArgs> createArgs() {
         HashMap<String, SummaryArgs> a = new HashMap<>();
+        a.put("company_branch", new SummaryArgs(
+                "ng_rt_company_branch"
+                , Collections.singletonList(new ArgsInfo(
+                "company_id"
+                , ""
+                , "deleted"))
+                , Arrays.asList(new ValueAlias(
+                        ""
+                        , "0"
+                        , "company_branch_del_0"),
+                new ValueAlias(
+                        ""
+                        , "1"
+                        , "company_branch_del_1")
+        )
+                , (j) -> Collections.singletonList(j.getString("parent_company_id"))
+        ));
         a.put("company_holder", new SummaryArgs(
                 "ng_rt_company_holder"
                 , Arrays.asList(new ArgsInfo(

+ 2 - 2
src/test/java/com/winhc/task/DataWorksSummaryJob.java

@@ -35,7 +35,7 @@ public class DataWorksSummaryJob {
 
         //确定重跑集合(集合为空跑全量)
 
-        List<String> add = Arrays.asList("company_patent");
+        List<String> add = Arrays.asList("company_branch");
         //List<String> add = Arrays.asList("company_copyright_reg","company_copyright_works");
         //List<String> add = Arrays.asList("company_patent","company_copyright_reg","company_copyright_works");
 
@@ -46,7 +46,7 @@ public class DataWorksSummaryJob {
                     return b1 && b2;
                 }).distinct().collect(Collectors.toList());
 
-        assertEquals(add.size(),list.size());
+        assertEquals(add.size(), list.size());
         String project = "winhc_ng";
         String flow = "syn_summary";
         String taskName = "summary_v9";