Parcourir la source

修改分公司逻辑

xufei il y a 2 ans
Parent
commit
81dc45eb19
1 fichiers modifiés avec 4 ajouts et 4 suppressions
  1. 4 4
      src/main/java/com/winhc/bigdata/udf/CompanyOrgTypeNew.java

+ 4 - 4
src/main/java/com/winhc/bigdata/udf/CompanyOrgTypeNew.java

@@ -16,11 +16,11 @@ public class CompanyOrgTypeNew extends UDF {
     private static final List<String> FOREIGN = Arrays.asList("中外", "外资", "外国", "外商");
     private static final List<String> CHINA = Arrays.asList("国有", "国资");
 
-    public String evaluate(String company_org_type, String company_name, String credit_code) {
+    public List<String> evaluate(String company_org_type, String company_name, String credit_code) {
         if (isEmpty(company_name)) return null;
 //分公司判断
         if (contains(company_org_type, "分公司") || endsWith(company_name, "分公司")) {
-            return JSON.toJSONString(Collections.singletonList("分公司"));
+            return null;
         }
 
 //工商注册类型判断
@@ -61,10 +61,10 @@ public class CompanyOrgTypeNew extends UDF {
         if (result.isEmpty()) {
             return null;
         }
-        return JSON.toJSONString(result);
+        return new ArrayList<>(result);
     }
 
     public static void main(String[] args) {
-        System.out.println(new CompanyOrgTypeNew().evaluate("分公司", "雅诗兰黛(上海)商贸有限公司北京分公司", "91110105MA01Q19Q2L"));
+        System.out.println(new CompanyOrgTypeNew().evaluate("有限公司", "雅诗兰黛(上海)商贸有限公司", "91110105MA01Q19Q2L"));
     }
 }