src/main/java/com/winhc/bigdata/udf/justice/PartySummaryUDF.java

@@ -11,6 +11,7 @@ import com.winhc.bigdata.utils.CaseNoPropertiesUtils;
 import org.apache.commons.lang3.StringUtils;
 
 import java.util.*;
+import java.util.stream.Collectors;
 
 /**
  * @Author: XuJiakai
@@ -56,6 +57,7 @@ public class PartySummaryUDF extends UDF {
             put("异议人", "其他");
             put("申请机关", "其他");
             put("公诉人", "其他");
+            put("第三人", "第三人");
 
         }
     };
@@ -156,23 +158,26 @@ public class PartySummaryUDF extends UDF {
     }
 
     public static void main(String[] args) {
-        String res = new PartySummaryUDF().evaluate("[{\"party_info\":[{\"litigant_id\":\"\",\"name\":\"平安保险东莞分公司\"}],\"show_level\":0,\"title_original\":\"原告\",\"title_std\":\"原告\"},{\"party_info\":[{\"litigant_id\":\"\",\"name\":\"沧州顺驰汽车公司\"},{\"litigant_id\":\"\",\"name\":\"崔建伟\"}],\"show_level\":1,\"title_original\":\"被告\",\"title_std\":\"被告\"}]");
-        System.out.println(res);
-       /* Entity<List<String>, Map<String, Set<CaseNoTitleEntity>>> caseNoKeyword =
+//        String res = new PartySummaryUDF().evaluate("[{\"party_info\":[{\"litigant_id\":\"\",\"name\":\"平安保险东莞分公司\"}],\"show_level\":0,\"title_original\":\"原告\",\"title_std\":\"原告\"},{\"party_info\":[{\"litigant_id\":\"\",\"name\":\"沧州顺驰汽车公司\"},{\"litigant_id\":\"\",\"name\":\"崔建伟\"}],\"show_level\":1,\"title_original\":\"被告\",\"title_std\":\"被告\"}]");
+//        System.out.println(res);
+        Entity<List<String>, Map<String, Set<CaseNoTitleEntity>>> caseNoKeyword =
                 CaseNoPropertiesUtils.getCaseNoKeyword();
         Set<String> set = new HashSet<>();
 
         Collection<Set<CaseNoTitleEntity>> values = caseNoKeyword._2.values();
         for (Set<CaseNoTitleEntity> value : values) {
             for (CaseNoTitleEntity caseNoTitleEntity : value) {
+                Set<String> collect = caseNoTitleEntity.getAll().stream().flatMap(Collection::stream).collect(Collectors.toSet());
+                set.addAll(collect);
                 set.addAll(caseNoTitleEntity.getCore());
             }
         }
         Set<String> ss = new HashSet<>();
         for (String s : set) {
-            ss.add(getTitleClassesByMap(s));
-            System.out.println(s + " " + getTitleClassesByMap(s));
+            System.out.println("tree.build_tree('"+s+"')");
+//            ss.add(getTitleClassesByMap(s));
+//            System.out.println(s + " " + getTitleClassesByMap(s));
         }
-        System.out.println(ss);*/
+//        System.out.println(ss);
     }
 }