From 02f8fa00344f77c121fcad5ead5ca97dd390a4f8 Mon Sep 17 00:00:00 2001 From: renkang <8417338+k94314517@user.noreply.gitee.com> Date: 星期二, 26 十一月 2024 16:29:05 +0800 Subject: [PATCH] 客户资料 巡检任务业务 --- server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwProjectServiceImpl.java | 95 ++++++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 93 insertions(+), 2 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..47c6138 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,26 @@ 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.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 +38,15 @@ @Autowired private YwProjectMapper ywProjectMapper; + + @Autowired + private YwFloorMapper ywFloorMapper; + + @Autowired + private YwRoomMapper ywRoomMapper; + + @Autowired + private YwBuildingMapper ywBuildingMapper; @Override public Integer create(YwProject model) { @@ -162,4 +179,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