diff --git a/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/CompanyServiceImpl.java b/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/CompanyServiceImpl.java index e781f0164..4c5e9cb71 100644 --- a/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/CompanyServiceImpl.java +++ b/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/CompanyServiceImpl.java @@ -36,6 +36,7 @@ import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import java.util.*; +import java.util.stream.Collectors; /** * @Description: 企业 @@ -307,6 +308,7 @@ public class CompanyServiceImpl extends ServiceImpl impl } secondComapnyList = getGroupStatisticsCompany(secondComapnyList, childComapnyList); fistComapnyList = getGroupStatisticsCompany(fistComapnyList, secondComapnyList); + fistComapnyList = filterCompanyByProject(fistComapnyList, map); data.put("companyList", fistComapnyList); } else if (tempCompany.getCompanyType() == 2) { data.put("companyInfo", tempCompany); @@ -330,6 +332,7 @@ public class CompanyServiceImpl extends ServiceImpl impl } } fistComapnyList = getGroupStatisticsCompany(fistComapnyList, companyList); + fistComapnyList = filterCompanyByProject(fistComapnyList, map); data.put("companyList", fistComapnyList); } else if (tempCompany.getCompanyType() == 3) { Company parentCompany = companyMapper.selectById(tempCompany.getParentId()); @@ -352,6 +355,7 @@ public class CompanyServiceImpl extends ServiceImpl impl companyList = getGroupStatisticsProject(companyList, projectList); } } + companyList = filterCompanyByProject(companyList, map); data.put("companyList", companyList); } else { Company parentCompany = companyMapper.selectById(tempCompany.getParentId()); @@ -382,6 +386,14 @@ public class CompanyServiceImpl extends ServiceImpl impl return data; } + @NotNull + private List filterCompanyByProject(List fistComapnyList, Map map) { + if (StringUtils.isBlank(MapUtils.getString(map, "projectName"))) { + return fistComapnyList; + } + return fistComapnyList.stream().filter(entityMap -> MapUtils.getInteger(entityMap, "projectNum") != 0).collect(Collectors.toList()); + } + @Override public void editCompanyBySn(Company company) { QueryWrapper queryWrapper = new QueryWrapper<>(); diff --git a/src/main/java/com/zhgd/xmgl/modules/bigdevice/mapper/xml/TowerMapper.xml b/src/main/java/com/zhgd/xmgl/modules/bigdevice/mapper/xml/TowerMapper.xml index 7c236838e..33f379d36 100644 --- a/src/main/java/com/zhgd/xmgl/modules/bigdevice/mapper/xml/TowerMapper.xml +++ b/src/main/java/com/zhgd/xmgl/modules/bigdevice/mapper/xml/TowerMapper.xml @@ -397,14 +397,14 @@ and f.company_sn = #{sn} - #{sn} + and b.company_sn = #{sn} - #{sn} + and a.project_sn = #{sn} + GROUP BY a.project_id - #{sn} - GROUP BY a.project_id + and cp.company_sn = #{sn} order by devNum desc diff --git a/src/main/java/com/zhgd/xmgl/modules/project/mapper/xml/ProjectMapper.xml b/src/main/java/com/zhgd/xmgl/modules/project/mapper/xml/ProjectMapper.xml index 0fe60d1e7..e3aa82479 100644 --- a/src/main/java/com/zhgd/xmgl/modules/project/mapper/xml/ProjectMapper.xml +++ b/src/main/java/com/zhgd/xmgl/modules/project/mapper/xml/ProjectMapper.xml @@ -215,6 +215,9 @@ and cp.company_sn = #{sn} + + and a.project_name like concat('%', #{projectName}, '%') +