From 1d0ec84e35d89ef84190b4f8f224957bd6f1bc0c Mon Sep 17 00:00:00 2001
From: zhengyf <zyf654042662@163.com>
Date: Thu, 3 Jul 2025 11:40:00 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=BD=E7=9F=A5=EF=BC=9A1=E3=80=81=E5=AE=A2?=
 =?UTF-8?q?=E5=95=86=E6=A1=A3=E6=A1=88=E5=88=86=E9=85=8D=E8=BF=87=E6=BB=A4?=
 =?UTF-8?q?=E5=85=AC=E5=8F=B8=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../CustdocAutoAllotPluginInitializer.java    | 20 +++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/buildpackage/src/main/java/com/hzya/frame/plugin/lets/plugin/base/CustdocAutoAllotPluginInitializer.java b/buildpackage/src/main/java/com/hzya/frame/plugin/lets/plugin/base/CustdocAutoAllotPluginInitializer.java
index 7a672085..e8511cd9 100644
--- a/buildpackage/src/main/java/com/hzya/frame/plugin/lets/plugin/base/CustdocAutoAllotPluginInitializer.java
+++ b/buildpackage/src/main/java/com/hzya/frame/plugin/lets/plugin/base/CustdocAutoAllotPluginInitializer.java
@@ -17,6 +17,7 @@ import com.hzya.frame.plugin.lets.queryvo.CustdocRequestVo;
 import com.hzya.frame.plugin.lets.resultvo.CbdocResultVo;
 import com.hzya.frame.plugin.lets.u8cdto.ReusltStrDto;
 import com.hzya.frame.plugin.lets.util.PushDataByU8cUtil;
+import com.hzya.frame.plugin.lets.util.QueryU8CEntityUtil;
 import com.hzya.frame.plugin.lets.util.SaveOrUpdateBusinessLogUtil;
 import com.hzya.frame.plugin.lets.util.pushData.Attribute;
 import com.hzya.frame.plugin.lets.util.pushData.PushU8CByApiCode;
@@ -86,6 +87,8 @@ public class CustdocAutoAllotPluginInitializer extends PluginBaseEntity {
     private PushU8CByApiCode pushU8CByApiCode;
     @Autowired
     private IBdCubasdocDao bdCubasdocDao;
+    @Autowired
+    private QueryU8CEntityUtil queryU8CEntityUtil;
 
     @Override
     public JsonResultEntity executeBusiness(JSONObject requestJson) throws Exception {
@@ -282,7 +285,7 @@ public class CustdocAutoAllotPluginInitializer extends PluginBaseEntity {
             try {
 
                 ArrayList<CustDocAssginVo> assginvolist = new ArrayList<>();
-                List<BdCorpEntity> bdCorpEntities = null;
+                List<BdCorpEntity> corpAll = null;
                 //查询公司档案
                 try {
 //                    String corpresponse = pushDataByU8cUtil.pushU8CByCode(OverallConstant.getOverAllValue("bdcorpQuery"), "{}");
@@ -295,13 +298,26 @@ public class CustdocAutoAllotPluginInitializer extends PluginBaseEntity {
                     String data = attribute.getData();
                     Map map = JSONObject.parseObject(data, Map.class);
                     Object o = map.get("datas");
-                    bdCorpEntities = JSONObject.parseArray(JSON.toJSONString(o), BdCorpEntity.class);
+                    corpAll = JSONObject.parseArray(JSON.toJSONString(o), BdCorpEntity.class);
 
 
                 } catch (Exception e) {
                     throw new RuntimeException(e);
                 }
 
+                //过滤公司
+                List<BdCorpEntity> bdCorpEntities = new ArrayList<>();
+                //查询需要过滤的公司
+                String filterCompanyStr = queryU8CEntityUtil.queryBdDefDocByListPk();
+                String[] splitFilterCompany = filterCompanyStr.split(",");
+                List<String> asList = Arrays.asList(splitFilterCompany);
+                for (BdCorpEntity bdCorpEntity : corpAll) {
+                    String unitCode = bdCorpEntity.getUnitcode();
+                    if(!asList.contains(unitCode)){
+                        bdCorpEntities.add(bdCorpEntity);
+                    }
+                }
+
                 for (BdCorpEntity bdCorpEntity : bdCorpEntities) {
                     CustDocAssginVo custDocAssginVo = new CustDocAssginVo();
                     custDocAssginVo.setPk_cubasdoc(custcode);