From a856cfc04747d4d8f3605168531b253240d2e87c Mon Sep 17 00:00:00 2001 From: liukangdong <898885815@qq.com> Date: 星期三, 27 十一月 2024 08:55:07 +0800 Subject: [PATCH] Merge branch 'master' of http://139.186.142.91:10010/r/productDev/funingyunwei --- server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwProjectServiceImpl.java | 115 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 112 insertions(+), 3 deletions(-) diff --git a/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwProjectServiceImpl.java b/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwProjectServiceImpl.java index 5e27677..7c9d35f 100644 --- a/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwProjectServiceImpl.java +++ b/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwProjectServiceImpl.java @@ -5,18 +5,27 @@ import com.doumee.core.model.PageWrap; import com.doumee.core.utils.Constants; import com.doumee.core.utils.Utils; +import com.doumee.dao.business.YwBuildingMapper; +import com.doumee.dao.business.YwFloorMapper; import com.doumee.dao.business.YwProjectMapper; -import com.doumee.dao.business.model.Company; -import com.doumee.dao.business.model.YwProject; +import com.doumee.dao.business.YwRoomMapper; +import com.doumee.dao.business.model.*; +import com.doumee.dao.business.vo.CompanyTree; +import com.doumee.dao.business.vo.ProjectDataVO; +import com.doumee.dao.business.vo.ProjectTree; +import com.doumee.service.business.AreasService; import com.doumee.service.business.YwProjectService; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; +import org.checkerframework.checker.units.qual.A; +import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.util.CollectionUtils; +import java.util.ArrayList; import java.util.Date; import java.util.List; @@ -30,6 +39,17 @@ @Autowired private YwProjectMapper ywProjectMapper; + @Autowired + private AreasService areasService; + + @Autowired + private YwFloorMapper ywFloorMapper; + + @Autowired + private YwRoomMapper ywRoomMapper; + + @Autowired + private YwBuildingMapper ywBuildingMapper; @Override public Integer create(YwProject model) { @@ -154,7 +174,22 @@ queryWrapper.lambda().eq(YwProject::getAddr, pageWrap.getModel().getAddr()); } queryWrapper.lambda().orderByDesc(YwProject::getCreateDate); - return PageData.from(ywProjectMapper.selectPage(page, queryWrapper)); + PageData<YwProject> data = PageData.from(ywProjectMapper.selectPage(page, queryWrapper)); + if(data!=null && data.getRecords()!=null){ + for(YwProject model :data.getRecords()){ + if(model.getAreaId()!=null){ + Areas a = areasService.findById(model.getAreaId(),Constants.TWO); + if(a != null){ + model.setAreaName(a.getName()); + model.setCityId(a.getCityId()); + model.setCityName(a.getCityName()); + model.setProvinceId(a.getProvinceId()); + model.setProvinceName(a.getProvinceName()); + } + } + } + } + return data; } @Override @@ -162,4 +197,78 @@ QueryWrapper<YwProject> wrapper = new QueryWrapper<>(ywProject); return ywProjectMapper.selectCount(wrapper); } + + + @Override + public List<ProjectDataVO> projectTree(){ + List<ProjectDataVO> projectDataVOList = new ArrayList<>(); + List<YwProject> ywProjectList = ywProjectMapper.selectList(new QueryWrapper<YwProject>().lambda() + .eq(YwProject::getIsdeleted,Constants.ZERO) + .eq(YwProject::getStatus,Constants.ZERO) + .orderByAsc(YwProject::getSortnum) + ); + for (YwProject ywProject:ywProjectList) { + ProjectDataVO projectDataVO = new ProjectDataVO(); + BeanUtils.copyProperties(ywProject,projectDataVO); + projectDataVO.setLv(Constants.ZERO); + projectDataVOList.add(projectDataVO); + } + + List<YwBuilding> ywBuildingList = ywBuildingMapper.selectList(new QueryWrapper<YwBuilding>().lambda() + .eq(YwBuilding::getIsdeleted,Constants.ZERO) + .eq(YwBuilding::getStatus,Constants.ZERO) + .orderByAsc(YwBuilding::getSortnum) + ); + + for (YwBuilding data:ywBuildingList) { + ProjectDataVO projectDataVO = new ProjectDataVO(); + BeanUtils.copyProperties(data,projectDataVO); + projectDataVO.setPId(data.getProjectId()); + projectDataVO.setLv(Constants.ONE); + projectDataVOList.add(projectDataVO); + } + + + List<YwFloor> ywFloorList = ywFloorMapper.selectList(new QueryWrapper<YwFloor>().lambda() + .eq(YwFloor::getIsdeleted,Constants.ZERO) + .eq(YwFloor::getStatus,Constants.ZERO) + .orderByAsc(YwFloor::getSortnum) + ); + + for (YwFloor data:ywFloorList) { + ProjectDataVO projectDataVO = new ProjectDataVO(); + BeanUtils.copyProperties(data,projectDataVO); + projectDataVO.setPId(data.getBuildingId()); + projectDataVO.setLv(Constants.TWO); + projectDataVOList.add(projectDataVO); + } + + + + List<YwRoom> ywRoomList = ywRoomMapper.selectList(new QueryWrapper<YwRoom>().lambda() + .eq(YwRoom::getIsdeleted,Constants.ZERO) + .eq(YwRoom::getStatus,Constants.ZERO) + .orderByAsc(YwRoom::getSortnum) + ); + + + for (YwRoom data:ywRoomList) { + ProjectDataVO projectDataVO = new ProjectDataVO(); + BeanUtils.copyProperties(data,projectDataVO); + projectDataVO.setName(data.getCode()); + projectDataVO.setPId(data.getFloor()); + projectDataVO.setLv(Constants.THREE); + projectDataVOList.add(projectDataVO); + } + + ProjectTree treeBuild = new ProjectTree(projectDataVOList); + projectDataVOList = treeBuild.buildTree(); + return projectDataVOList; + + } + + + + + } -- Gitblit v1.9.3