射洪市项目管理系统
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
fuliqi
2025-01-20 9e28453dba6ecf93c6c2234ba38def7c41cf12f0 
 business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java


@@ -1,34 +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.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.domain.entity.File;


import com.ycl.domain.entity.ProjectInfo;


import com.ycl.domain.form.DocumentInfoForm;


import com.ycl.domain.form.ProjectInfoForm;


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


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




/**


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


@@ -41,8 +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;




    /**


     * 添加


@@ -56,9 +106,12 @@


        //添加基本信息


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


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


        fileList.forEach(item -> {


@@ -67,6 +120,17 @@


        });


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


    }




    /**


@@ -109,7 +173,6 @@


     * @return


     */


    @Override


    //TODO:待完善


    public Result remove(List<String> ids) {


        baseMapper.deleteBatchIds(ids);


        return Result.ok("删除成功");


@@ -122,8 +185,18 @@


     * @return


     */


    @Override


    //TODO:待完善


    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("删除成功");


    }


@@ -143,6 +216,7 @@


            query.setProjectEndTime(DateUtils.getDayEnd(query.getProjectEndTime()));


        }


        String projectCategory = query.getProjectCategory();


        //实施阶段有两个值放在sql处理


        if (ProjectCategoryEnum.RESERVE.getType().equals(projectCategory)) {


            query.setProjectStatus(ProjectCategoryEnum.RESERVE.getStatus());


            query.setReserveOrPrevious(ProjectCategoryEnum.RESERVE.getCode());


@@ -160,15 +234,27 @@


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


        baseMapper.getPage(query, page);


        List<ProjectInfoVO> records = page.getRecords();


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


        records.forEach(vo -> {


                    ProjectInfoVO.transform(vo);


                    vo.setProjectColorCode("green");


            ProjectInfoVO.transform(vo);


            ProjectVO projectVO = new ProjectVO();


            copyToProjectVO(vo,projectVO);


            copyToProjectVO(vo, projectVO);


            //翻译项目阶段


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


            projectVO.setProjectPhase(phase);


@@ -177,13 +263,17 @@


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


    }




    private void copyToProjectVO(ProjectInfoVO vo,ProjectVO projectVO) {


    private void copyToProjectVO(ProjectInfoVO vo, ProjectVO projectVO) {


        //忽略null值的复制


        CopyUtils.copyNoNullProperties(vo, projectVO);


        if(vo.getProjectInvestmentFunding()!=null) CopyUtils.copyNoNullProperties(vo.getProjectInvestmentFunding(),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.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);


    }




    /**


@@ -229,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("竣工投用阶段", 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("竣工投用阶段", 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;


@@ -286,6 +439,9 @@


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


@@ -295,4 +451,321 @@


            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 {


            //需要附件,导出为zip


            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; // 或者记录日志、抛出异常等


                    }


                    // 目标文件路径，使用attachmentDir作为父目录，并保持原文件名


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


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


        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("解压失败，尝试使用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


    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, "项目名称重复");


    }




}