射洪市项目管理系统
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
fuliqi
2024-12-02 1462476c93011079d6cec65be2877729571bba16 
 business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java


@@ -1,19 +1,30 @@


package com.ycl.service.impl;




import cn.hutool.core.collection.CollUtil;


import cn.hutool.core.util.ObjectUtil;


import cn.hutool.core.util.ZipUtil;


import com.alibaba.excel.EasyExcel;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;


import com.baomidou.mybatisplus.core.metadata.IPage;


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


import com.ycl.domain.excel.ProjectExcelTemplate;


import com.ycl.domain.form.DocumentInfoForm;


import com.ycl.domain.form.ProjectInfoForm;


import com.ycl.domain.query.ProjectExportQuery;


import com.ycl.domain.query.ProjectInfoQuery;


import com.ycl.domain.vo.*;


import com.ycl.framework.utils.PageUtil;


@@ -21,14 +32,25 @@


import com.ycl.service.FileService;


import com.ycl.service.ProjectInfoService;


import lombok.RequiredArgsConstructor;


import org.apache.commons.codec.Charsets;


import org.springframework.stereotype.Service;


import org.springframework.transaction.annotation.Transactional;


import org.springframework.util.Assert;




import javax.servlet.ServletOutputStream;


import javax.servlet.http.HttpServletResponse;


import java.io.FileOutputStream;


import java.io.IOException;


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


import java.text.SimpleDateFormat;


import java.util.*;


import java.util.stream.Collectors;


import java.util.zip.ZipOutputStream;




/**


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


@@ -156,6 +178,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());


@@ -375,4 +398,124 @@


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


            }


        }


    }


}