From 8f68d17500c321405c0abad309f7f0dc443c6072 Mon Sep 17 00:00:00 2001
From: luohairen <3399054449@qq.com>
Date: 星期三, 04 十二月 2024 09:34:13 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'
---
business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java | 145 ++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 144 insertions(+), 1 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 08b62f6..c9b5b18 100644
--- a/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java
+++ b/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java
@@ -1,19 +1,31 @@
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.LambdaQueryWrapper;
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 +33,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;
/**
* 椤圭洰绠＄悊鍩虹淇℃伅琛� 鏈嶅姟瀹炵幇绫�
@@ -163,7 +186,7 @@
} else if (ProjectCategoryEnum.PREVIOUS.getType().equals(projectCategory)) {
query.setProjectStatus(ProjectCategoryEnum.PREVIOUS.getStatus());
query.setReserveOrPrevious(ProjectCategoryEnum.PREVIOUS.getCode());
- } else if (ProjectCategoryEnum.FINISH.getType().equals(projectCategory)) {
+ } else if (ProjectCategoryEnum.FINISH.getType().equals(projectCategory)) {
query.setProjectStatus(ProjectCategoryEnum.FINISH.getStatus());
} else if (ProjectCategoryEnum.EXCEPTION.getType().equals(projectCategory)) {
//TODO
@@ -376,4 +399,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 {
+ //闇�瑕侀檮浠�,瀵煎嚭涓簔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);
+ }
+ }
+ }
}
--
Gitblit v1.8.0