|
@@ -9,13 +9,13 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
import com.usky.common.core.bean.CommonPage;
|
|
|
import com.usky.common.mybatis.core.AbstractCrudService;
|
|
|
import com.usky.common.security.utils.SecurityUtils;
|
|
|
-import com.usky.fire.domain.BaseBuildAttach;
|
|
|
-import com.usky.fire.domain.BaseBuildExtinguish;
|
|
|
+import com.usky.fire.domain.*;
|
|
|
+import com.usky.fire.mapper.BaseCompanyMapper;
|
|
|
import com.usky.fire.service.BaseBuildAttachService;
|
|
|
import com.usky.fire.service.BaseBuildExtinguishService;
|
|
|
import com.usky.fire.service.BaseBuildService;
|
|
|
-import com.usky.fire.domain.BaseBuild;
|
|
|
import com.usky.fire.mapper.BaseBuildMapper;
|
|
|
+import com.usky.fire.service.BaseCompanyService;
|
|
|
import com.usky.fire.service.vo.BaseBuildAddVO;
|
|
|
import com.usky.fire.service.vo.BaseBuildExportVO;
|
|
|
import com.usky.fire.service.vo.CompanyDataVo;
|
|
@@ -47,20 +47,34 @@ public class BaseBuildServiceImpl extends AbstractCrudService<BaseBuildMapper, B
|
|
|
@Autowired
|
|
|
private BaseBuildExtinguishService baseBuildExtinguishService;
|
|
|
|
|
|
+ @Autowired
|
|
|
+ private BaseCompanyMapper baseCompanyMapper;
|
|
|
+
|
|
|
@Override
|
|
|
- public CommonPage<Object> baseBuildList(String buildNum, String buildName, Integer pageNum, Integer pageSize) {
|
|
|
+ public CommonPage<Object> baseBuildList(String buildNum, String buildName, String companyId, Integer pageNum, Integer pageSize) {
|
|
|
List<Object> list = new ArrayList<>();
|
|
|
IPage<BaseBuild> page = new Page<>(pageNum, pageSize);
|
|
|
LambdaQueryWrapper<BaseBuild> queryWrapper = Wrappers.lambdaQuery();
|
|
|
queryWrapper.select(BaseBuild::getId, BaseBuild::getBuildNum, BaseBuild::getBuildName, BaseBuild::getAddress, BaseBuild::getBuildArea,
|
|
|
BaseBuild::getFireRating, BaseBuild::getUseCharacter, BaseBuild::getBuildStructure, BaseBuild::getBuildHigh,
|
|
|
- BaseBuild::getCompleteYear, BaseBuild::getSafePerson, BaseBuild::getManagePerson, BaseBuild::getFireRisk, BaseBuild::getBuildPlan)
|
|
|
+ BaseBuild::getCompleteYear, BaseBuild::getSafePerson, BaseBuild::getManagePerson,
|
|
|
+ BaseBuild::getFireRisk, BaseBuild::getBuildPlan, BaseBuild::getCompanyId)
|
|
|
.eq(BaseBuild::getDeleteFlag, 0)
|
|
|
.like(StringUtils.isNotBlank(buildNum), BaseBuild::getBuildNum, buildNum)
|
|
|
.like(StringUtils.isNotBlank(buildName), BaseBuild::getBuildName, buildName)
|
|
|
+ .eq(StringUtils.isNotBlank(companyId), BaseBuild::getCompanyId, companyId)
|
|
|
.orderByDesc(BaseBuild::getId);
|
|
|
page = this.page(page, queryWrapper);
|
|
|
if (CollectionUtils.isNotEmpty(page.getRecords())) {
|
|
|
+ List<String> companyIdList = new ArrayList<>();
|
|
|
+ for (int j = 0; j < page.getRecords().size(); j++) {
|
|
|
+ companyIdList.add(page.getRecords().get(j).getCompanyId());
|
|
|
+ }
|
|
|
+ List<BaseCompany> list1 = new ArrayList<>();
|
|
|
+ if (CollectionUtils.isNotEmpty(companyIdList)) {
|
|
|
+ String[] a2 = companyIdList.toArray(new String[companyIdList.size()]);
|
|
|
+ list1 = baseCompanyMapper.companyNameList(a2);
|
|
|
+ }
|
|
|
for (int i = 0; i < page.getRecords().size(); i++) {
|
|
|
Map<String, Object> map = new HashMap<>();
|
|
|
map.put("id", page.getRecords().get(i).getId());
|
|
@@ -77,6 +91,13 @@ public class BaseBuildServiceImpl extends AbstractCrudService<BaseBuildMapper, B
|
|
|
map.put("managePerson", page.getRecords().get(i).getManagePerson());
|
|
|
map.put("fireRisk", page.getRecords().get(i).getFireRisk());
|
|
|
map.put("buildPlan", page.getRecords().get(i).getBuildPlan());
|
|
|
+ if (CollectionUtils.isNotEmpty(list1)) {
|
|
|
+ for (int j = 0; j < list1.size(); j++) {
|
|
|
+ if (page.getRecords().get(i).getCompanyId().equals(list1.get(j).getCompanyId())) {
|
|
|
+ map.put("companyName", list1.get(j).getCompanyName());
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
list.add(map);
|
|
|
}
|
|
|
}
|