From a637fbc85648a859e07661112f0190ae1c93e4cf Mon Sep 17 00:00:00 2001
From: zhanghua <314079846@qq.com>
Date: 星期四, 16 一月 2025 22:33:04 +0800
Subject: [PATCH] 地图上项目查询

---
 business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java |   98 ++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 89 insertions(+), 9 deletions(-)

diff --git a/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java b/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java
index 7a70638..2d11117 100644
--- a/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java
+++ b/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java
@@ -7,6 +7,7 @@
 import cn.hutool.core.util.StrUtil;
 import cn.hutool.core.util.ZipUtil;
 import com.alibaba.excel.EasyExcel;
+import com.baomidou.mybatisplus.core.conditions.Wrapper;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
@@ -28,11 +29,10 @@
 import com.ycl.common.utils.file.FileUploadUtils;
 import com.ycl.common.utils.file.FileUtils;
 import com.ycl.common.utils.poi.ExcelUtil;
+import com.ycl.controller.ProjectInfoController;
 import com.ycl.domain.entity.*;
 import com.ycl.domain.excel.ProjectExcelTemplate;
-import com.ycl.domain.form.DocumentInfoForm;
-import com.ycl.domain.form.PlanForm;
-import com.ycl.domain.form.ProjectInfoForm;
+import com.ycl.domain.form.*;
 import com.ycl.domain.query.ProjectExportQuery;
 import com.ycl.domain.query.ProjectInfoQuery;
 import com.ycl.domain.vo.*;
@@ -41,6 +41,7 @@
 import com.ycl.mapper.*;
 import com.ycl.service.FileService;
 import com.ycl.service.ProjectInfoService;
+import com.ycl.system.mapper.SysDeptMapper;
 import lombok.RequiredArgsConstructor;
 import org.apache.commons.codec.Charsets;
 import org.springframework.stereotype.Service;
@@ -85,6 +86,11 @@
     private final FileService fileService;
     private final FileMapper fileMapper;
     private final PlanMapper planMapper;
+    private final ProjectInvestmentInfoServiceImpl projectInvestmentInfoServiceImpl;
+    private final ProjectInvestmentFundingServiceImpl projectInvestmentFundingServiceImpl;
+    private final ProjectInvestmentPolicyComplianceServiceImpl projectInvestmentPolicyComplianceServiceImpl;
+    private final ProjectUnitRegistrationInfoServiceImpl projectUnitRegistrationInfoServiceImpl;
+    private final SysDeptMapper sysDeptMapper;
 
     /**
      * 娣诲姞
@@ -224,13 +230,25 @@
             return Result.ok().data(list).total(0);
         }
 
+        Long userId = SecurityUtils.getUserId();
+        String ancestors = sysDeptMapper.selectAncestors(userId);
+        String[] ancestorArr = ancestors.split(",");
+        for (String ancestor : ancestorArr) {
+            if ("101".equals(ancestor)) { // 涓氫富
+                query.setCreateBy(userId);
+                query.setRole(0);
+            } else if ("102".equals(ancestor)) { // 瀹℃壒
+                query.setUsedStatus(1);
+                query.setRole(1);
+            }
+        }
+
         IPage<ProjectInfoVO> page = PageUtil.getPage(query, ProjectInfoVO.class);
         baseMapper.getPage(page, query);
         List<ProjectInfoVO> records = page.getRecords();
         List<ProjectVO> list = new ArrayList<>();
         records.forEach(vo -> {
             ProjectInfoVO.transform(vo);
-            vo.setProjectColorCode("green");
             ProjectVO projectVO = new ProjectVO();
             copyToProjectVO(vo, projectVO);
             //缈昏瘧椤圭洰闃舵
@@ -417,7 +435,7 @@
         ProjectInfo projectInfo = baseMapper.selectById(recordId);
         // 鍒ゆ柇褰撳墠鐢ㄦ埛id鏄惁鍦ㄤ富绠″垪琛ㄤ腑
         String competentDepartment = projectInfo.getCompetentDepartment();
-        if (StringUtils.isEmpty(competentDepartment)){
+        if (StringUtils.isEmpty(competentDepartment)) {
             return Result.ok().data(false);
         }
         List<String> list = Arrays.asList(competentDepartment.split(","));
@@ -567,7 +585,7 @@
         java.io.File tempDir = null;
         try {
             tempZip = java.io.File.createTempFile("temp", ".zip");
-            try(InputStream inputStream = file.getInputStream()) {
+            try (InputStream inputStream = file.getInputStream()) {
                 Files.copy(inputStream, tempZip.toPath(), StandardCopyOption.REPLACE_EXISTING);
             }
 
@@ -575,9 +593,9 @@
 
             try {
                 ZipUtil.unzip(tempZip, tempDir, CharsetUtil.CHARSET_UTF_8);
-            }catch (IORuntimeException e){
-                    log.error("瑙e帇澶辫触锛屽皾璇曚娇鐢℅BK缂栫爜瑙e帇...");
-                    ZipUtil.unzip(tempZip, tempDir, CharsetUtil.CHARSET_GBK);
+            } catch (IORuntimeException e) {
+                log.error("瑙e帇澶辫触锛屽皾璇曚娇鐢℅BK缂栫爜瑙e帇...");
+                ZipUtil.unzip(tempZip, tempDir, CharsetUtil.CHARSET_GBK);
             }
 
             Path path = tempDir.toPath();
@@ -658,6 +676,66 @@
         }
     }
 
+    @Override
+    public Result editProject(ProjectForm form) {
+        if (ObjectUtil.isNotNull(form.getProjectInfoForm())) {
+            if (ObjectUtil.isNull(form.getProjectInfoForm().getId())) {
+                //鏂板
+                add(form.getProjectInfoForm());
+            } else {
+                //鏇存柊
+                update(form.getProjectInfoForm());
+            }
+        }
+        if (ObjectUtil.isNotNull(form.getProjectInvestmentInfoForm())) {
+            if (ObjectUtil.isNull(form.getProjectInvestmentInfoForm().getId())) {
+                projectInvestmentInfoServiceImpl.add(form.getProjectInvestmentInfoForm());
+            } else {
+                projectInvestmentInfoServiceImpl.update(form.getProjectInvestmentInfoForm());
+            }
+        }
+        if (ObjectUtil.isNotNull(form.getProjectInvestmentFundingForm())) {
+            if (ObjectUtil.isNull(form.getProjectInvestmentFundingForm().getId())) {
+                projectInvestmentFundingServiceImpl.add(form.getProjectInvestmentFundingForm());
+            } else {
+                projectInvestmentFundingServiceImpl.update(form.getProjectInvestmentFundingForm());
+            }
+        }
+        if (ObjectUtil.isNotNull(form.getProjectInvestmentPolicyComplianceForm())) {
+            if (ObjectUtil.isNull(form.getProjectInvestmentPolicyComplianceForm().getId())) {
+                projectInvestmentPolicyComplianceServiceImpl.add(form.getProjectInvestmentPolicyComplianceForm());
+            } else {
+                projectInvestmentPolicyComplianceServiceImpl.update(form.getProjectInvestmentPolicyComplianceForm());
+            }
+        }
+        if (ObjectUtil.isNotNull(form.getProjectUnitRegistrationInfoForm())) {
+            if (ObjectUtil.isNull(form.getProjectUnitRegistrationInfoForm().getId())) {
+                projectUnitRegistrationInfoServiceImpl.add(form.getProjectUnitRegistrationInfoForm());
+            } else {
+                projectUnitRegistrationInfoServiceImpl.update(form.getProjectUnitRegistrationInfoForm());
+            }
+        }
+        if (ObjectUtil.isNotNull(form.getDocumentInfoForm())) {
+            if (ObjectUtil.isNull(form.getDocumentInfoForm().getProjectId())) {
+                return Result.error("璇峰厛淇濆瓨鎶曡祫绠$悊鍩烘湰淇℃伅");
+            } else {
+                addDoc(form.getDocumentInfoForm());
+            }
+        }
+
+        return Result.ok("鎻愪氦鎴愬姛");
+    }
+
+    @Override
+    public Result searchByKey(String wordKey) {
+        Wrapper wrapper = null;
+        if (!StringUtils.isEmpty(wordKey)) {
+            wrapper = Wrappers.<ProjectInfo>lambdaQuery().like(ProjectInfo::getProjectName, wordKey).or().like(ProjectInfo::getProjectCode, wordKey);
+        }
+        List<ProjectInfo> list = baseMapper.selectList(wrapper);
+        return Result.ok().data(list);
+    }
+
     private static void deleteDirectoryOrFile(java.io.File file) {
         if (ObjectUtil.isNull(file)) {
             return;
@@ -673,8 +751,10 @@
         }
         file.delete();
     }
+
     /**
      * 椤圭洰鍚嶇О鏍¢獙閲嶅
+     *
      * @return
      */
     public void checkProjectNameUnique(ProjectInfo projectInfo) {

--
Gitblit v1.8.0