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