射洪市项目管理系统
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
xiangpei
2025-02-22 d08fd7c3f0d3b11c19314b8f9677ee8626d50a49 
 business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java


@@ -1,19 +1,25 @@


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.query.LambdaQueryWrapper;


import com.baomidou.mybatisplus.core.conditions.Wrapper;


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.base.Result;


import com.ycl.common.config.SysConfig;


import com.ycl.common.constant.Constants;


import com.ycl.common.core.domain.BaseEntity;


import com.ycl.common.core.domain.entity.SysDictData;


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.enums.business.ProjectStatusEnum;


import com.ycl.common.exception.base.BaseException;


import com.ycl.common.utils.CopyUtils;


import com.ycl.common.utils.DateUtils;


@@ -21,38 +27,47 @@


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.constant.ProjectConstant;


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.*;


import com.ycl.framework.utils.PageUtil;


import com.ycl.listener.excel.ProjectImportListener;


import com.ycl.mapper.*;


import com.ycl.service.FileService;


import com.ycl.service.ProjectInfoService;


import com.ycl.service.*;


import com.ycl.system.mapper.SysDeptMapper;


import com.ycl.system.service.ISysDeptService;


import com.ycl.system.service.ISysDictDataService;


import lombok.RequiredArgsConstructor;


import org.apache.commons.codec.Charsets;


import org.springframework.aop.framework.AopContext;


import org.springframework.stereotype.Service;


import org.springframework.transaction.annotation.Transactional;


import org.springframework.util.Assert;


import org.springframework.util.CollectionUtils;


import org.springframework.web.multipart.MultipartFile;




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.zip.ZipOutputStream;


import java.util.stream.Stream;




/**


 * 项目管理基础信息表 服务实现类


@@ -65,6 +80,8 @@


public class ProjectInfoServiceImpl extends ServiceImpl<ProjectInfoMapper, ProjectInfo> implements ProjectInfoService {




    private final ProjectInfoMapper projectInfoMapper;


    private final ProjectInfoWinUnitService projectInfoWinUnitService;


    private final ProjectInfoYearPlanService projectInfoYearPlanService;


    private final ProjectInvestmentFundingMapper investmentFundingMapper;


    private final ProjectInvestmentInfoMapper investmentInfoMapper;


    private final ProjectInvestmentPolicyComplianceMapper policyMapper;


@@ -72,6 +89,13 @@


    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;


    private final ISysDictDataService dictDataService;






    /**


     * 添加


@@ -80,13 +104,21 @@


     * @return


     */


    @Override


    @Transactional(rollbackFor = Exception.class)


    public Result add(ProjectInfoForm form) {


        //添加基本信息


        ProjectInfo entity = ProjectInfoForm.getEntityByForm(form, null);


        Long userId = SecurityUtils.getUserId();


//        Long deptId = SecurityUtils.getDeptId();


//        entity.setProjectOwnerUnit(deptId);


        entity.setCreateBy(userId);


        entity.setUpdateBy(userId);


        //如果是储备项目(未开工项目)不需要审核


        if (ProjectConstant.COMMIT.equals(entity.getUsedStatus()) && ProjectStatusEnum.PENDDING.getType().equals(entity.getProjectStatus())) {


            entity.setUsedStatus(ProjectConstant.PASS);


        }


        if(!checkProjectNameAndIdIsUnique(entity,null)){


            throw new RuntimeException("项目重复，项目名称、项目码已存在");


        }


        baseMapper.insert(entity);


        addPlan(entity.getId()); //添加计划表


        //添加文件


@@ -96,7 +128,39 @@


            item.setType(FileTypeEnum.PROJECT_INFO);


        });


        fileService.saveBatch(fileList);




        //添加项目中标单位


        saveUnit(entity.getId(), form.getWinUnitList());


        //添加年度投资计划


        saveYearPlan(entity.getId(), form.getYearPlanList());


        return Result.ok("添加成功").data(entity.getId());


    }




    /**


     * 判断项目名和项目码是否唯一


     * @param entity from true 来自add() false 来自edit()


     * @return


     */


    public boolean  checkProjectNameAndIdIsUnique(ProjectInfo entity,Long updateId){


        QueryWrapper<ProjectInfo> queryWrapper = new QueryWrapper<ProjectInfo>();


        if (updateId == null) {




            queryWrapper.eq("project_name", entity.getProjectName());




            if (StringUtils.isNotEmpty(entity.getProjectCode())) {


                queryWrapper.eq("project_code", entity.getProjectCode());


            }




            List<ProjectInfo> project = projectInfoMapper.selectList(queryWrapper);




            return project == null || project.isEmpty();


        }else {




            List<ProjectInfo> project = projectInfoMapper.checkProjectNameAndIdIsUnique(updateId,entity.getProjectName(),entity.getProjectCode());




            return project == null || project.isEmpty();


        }




    }




    public void addPlan(Long projectInfoId) {


@@ -117,14 +181,21 @@


     * @return


     */


    @Override


    @Transactional(rollbackFor = Exception.class)


    public Result update(ProjectInfoForm form) {


        ProjectInfo entity = baseMapper.selectById(form.getId());




        // 为空抛IllegalArgumentException，做全局异常处理


        Assert.notNull(entity, "记录不存在");


        ProjectInfoForm.getEntityByForm(form, entity);


        Long userId = SecurityUtils.getUserId();


        entity.setUpdateBy(userId);


        //如果是储备项目(未开工项目)不需要审核


        if (ProjectConstant.COMMIT.equals(entity.getUsedStatus()) && ProjectStatusEnum.PENDDING.getType().equals(entity.getProjectStatus())) {


            entity.setUsedStatus(ProjectConstant.PASS);


        }


        if(!checkProjectNameAndIdIsUnique(entity,form.getId())){


            throw new RuntimeException("项目重复，项目名称、项目码已存在");


        }


        //更新项目信息


        baseMapper.updateById(entity);


        List<File> fileList = form.getFileList();


@@ -140,7 +211,49 @@


        fileMapper.delete(fileQueryWrapper);


        //替换成现有


        fileService.saveBatch(fileList);




        //修改项目中标单位


        //删除原数据


        QueryWrapper<ProjectInfoWinUnit> unitWrapper = new QueryWrapper<>();


        unitWrapper.eq("project_info_id", form.getId());


        projectInfoWinUnitService.remove(unitWrapper);


        //添加项目中标单位


        saveUnit(entity.getId(), form.getWinUnitList());




        //修改年度投资计划


        //删除原数据


        QueryWrapper<ProjectInfoYearPlan> yearPlanWrapper = new QueryWrapper<>();


        unitWrapper.eq("project_info_id", form.getId());


        projectInfoYearPlanService.remove(yearPlanWrapper);


        //添加年度投资计划


        saveYearPlan(entity.getId(), form.getYearPlanList());


        return Result.ok("修改成功");


    }




    private void saveUnit(Long id, List<ProjectInfoWinUnitForm> projectInfoWinUnitForm) {


        List<ProjectInfoWinUnit> projectInfoWinUnits = new ArrayList<>();


        for (ProjectInfoWinUnitForm winUnitForm : projectInfoWinUnitForm) {


            ProjectInfoWinUnit projectInfoWinUnit = ProjectInfoWinUnitForm.getEntityByForm(winUnitForm, null);


            projectInfoWinUnit.setId(null);


            projectInfoWinUnit.setProjectInfoId(id);


            projectInfoWinUnits.add(projectInfoWinUnit);


        }


        if (!CollectionUtils.isEmpty(projectInfoWinUnits)) {


            projectInfoWinUnitService.saveBatch(projectInfoWinUnits);


        }


    }




    private void saveYearPlan(Long id, List<ProjectInfoYearPlanForm> yearPlanList) {


        List<ProjectInfoYearPlan> projectInfoYearPlans = new ArrayList<>();


        for (ProjectInfoYearPlanForm yearPlanForm : yearPlanList) {


            ProjectInfoYearPlan yearPlan = ProjectInfoYearPlanForm.getEntityByForm(yearPlanForm, null);


            yearPlan.setId(null);


            yearPlan.setProjectInfoId(id);


            projectInfoYearPlans.add(yearPlan);


        }


        if (!CollectionUtils.isEmpty(projectInfoYearPlans)) {


            projectInfoYearPlanService.saveBatch(projectInfoYearPlans);


        }


    }




    /**


@@ -168,6 +281,8 @@


        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));


        projectInfoWinUnitService.remove(new QueryWrapper<ProjectInfoWinUnit>().eq("project_info_id", id));


        projectInfoYearPlanService.remove(new QueryWrapper<ProjectInfoYearPlan>().eq("project_info_id", id));




        List<String> types = new ArrayList<>();


        types.add(FileTypeEnum.PROJECT_INFO.getType());


@@ -179,7 +294,7 @@


    }




    /**


     * 分页查询


     * 分页查询，TODO 项目中心的分页使用单独的接口，去掉写死的逻辑换成dept or查询


     *


     * @param query


     * @return


@@ -211,18 +326,41 @@


            return Result.ok().data(list).total(0);


        }




        Long userId = SecurityUtils.getUserId();


        String ancestors = sysDeptMapper.selectAncestors(userId);


        String[] ancestorArr = ancestors.split(",");


        List<String> ancestorList = Arrays.stream(ancestorArr).collect(Collectors.toList());


        ancestorList.add(SecurityUtils.getDeptId() + "");


        if (!CollectionUtils.isEmpty(ancestorList) && ancestorList.contains("101")) {


            query.setProjectOwnerUnit(SecurityUtils.getDeptId());


            //业主


            query.setRole(0);


        } else if (!CollectionUtils.isEmpty(ancestorList) && ancestorList.contains("102")) {


            query.setCompetentDepartment(SecurityUtils.getDeptId());


            //审批 对应主管部门


            query.setRole(1);


        }




        IPage<ProjectInfoVO> page = PageUtil.getPage(query, ProjectInfoVO.class);


        baseMapper.getPage(page, query);


        baseMapper.getPage(query, page);


        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);


            //翻译项目阶段


            List<Long> departmentList = vo.getCompetentDepartmentList();


            projectVO.setAuditRole(SecurityUtils.isAdmin(SecurityUtils.getUserId()) || (!CollectionUtils.isEmpty(departmentList) && departmentList.contains(SecurityUtils.getDeptId())));


            // 翻译项目阶段


            String phase = ProjectCategoryEnum.getPhaseByProjectStatus(projectVO.getProjectStatus(), projectVO.getProcessId() != null);


            projectVO.setProjectPhase(phase);


            if (StringUtils.isNotEmpty(projectVO.getProjectSubType())) {


                // 翻译项目子类型


                SysDictData dictData = dictDataService.selectDictDataByValue(projectVO.getProjectType(), projectVO.getProjectSubType(), "sys_project_type");


                if (Objects.nonNull(dictData)) {


                    projectVO.setProjectSubType(dictData.getDictLabel());


                }


            }


            list.add(projectVO);


        });


        return Result.ok().data(list).total(page.getTotal());


@@ -252,11 +390,33 @@


        ProjectInfo entity = baseMapper.getById(id);


        Assert.notNull(entity, "记录不存在");


        ProjectInfoVO vo = ProjectInfoVO.getVoByEntity(entity, null);


        List<Long> departmentList = vo.getCompetentDepartmentList();


        vo.setAuditRole(SecurityUtils.isAdmin(SecurityUtils.getUserId()) || (!CollectionUtils.isEmpty(departmentList) && departmentList.contains(SecurityUtils.getDeptId())));


        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);




        //中标单位


        List<ProjectInfoWinUnitVO> unitVOS = new ArrayList<>();


        List<ProjectInfoWinUnit> unitList = projectInfoWinUnitService.list(new QueryWrapper<ProjectInfoWinUnit>().eq("project_info_id", id));


        for (ProjectInfoWinUnit projectInfoWinUnit : unitList) {


            ProjectInfoWinUnitVO unitVO = ProjectInfoWinUnitVO.getVoByEntity(projectInfoWinUnit, null);


            unitVO.setId(null);


            unitVOS.add(unitVO);


        }


        vo.setWinUnitList(unitVOS);


        //年度投资计划


        List<ProjectInfoYearPlanVO> yearPlanVOS = new ArrayList<>();


        List<ProjectInfoYearPlan> yearPlaList = projectInfoYearPlanService.list(new QueryWrapper<ProjectInfoYearPlan>().eq("project_info_id", id));


        for (ProjectInfoYearPlan yearPlan : yearPlaList) {


            ProjectInfoYearPlanVO yearPlanVO = ProjectInfoYearPlanVO.getVoByEntity(yearPlan, null);


            yearPlanVO.setId(null);


            yearPlanVOS.add(yearPlanVO);


        }


        vo.setYearPlanList(yearPlanVOS);


        return Result.ok().data(vo);


    }




@@ -284,7 +444,7 @@


        // {"type":"省重点项目","count":0,"amount":"0.00","text":"省"}]}}


        IndexCountVO indexCountVO = new IndexCountVO();


        List<IndexProPhaseCountVO> proPhaseCountVO = new ArrayList<>();


        List<ProjectVO> projectVOS = baseMapper.homeCount();


        List<ProjectVO> projectVOS = baseMapper.homeCount(new BaseEntity());


        List<ProjectVO> reserve = new ArrayList<>();


        List<ProjectVO> previous = new ArrayList<>();


        List<ProjectVO> implement = new ArrayList<>();


@@ -310,7 +470,8 @@


                city.add(item);


            } else if (ImportanceTypeEnum.SHEHONG_KEY.getType().equals(item.getImportanceType())) {


                county.add(item);


            } else if (ImportanceTypeEnum.NORMAL.getType().equals(item.getImportanceType())) {


            } else // if  (ImportanceTypeEnum.NORMAL.getType().equals(item.getImportanceType()))


            {


                normal.add(item);


            }


        });


@@ -318,48 +479,46 @@


        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(), "储"));


                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("10000"), 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(), "新"));


                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("10000"), 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(), "建"));


                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("10000"), 2, RoundingMode.HALF_UP).toString(), "建"));




        proPhaseCountVO.add(new IndexProPhaseCountVO("竣工投用阶段", 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(), "竣"));


                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("10000"), 2, RoundingMode.HALF_UP).toString(), "竣"));




        indexCountVO.setProPhaseCountVO(proPhaseCountVO);




        List<IndexImpTypeCountVO> impTypeCountVO = new ArrayList<>();


        impTypeCountVO.add(new IndexImpTypeCountVO("一般项目", province.size(), province.stream().


        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(), "普"));


                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("10000"), 2, RoundingMode.HALF_UP).toString(), "普"));




        impTypeCountVO.add(new IndexImpTypeCountVO("县重点项目", city.size(), city.stream().


        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(), "县"));


                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("10000"), 2, RoundingMode.HALF_UP).toString(), "县"));




        impTypeCountVO.add(new IndexImpTypeCountVO("市重点项目", county.size(), county.stream().


        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(), "市"));


                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("10000"), 2, RoundingMode.HALF_UP).toString(), "市"));




        impTypeCountVO.add(new IndexImpTypeCountVO("省重点项目", normal.size(), normal.stream().


        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(), "省"));


                .reduce(BigDecimal.ZERO, BigDecimal::add).divide(new BigDecimal("10000"), 2, RoundingMode.HALF_UP).toString(), "省"));


        indexCountVO.setImpTypeCountVO(impTypeCountVO);






        return indexCountVO;


    }




@@ -404,7 +563,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(","));


@@ -546,4 +705,193 @@


        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("解压失败，尝试使用GBK编码解压...");


                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和busId字段


            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


    @Transactional(rollbackFor = Exception.class)


    public Result editProject(ProjectForm form) {


        if (ObjectUtil.isNotNull(form.getProjectInfoForm())) {




            if (ObjectUtil.isNull(form.getProjectInfoForm().getId())) {


                //新增


                ((ProjectInfoServiceImpl) AopContext.currentProxy()).add(form.getProjectInfoForm());


            } else {


                //更新


                ((ProjectInfoServiceImpl) AopContext.currentProxy()).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, "项目名称重复");


    }




}