- 来源去重

dsk-system/src/main/java/com/dsk/system/dskService/EnterpriseBussinessService.java

@@ -84,12 +84,22 @@ public class EnterpriseBussinessService {
             dataMap.put("sourceList", null);
             String source = MapUtils.getString(dataMap, "source");
             if (ObjectUtil.isNotEmpty(source)) {
+                List<EnterpriseBussinessVo> enterpriseSource = new ArrayList<>();
                 List<EnterpriseBussinessVo> enterpriseBussinessVoList = JSONUtil.toList(source, EnterpriseBussinessVo.class);
                 if (ObjectUtil.isNotEmpty(enterpriseBussinessVoList.get(0))) {
                     EnterpriseBussinessVo vo = enterpriseBussinessVoList.get(0);
                     dataMap.put("sourceId", vo.getSource_id());
                     dataMap.put("sourceType", vo.getSource_type());
                     dataMap.put("sourceList", enterpriseBussinessVoList);
+                    if (ObjectUtil.isNotEmpty(body.getSources())) {
+                        for (EnterpriseBussinessVo bussinessVo : enterpriseBussinessVoList) {
+                            String source_type = bussinessVo.getSource_type();
+                            if (body.getSources().contains(source_type)) {
+                                enterpriseSource.add(bussinessVo);
+                            }
+                        }
+                        dataMap.put("sourceList", enterpriseSource);
+                    }
                 }
             }
         }
@@ -173,12 +183,22 @@ public class EnterpriseBussinessService {
             dataMap.put("sourceType", null);
             String source = MapUtils.getString(dataMap, "source");
             if (ObjectUtil.isNotEmpty(source)) {
+                List<EnterpriseBussinessVo> enterpriseSource = new ArrayList<>();
                 List<EnterpriseBussinessVo> enterpriseBussinessVoList = JSONUtil.toList(source, EnterpriseBussinessVo.class);
                 if (ObjectUtil.isNotEmpty(enterpriseBussinessVoList.get(0))) {
                     EnterpriseBussinessVo vo = enterpriseBussinessVoList.get(0);
                     dataMap.put("sourceId", vo.getSource_id());
                     dataMap.put("sourceType", vo.getSource_type());
                     dataMap.put("sourceList", enterpriseBussinessVoList);
+                    if (ObjectUtil.isNotEmpty(body.getSources())) {
+                        for (EnterpriseBussinessVo bussinessVo : enterpriseBussinessVoList) {
+                            String source_type = bussinessVo.getSource_type();
+                            if (body.getSources().contains(source_type)) {
+                                enterpriseSource.add(bussinessVo);
+                            }
+                        }
+                        dataMap.put("sourceList", enterpriseSource);
+                    }
                 }
             }
         }