From 9e28453dba6ecf93c6c2234ba38def7c41cf12f0 Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期一, 20 一月 2025 18:25:51 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java |  661 ++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 633 insertions(+), 28 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 13afe3e..4d0d216 100644
--- a/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java
+++ b/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java
@@ -1,26 +1,74 @@
 package com.ycl.service.impl;
 
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.io.IORuntimeException;
+import cn.hutool.core.util.CharsetUtil;
+import cn.hutool.core.util.ObjectUtil;
+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;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ycl.common.annotation.DataScope;
 import com.ycl.common.base.Result;
-import com.ycl.domain.entity.File;
-import com.ycl.domain.entity.ProjectInfo;
-import com.ycl.domain.form.ProjectInfoForm;
+import com.ycl.common.config.SysConfig;
+import com.ycl.common.constant.Constants;
+import com.ycl.common.core.domain.BaseEntity;
+import com.ycl.common.enums.business.FileTypeEnum;
+import com.ycl.common.enums.business.ImportanceTypeEnum;
+import com.ycl.common.enums.business.ProjectCategoryEnum;
+import com.ycl.common.exception.base.BaseException;
+import com.ycl.common.utils.CopyUtils;
+import com.ycl.common.utils.DateUtils;
+import com.ycl.common.utils.SecurityUtils;
+import com.ycl.common.utils.StringUtils;
+import com.ycl.common.utils.bean.BeanUtils;
+import com.ycl.common.utils.excel.OutputExcelUtils;
+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.*;
+import com.ycl.domain.query.ProjectExportQuery;
 import com.ycl.domain.query.ProjectInfoQuery;
 import com.ycl.domain.vo.*;
 import com.ycl.framework.utils.PageUtil;
-import com.ycl.mapper.ProjectInfoMapper;
+import com.ycl.listener.excel.ProjectImportListener;
+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.springframework.beans.BeanUtils;
+import org.apache.commons.codec.Charsets;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.Assert;
+import org.springframework.web.multipart.MultipartFile;
 
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletResponse;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.math.BigDecimal;
+import java.math.RoundingMode;
+import java.net.URLEncoder;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.StandardCopyOption;
+import java.text.SimpleDateFormat;
+import java.util.*;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
+import java.util.zip.ZipOutputStream;
 
 /**
  * 椤圭洰绠$悊鍩虹淇℃伅琛� 鏈嶅姟瀹炵幇绫�
@@ -33,6 +81,18 @@
 public class ProjectInfoServiceImpl extends ServiceImpl<ProjectInfoMapper, ProjectInfo> implements ProjectInfoService {
 
     private final ProjectInfoMapper projectInfoMapper;
+    private final ProjectInvestmentFundingMapper investmentFundingMapper;
+    private final ProjectInvestmentInfoMapper investmentInfoMapper;
+    private final ProjectInvestmentPolicyComplianceMapper policyMapper;
+    private final ProjectUnitRegistrationInfoMapper unitRegistrationInfoMapper;
+    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;
 
     /**
      * 娣诲姞
@@ -41,13 +101,36 @@
      * @return
      */
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public Result add(ProjectInfoForm form) {
         //娣诲姞鍩烘湰淇℃伅
         ProjectInfo entity = ProjectInfoForm.getEntityByForm(form, null);
-//        baseMapper.insert(entity);
+        Long userId = SecurityUtils.getUserId();
+        Long deptId = SecurityUtils.getDeptId();
+        entity.setProjectOwnerUnit(deptId);
+        entity.setCreateBy(userId);
+        entity.setUpdateBy(userId);
+        baseMapper.insert(entity);
+        addPlan(entity.getId()); //娣诲姞璁″垝琛�
         //娣诲姞鏂囦欢
         List<File> fileList = form.getFileList();
-        return Result.ok("娣诲姞鎴愬姛");
+        fileList.forEach(item -> {
+            item.setBusId(entity.getId());
+            item.setType(FileTypeEnum.PROJECT_INFO);
+        });
+        fileService.saveBatch(fileList);
+        return Result.ok("娣诲姞鎴愬姛").data(entity.getId());
+    }
+
+    public void addPlan(Long projectInfoId) {
+        Plan plan = new Plan();
+        plan.setProjectInfoId(projectInfoId);
+        plan.setReportStatus(1);
+        plan.setMonthStatus(1);
+        plan.setSeasonStatus(1);
+        plan.setYearStatus(1);
+        plan.setDeleted(0);
+        planMapper.insert(plan);
     }
 
     /**
@@ -57,13 +140,29 @@
      * @return
      */
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public Result update(ProjectInfoForm form) {
         ProjectInfo entity = baseMapper.selectById(form.getId());
-
         // 涓虹┖鎶汭llegalArgumentException锛屽仛鍏ㄥ眬寮傚父澶勭悊
         Assert.notNull(entity, "璁板綍涓嶅瓨鍦�");
-        BeanUtils.copyProperties(form, entity);
+        ProjectInfoForm.getEntityByForm(form, entity);
+        Long userId = SecurityUtils.getUserId();
+        entity.setUpdateBy(userId);
+        //鏇存柊椤圭洰淇℃伅
         baseMapper.updateById(entity);
+        List<File> fileList = form.getFileList();
+        fileList.forEach(item -> {
+            item.setId(null);
+            item.setBusId(entity.getId());
+            item.setType(FileTypeEnum.PROJECT_INFO);
+        });
+        //鍒犻櫎鍘熸湁鏂囦欢
+        QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>();
+        fileQueryWrapper.eq("type", FileTypeEnum.PROJECT_INFO.getType());
+        fileQueryWrapper.eq("bus_id", entity.getId());
+        fileMapper.delete(fileQueryWrapper);
+        //鏇挎崲鎴愮幇鏈�
+        fileService.saveBatch(fileList);
         return Result.ok("淇敼鎴愬姛");
     }
 
@@ -86,7 +185,18 @@
      * @return
      */
     @Override
-    public Result removeById(String id) {
+    @Transactional(rollbackFor = Exception.class)
+    public Result removeById(Long id) {
+        investmentFundingMapper.delete(new QueryWrapper<ProjectInvestmentFunding>().eq("project_id", id));
+        investmentInfoMapper.delete(new QueryWrapper<ProjectInvestmentInfo>().eq("project_id", id));
+        policyMapper.delete(new QueryWrapper<ProjectInvestmentPolicyCompliance>().eq("project_id", id));
+        unitRegistrationInfoMapper.delete(new QueryWrapper<ProjectUnitRegistrationInfo>().eq("project_id", id));
+
+        List<String> types = new ArrayList<>();
+        types.add(FileTypeEnum.PROJECT_INFO.getType());
+        types.add(FileTypeEnum.INVEST_POLICY.getType());
+        types.add(FileTypeEnum.DOCUMENT_INFO.getType());
+        fileMapper.delete(new QueryWrapper<File>().eq("bus_id", id).in("type", types));
         baseMapper.deleteById(id);
         return Result.ok("鍒犻櫎鎴愬姛");
     }
@@ -99,9 +209,71 @@
      */
     @Override
     public Result page(ProjectInfoQuery query) {
+        if (query.getProjectStartTime() != null) {
+            query.setProjectStartTime(DateUtils.getDayStart(query.getProjectStartTime()));
+        }
+        if (query.getProjectEndTime() != null) {
+            query.setProjectEndTime(DateUtils.getDayEnd(query.getProjectEndTime()));
+        }
+        String projectCategory = query.getProjectCategory();
+        //瀹炴柦闃舵鏈変袱涓�兼斁鍦╯ql澶勭悊
+        if (ProjectCategoryEnum.RESERVE.getType().equals(projectCategory)) {
+            query.setProjectStatus(ProjectCategoryEnum.RESERVE.getStatus());
+            query.setReserveOrPrevious(ProjectCategoryEnum.RESERVE.getCode());
+        } else if (ProjectCategoryEnum.PREVIOUS.getType().equals(projectCategory)) {
+            query.setProjectStatus(ProjectCategoryEnum.PREVIOUS.getStatus());
+            query.setReserveOrPrevious(ProjectCategoryEnum.PREVIOUS.getCode());
+        } else if (ProjectCategoryEnum.FINISH.getType().equals(projectCategory)) {
+            query.setProjectStatus(ProjectCategoryEnum.FINISH.getStatus());
+        } else if (ProjectCategoryEnum.EXCEPTION.getType().equals(projectCategory)) {
+            //TODO
+            //鍏堟煡鍑哄紓甯告祦绋嬫垨鑰呭紓甯歌繘搴︽垨鑰呭紓甯歌鍒掔殑projectId鍜屽紓甯哥绫�
+            //閫氳繃projectId鏌ュ嚭椤圭洰鏁版嵁
+            //琛ュ厖鐩稿簲鐨勫紓甯告暟鎹紙寮傚父绉嶇被銆佸紓甯告祦绋嬭妭鐐圭瓑锛�
+            List<ProjectVO> list = new ArrayList<>();
+            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);
-        return Result.ok().data(page.getRecords()).total(page.getTotal());
+        baseMapper.getPage(query, page);
+        List<ProjectInfoVO> records = page.getRecords();
+        List<ProjectVO> list = new ArrayList<>();
+        records.forEach(vo -> {
+            ProjectInfoVO.transform(vo);
+            ProjectVO projectVO = new ProjectVO();
+            copyToProjectVO(vo, projectVO);
+            //缈昏瘧椤圭洰闃舵
+            String phase = ProjectCategoryEnum.getPhaseByProjectStatus(projectVO.getProjectStatus(), projectVO.getProcessId() != null);
+            projectVO.setProjectPhase(phase);
+            list.add(projectVO);
+        });
+        return Result.ok().data(list).total(page.getTotal());
+    }
+
+    private void copyToProjectVO(ProjectInfoVO vo, ProjectVO projectVO) {
+        //蹇界暐null鍊肩殑澶嶅埗
+        CopyUtils.copyNoNullProperties(vo, projectVO);
+        if (vo.getProjectInvestmentInfo() != null)
+            CopyUtils.copyNoNullProperties(vo.getProjectInvestmentInfo(), projectVO);
+        if (vo.getProjectUnitRegistrationInfo() != null)
+            CopyUtils.copyNoNullProperties(vo.getProjectUnitRegistrationInfo(), projectVO);
+        if (vo.getProjectInvestmentPolicyCompliance() != null)
+            CopyUtils.copyNoNullProperties(vo.getProjectInvestmentPolicyCompliance(), projectVO);
+        if (vo.getProjectInvestmentFunding() != null)
+            CopyUtils.copyNoNullProperties(vo.getProjectInvestmentFunding(), projectVO);
     }
 
     /**
@@ -112,8 +284,14 @@
      */
     @Override
     public Result detail(Integer id) {
-        ProjectInfoVO vo = baseMapper.getById(id);
-        Assert.notNull(vo, "璁板綍涓嶅瓨鍦�");
+        ProjectInfo entity = baseMapper.getById(id);
+        Assert.notNull(entity, "璁板綍涓嶅瓨鍦�");
+        ProjectInfoVO vo = ProjectInfoVO.getVoByEntity(entity, null);
+        QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>();
+        fileQueryWrapper.eq("type", FileTypeEnum.PROJECT_INFO.getType());
+        fileQueryWrapper.eq("bus_id", vo.getId());
+        List<File> files = fileMapper.selectList(fileQueryWrapper);
+        vo.setFileList(files);
         return Result.ok().data(vo);
     }
 
@@ -141,17 +319,80 @@
         // {"type":"鐪侀噸鐐归」鐩�","count":0,"amount":"0.00","text":"鐪�"}]}}
         IndexCountVO indexCountVO = new IndexCountVO();
         List<IndexProPhaseCountVO> proPhaseCountVO = new ArrayList<>();
-        proPhaseCountVO.add(new IndexProPhaseCountVO("鍌ㄥ瑙勫垝闃舵", 0, "0.00", "鍌�"));
-        proPhaseCountVO.add(new IndexProPhaseCountVO("椤圭洰鍓嶆湡闃舵", 0, "0.00", "鏂�"));
-        proPhaseCountVO.add(new IndexProPhaseCountVO("瀹炴柦闃舵", 0, "0.00", "寤�"));
-        proPhaseCountVO.add(new IndexProPhaseCountVO("绔e伐鎶曠敤闃舵", 0, "0.00", "绔�"));
-        List<IndexImpTypeCountVO> impTypeCountVO = new ArrayList<>();
-        impTypeCountVO.add(new IndexImpTypeCountVO("涓�鑸」鐩�", 0, "0.00", "鏅�"));
-        impTypeCountVO.add(new IndexImpTypeCountVO("鍘块噸鐐归」鐩�", 0, "0.00", "鍘�"));
-        impTypeCountVO.add(new IndexImpTypeCountVO("甯傞噸鐐归」鐩�", 0, "0.00", "甯�"));
-        impTypeCountVO.add(new IndexImpTypeCountVO("鐪侀噸鐐归」鐩�", 0, "0.00", "鐪�"));
-        indexCountVO.setImpTypeCountVO(impTypeCountVO);
+        List<ProjectVO> projectVOS = baseMapper.homeCount(new BaseEntity());
+        List<ProjectVO> reserve = new ArrayList<>();
+        List<ProjectVO> previous = new ArrayList<>();
+        List<ProjectVO> implement = new ArrayList<>();
+        List<ProjectVO> finish = new ArrayList<>();
+        List<ProjectVO> normal = new ArrayList<>();
+        List<ProjectVO> province = new ArrayList<>();
+        List<ProjectVO> city = new ArrayList<>();
+        List<ProjectVO> county = new ArrayList<>();
+        projectVOS.forEach(item -> {
+            if (ProjectCategoryEnum.RESERVE.getDesc().equals(ProjectCategoryEnum.getPhaseByProjectStatus(item.getProjectStatus(), item.getProcessId() != null))) {
+                reserve.add(item);
+            } else if (ProjectCategoryEnum.PREVIOUS.getDesc().equals(ProjectCategoryEnum.getPhaseByProjectStatus(item.getProjectStatus(), item.getProcessId() != null))) {
+                previous.add(item);
+            } else if (ProjectCategoryEnum.IMPLEMENT.getDesc().equals(ProjectCategoryEnum.getPhaseByProjectStatus(item.getProjectStatus(), item.getProcessId() != null))) {
+                implement.add(item);
+            } else if (ProjectCategoryEnum.FINISH.getDesc().equals(ProjectCategoryEnum.getPhaseByProjectStatus(item.getProjectStatus(), item.getProcessId() != null))) {
+                finish.add(item);
+            }
+
+            if (ImportanceTypeEnum.PROVINCIAL_KEY.getType().equals(item.getImportanceType())) {
+                province.add(item);
+            } else if (ImportanceTypeEnum.SUINING_KEY.getType().equals(item.getImportanceType())) {
+                city.add(item);
+            } else if (ImportanceTypeEnum.SHEHONG_KEY.getType().equals(item.getImportanceType())) {
+                county.add(item);
+            } else if (ImportanceTypeEnum.NORMAL.getType().equals(item.getImportanceType())) {
+                normal.add(item);
+            }
+        });
+        //鍗曚綅鍏堥粯璁や负鍏�
+        proPhaseCountVO.add(new IndexProPhaseCountVO("鍌ㄥ瑙勫垝闃舵", reserve.size(), reserve.stream().
+                filter(item -> !StringUtils.isEmpty(item.getTotalInvestment()))
+                .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP))
+                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "鍌�"));
+
+        proPhaseCountVO.add(new IndexProPhaseCountVO("椤圭洰鍓嶆湡闃舵", previous.size(), previous.stream().
+                filter(item -> !StringUtils.isEmpty(item.getTotalInvestment()))
+                .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP))
+                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "鏂�"));
+
+        proPhaseCountVO.add(new IndexProPhaseCountVO("瀹炴柦闃舵", implement.size(), implement.stream().
+                filter(item -> !StringUtils.isEmpty(item.getTotalInvestment()))
+                .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP))
+                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "寤�"));
+
+        proPhaseCountVO.add(new IndexProPhaseCountVO("绔e伐鎶曠敤闃舵", finish.size(), finish.stream().
+                filter(item -> !StringUtils.isEmpty(item.getTotalInvestment()))
+                .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP))
+                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "绔�"));
+
         indexCountVO.setProPhaseCountVO(proPhaseCountVO);
+
+        List<IndexImpTypeCountVO> impTypeCountVO = new ArrayList<>();
+        impTypeCountVO.add(new IndexImpTypeCountVO("涓�鑸」鐩�", province.size(), province.stream().
+                filter(item -> !StringUtils.isEmpty(item.getTotalInvestment()))
+                .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP))
+                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "鏅�"));
+
+        impTypeCountVO.add(new IndexImpTypeCountVO("鍘块噸鐐归」鐩�", city.size(), city.stream().
+                filter(item -> !StringUtils.isEmpty(item.getTotalInvestment()))
+                .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP))
+                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "鍘�"));
+
+        impTypeCountVO.add(new IndexImpTypeCountVO("甯傞噸鐐归」鐩�", county.size(), county.stream().
+                filter(item -> !StringUtils.isEmpty(item.getTotalInvestment()))
+                .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP))
+                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "甯�"));
+
+        impTypeCountVO.add(new IndexImpTypeCountVO("鐪侀噸鐐归」鐩�", normal.size(), normal.stream().
+                filter(item -> !StringUtils.isEmpty(item.getTotalInvestment()))
+                .map(item -> new BigDecimal(item.getTotalInvestment()).setScale(2, RoundingMode.HALF_UP))
+                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("100000000"), 2, RoundingMode.HALF_UP).toString(), "鐪�"));
+        indexCountVO.setImpTypeCountVO(impTypeCountVO);
 
 
         return indexCountVO;
@@ -163,4 +404,368 @@
         map.put("processExceptionProject", 0);
         return map;
     }
+
+    @Override
+    public Result docDetail(Integer id) {
+        DocumentInfoForm documentInfoForm = new DocumentInfoForm();
+        QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>();
+        fileQueryWrapper.eq("type", FileTypeEnum.DOCUMENT_INFO.getType());
+        fileQueryWrapper.eq("bus_id", id);
+        List<File> files = fileMapper.selectList(fileQueryWrapper);
+        documentInfoForm.setFileList(files);
+        return Result.ok().data(documentInfoForm);
+    }
+
+    @Override
+    public Result addDoc(DocumentInfoForm form) {
+        List<File> fileList = form.getFileList();
+        fileList.forEach(item -> {
+            item.setId(null);
+            item.setBusId(form.getProjectId());
+            item.setType(FileTypeEnum.DOCUMENT_INFO);
+        });
+        //鍒犻櫎鍘熸湁鏂囦欢
+        QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>();
+        fileQueryWrapper.eq("type", FileTypeEnum.DOCUMENT_INFO.getType());
+        fileQueryWrapper.eq("bus_id", form.getProjectId());
+        fileMapper.delete(fileQueryWrapper);
+        //鏇挎崲鎴愮幇鏈�
+        fileService.saveBatch(fileList);
+        return Result.ok();
+    }
+
+    @Override
+    public Result getManagerFlag(Integer recordId) {
+        ProjectInfo projectInfo = baseMapper.selectById(recordId);
+        // 鍒ゆ柇褰撳墠鐢ㄦ埛id鏄惁鍦ㄤ富绠″垪琛ㄤ腑
+        String competentDepartment = projectInfo.getCompetentDepartment();
+        if (StringUtils.isEmpty(competentDepartment)) {
+            return Result.ok().data(false);
+        }
+        List<String> list = Arrays.asList(competentDepartment.split(","));
+        // 鑾峰緱褰撳墠鐢ㄦ埛id
+        Long userId = SecurityUtils.getUserId();
+        if (list.contains(userId.toString())) {
+            return Result.ok().data(true);
+        } else {
+            return Result.ok().data(false);
+        }
+    }
+
+    @Override
+    public void export(HttpServletResponse response, ProjectExportQuery query) throws IOException {
+        List<ProjectVO> data = baseMapper.selectProjectDetailByIds(query.getDataIdList());
+        List<ProjectExcelTemplate> excelList = new ArrayList<>();
+        //瀛楀吀浣滅炕璇�
+        data.forEach(item -> {
+            ProjectExcelTemplate excel = new ProjectExcelTemplate();
+            BeanUtils.copyProperties(item, excel);
+            //椤圭洰闃舵
+            excel.setProjectPhase(ProjectCategoryEnum.getPhaseByProjectStatus(item.getProjectStatus(), item.getProcessId() != null));
+            excelList.add(excel);
+        });
+        Set<Integer> indexes = OutputExcelUtils.getSelectFields(query.getFieldList(), ProjectExcelTemplate.class);
+        //涓嶉渶瑕侀檮浠�
+        if (!query.getRequireFile()) {
+            try (ServletOutputStream outputStream = response.getOutputStream()) {
+                EasyExcel.write(outputStream, ProjectExcelTemplate.class).includeColumnIndexes(indexes).sheet("椤圭洰鍒楄〃").doWrite(excelList);
+            } catch (Exception e) {
+                log.error(e.getMessage(), e);
+                throw new BaseException("瀵煎嚭澶辫触锛�" + e.getMessage());
+            }
+        } else {
+            //闇�瑕侀檮浠�,瀵煎嚭涓簔ip
+            response.setContentType("application/zip");
+            response.setCharacterEncoding(Charsets.UTF_8.name());
+            String fileName = URLEncoder.encode("椤圭洰鍒楄〃", Charsets.UTF_8);
+            response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".zip");
+
+            java.io.File tempDir = null;
+            try {
+                //鏌ュ嚭闄勪欢
+                List<File> projectInfoFiles = fileMapper.selectList(new QueryWrapper<File>()
+                        .in("bus_id", data.stream().map(ProjectVO::getId).collect(Collectors.toList()))
+                        .and(q -> q.eq("type", FileTypeEnum.PROJECT_INFO.getType()).or().eq("type", FileTypeEnum.DOCUMENT_INFO.getType()))
+                );
+                List<File> policyFiles = fileMapper.selectList(new QueryWrapper<File>()
+                        .in("bus_id", data.stream().map(ProjectVO::getPolicyId).collect(Collectors.toList()))
+                        .eq("type", FileTypeEnum.INVEST_POLICY.getType())
+                );
+                projectInfoFiles.addAll(policyFiles);
+                projectInfoFiles.forEach(file -> {
+                    data.forEach(item -> {
+                        if (FileTypeEnum.PROJECT_INFO.equals(file.getType())) {
+                            if (file.getBusId().equals(item.getId())) {
+                                //瀹℃壒璁″垝涔�
+                                item.setApprovalPlan(file.getOriginalName());
+                            }
+                        } else if (FileTypeEnum.DOCUMENT_INFO.equals(file.getType())) {
+                            if (file.getBusId().equals(item.getId())) {
+                                //鐩稿叧鏂囦功
+                                item.setDocuments(file.getOriginalName());
+                            }
+                        } else if (FileTypeEnum.INVEST_POLICY.equals(file.getType())) {
+                            if (file.getBusId().equals(item.getPolicyId())) {
+                                //绗﹀悎浜т笟鏀跨瓥闄勪欢
+                                item.setPolicyComplianceAttachment(file.getOriginalName());
+                            }
+                        }
+                    });
+
+                });
+                // 鍒涘缓涓存椂鐩綍
+                tempDir = Files.createTempDirectory("temp").toFile();
+
+                java.io.File templateDir = new java.io.File(tempDir, "template");
+                if (!templateDir.exists()) {
+                    templateDir.mkdirs();
+                }
+
+                // 鍒涘缓 Excel 鏂囦欢
+                java.io.File excelFile = new java.io.File(templateDir, "excel.xlsx");
+                if (!excelFile.exists()) {
+                    excelFile.createNewFile();
+                }
+
+                // 鍐欏叆 Excel 妯℃澘鏁版嵁
+                try (FileOutputStream fileOutputStream = new FileOutputStream(excelFile, false)) {
+                    EasyExcel.write(fileOutputStream, ProjectExcelTemplate.class).sheet("椤圭洰鍒楄〃").doWrite(excelList);
+                }
+
+                // 鍒涘缓闄勪欢鐩綍
+                java.io.File attachmentDir = new java.io.File(templateDir, "attachment");
+                if (!attachmentDir.exists()) {
+                    attachmentDir.mkdirs();
+                }
+
+                //灏嗛檮浠跺鍒跺鐩綍涓�
+                for (File item : projectInfoFiles) {
+                    String url = item.getUrl();
+                    if (StringUtils.isBlank(url)) continue;
+                    url = url.replaceFirst(Constants.RESOURCE_PREFIX, SysConfig.getProfile());
+                    // 鐩存帴浣跨敤File绫诲鐞嗙郴缁熻矾寰�
+                    java.io.File sourceFile = new java.io.File(url);
+                    if (!sourceFile.exists() || !sourceFile.isFile()) {
+                        // 澶勭悊鏂囦欢涓嶅瓨鍦ㄦ垨涓嶆槸鏂囦欢鐨勬儏鍐�
+                        continue; // 鎴栬�呰褰曟棩蹇椼�佹姏鍑哄紓甯哥瓑
+                    }
+                    // 鐩爣鏂囦欢璺緞锛屼娇鐢╝ttachmentDir浣滀负鐖剁洰褰曪紝骞朵繚鎸佸師鏂囦欢鍚�
+                    java.io.File targetFile = new java.io.File(attachmentDir, sourceFile.getName());
+                    // 澶嶅埗鏂囦欢
+                    try {
+                        Files.copy(sourceFile.toPath(), targetFile.toPath(), StandardCopyOption.REPLACE_EXISTING);
+                    } catch (IOException e) {
+                        log.error(e.getMessage());
+                    }
+                }
+                // 鎵撳寘 ZIP 鏂囦欢
+                java.io.File zipFile = ZipUtil.zip(templateDir);
+                byte[] zipBytes = Files.readAllBytes(zipFile.toPath());
+
+                // 灏� ZIP 鏂囦欢鍐欏叆鍝嶅簲
+                try (ServletOutputStream outputStream = response.getOutputStream()) {
+                    outputStream.write(zipBytes);
+                }
+            } finally {
+                OutputExcelUtils.deleteDirectoryOrFile(tempDir);
+            }
+        }
+    }
+
+    @Override
+    public Result updateUsedStatus(Integer id, Integer usedStatus) {
+        ProjectInfo entity = baseMapper.selectById(id);
+        // 涓虹┖鎶汭llegalArgumentException锛屽仛鍏ㄥ眬寮傚父澶勭悊
+        Assert.notNull(entity, "璁板綍涓嶅瓨鍦�");
+        entity.setUsedStatus(usedStatus);
+        baseMapper.updateById(entity);
+        return Result.ok("鍒犻櫎鎴愬姛");
+    }
+
+    @Transactional
+    @Override
+    public void importProject(MultipartFile file) {
+        java.io.File tempZip = null;
+        java.io.File tempDir = null;
+        try {
+            tempZip = java.io.File.createTempFile("temp", ".zip");
+            try (InputStream inputStream = file.getInputStream()) {
+                Files.copy(inputStream, tempZip.toPath(), StandardCopyOption.REPLACE_EXISTING);
+            }
+
+            tempDir = Files.createTempDirectory("temp-dir").toFile();
+
+            try {
+                ZipUtil.unzip(tempZip, tempDir, CharsetUtil.CHARSET_UTF_8);
+            } catch (IORuntimeException e) {
+                log.error("瑙e帇澶辫触锛屽皾璇曚娇鐢℅BK缂栫爜瑙e帇...");
+                ZipUtil.unzip(tempZip, tempDir, CharsetUtil.CHARSET_GBK);
+            }
+
+            Path path = tempDir.toPath();
+            ProjectImportListener projectImportListener = new ProjectImportListener();
+            //闇�瑕佸瓨鍏ユ暟鎹簱
+            ArrayList<File> attachments = new ArrayList<>();
+            //瀹為檯鐨勬枃浠�
+            List<java.io.File> files = new ArrayList<>();
+            try (Stream<Path> pathStream = Files.walk(path)) {
+                pathStream.forEach(filePath -> {
+                    if (Files.isDirectory(filePath) && filePath.getFileName().toString().equals("attachment")) {
+                        //闄勪欢澶勭悊
+                        try (Stream<Path> stream = Files.walk(filePath)) {
+                            stream.forEach(attachmentPath -> {
+                                if (Files.isRegularFile(attachmentPath)) {
+                                    files.add(attachmentPath.toFile());
+                                }
+                            });
+                        } catch (IOException e) {
+                            log.error(e.getMessage());
+                            throw new RuntimeException(e);
+                        }
+                    } else if (Files.isRegularFile(filePath) && filePath.getFileName().toString().equals("excel.xlsx")) {
+                        //瀵煎叆鏁版嵁澶勭悊
+                        try (FileInputStream fileInputStream = new FileInputStream(filePath.toFile())) {
+                            ExcelUtil.importExcel(fileInputStream, ProjectExcelTemplate.class, projectImportListener);
+                        } catch (IOException e) {
+                            log.error(e.getMessage());
+                            throw new RuntimeException(e);
+                        }
+                    }
+                });
+            }
+            /* 鎶婇檮浠跺瓨鍏ヤ笂浼犺矾寰� */
+            //涓婁紶鏂囦欢璺緞
+            String filePath = SysConfig.getUploadPath();
+            files.forEach(attachmentFile -> {
+                try {
+                    //涓婁紶
+                    String url = FileUploadUtils.uploadIOFile(filePath, attachmentFile);
+                    //瀛樻斁鐨勬枃浠跺悕浼氬姞涓奯鏃堕棿鎴崇殑鍚庣紑
+                    String newName = FileUtils.getName(url);
+                    attachments.add(new File()
+                            .setUrl(url)
+                            .setName(newName)
+                            .setOriginalName(attachmentFile.getName()));
+                } catch (IOException e) {
+                    e.printStackTrace();
+                }
+            });
+
+            Map<String, Long> projectInfoMap = projectImportListener.getProjectInfoMap();
+            Map<String, Long> investmentProjectPolicyComplianceMap = projectImportListener.getInvestmentProjectPolicyComplianceMap();
+            Map<String, Long> documentsMap = projectImportListener.getDocumentsMap();
+            //琛ュ厖type鍜宐usId瀛楁
+            attachments.forEach(attachment -> {
+                String fileName = attachment.getOriginalName();
+                if (ObjectUtil.isNotNull(projectInfoMap.get(fileName))) {
+                    attachment.setBusId(projectInfoMap.get(fileName));
+                    attachment.setType(FileTypeEnum.PROJECT_INFO);
+                } else if (ObjectUtil.isNotNull(investmentProjectPolicyComplianceMap.get(fileName))) {
+                    attachment.setBusId(investmentProjectPolicyComplianceMap.get(fileName));
+                    attachment.setType(FileTypeEnum.INVEST_POLICY);
+                } else if (ObjectUtil.isNotNull(documentsMap.get(fileName))) {
+                    attachment.setBusId(documentsMap.get(fileName));
+                    attachment.setType(FileTypeEnum.DOCUMENT_INFO);
+                }
+            });
+
+            // 娣诲姞椤圭洰闄勪欢鍏宠仈琛�
+            fileService.saveBatch(attachments);
+        } catch (IOException e) {
+            log.error(e.getMessage());
+            throw new RuntimeException(e);
+        } finally {
+            deleteDirectoryOrFile(tempZip);
+            deleteDirectoryOrFile(tempDir);
+        }
+    }
+
+    @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;
+        }
+
+        if (file.isDirectory()) {
+            java.io.File[] files = file.listFiles();
+            if (files != null) {
+                for (java.io.File f : files) {
+                    deleteDirectoryOrFile(f);
+                }
+            }
+        }
+        file.delete();
+    }
+
+    /**
+     * 椤圭洰鍚嶇О鏍¢獙閲嶅
+     *
+     * @return
+     */
+    public void checkProjectNameUnique(ProjectInfo projectInfo) {
+        if (StrUtil.isBlank(projectInfo.getProjectName())) {
+            throw new BaseException("椤圭洰鍚嶇О涓嶈兘涓虹┖");
+        }
+        Assert.isTrue(baseMapper.selectCount(Wrappers.<ProjectInfo>lambdaQuery().eq(ProjectInfo::getProjectName, projectInfo.getProjectName())) == 0, "椤圭洰鍚嶇О閲嶅");
+    }
+
 }

--
Gitblit v1.8.0