瀏覽代碼

修改分公司逻辑

xufei 2 年之前
父節點
當前提交
81dc45eb19
共有 1 個文件被更改,包括 4 次插入4 次删除
  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"));
     }
 }