LK/income_payments_management
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add:工作组-全部文件导出的接口

Browse Source
This commit is contained in:
sunzheng
2024-07-09 15:06:36 +08:00
parent 8423a1b0e8
commit bd407f8dd4
3 changed files with 247 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
jero-boot-incoming-payment/src/main/java/com/jero/project/controller/PayWorkingGroupController.java
+14 -7
View File
@@ -17,7 +17,6 @@ import com.jero.common.util.oConvertUtils;
import com.jero.contract.entity.PayIncomeContract;
import com.jero.contract.service.IPayIncomeContractService;
import com.jero.mail.commont.constant;
import com.jero.meeting.entity.PayMeetingQuestion;
import com.jero.project.common.PayProjectCommon;
import com.jero.project.entity.PayWorkingGroup;
import com.jero.project.entity.PayWorkingGroupSubItem;
@@ -34,18 +33,14 @@ import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.ModelAndView;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.*;
import java.util.stream.Collectors;
/**
/**
* @Description: 工作组项目
* @Author: jeecg-boot
* @Date: 2021-08-27
@@ -466,4 +461,16 @@ public class PayWorkingGroupController extends JeroController<PayWorkingGroup, I
return super.importExcel(request, response, PayWorkingGroup.class);
}
/**
* 全部文件导出
* @param request
* @param response
*/
// @RequiresPermissions("workingGroupProject:exportAllFile")
@ApiOperation(value="全部文件导出", notes="全部文件导出")
@GetMapping(value = "/exportAllFile")
public void exportAllFile(HttpServletRequest request, HttpServletResponse response) {
payWorkingGroupService.exportAllFile(request, response);
}
}
jero-boot-incoming-payment/src/main/java/com/jero/project/service/IPayWorkingGroupService.java
+9 -1
View File
@@ -9,7 +9,8 @@ import com.jero.project.vo.PayWorkingGroupVO;
import com.jero.project.entity.PayWorkingGroup;
import com.baomidou.mybatisplus.extension.service.IService;
import com.jero.statistics.entity.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.List;
/**
@@ -248,4 +249,11 @@ public interface IPayWorkingGroupService extends IService<PayWorkingGroup> {
* @param payWorkingGroupSubItem 工作组成员
*/
void validAndSetDefaultValue(PayWorkingGroupSubItem payWorkingGroupSubItem);
/**
* 全部文件导出
* @param request
* @param response
*/
void exportAllFile(HttpServletRequest request, HttpServletResponse response);
}
jero-boot-incoming-payment/src/main/java/com/jero/project/service/impl/PayWorkingGroupServiceImpl.java
+224 -1
View File
@@ -23,6 +23,8 @@ import com.jero.contract.service.IPayIncomeContractAssociatedService;
import com.jero.meeting.common.MeetingCommon;
import com.jero.meeting.entity.PayMeeting;
import com.jero.meeting.service.IPayMeetingService;
import com.jero.modules.oss.entity.OSSFile;
import com.jero.modules.oss.service.IOSSFileService;
import com.jero.modules.system.entity.SysUser;
import com.jero.modules.system.service.ISysUserService;
import com.jero.project.common.PayProjectCommon;
@@ -30,6 +32,7 @@ import com.jero.project.entity.*;
import com.jero.project.mapper.PayWorkingGroupMapper;
import com.jero.project.mapper.PayWorkingGroupSubItemContactsMapper;
import com.jero.project.service.IPayResearchProjectService;
import com.jero.project.service.IPayWorkingGroupDataFileService;
import com.jero.project.service.IPayWorkingGroupService;
import com.jero.project.service.IPayWorkingGroupSubItemService;
import com.jero.project.vo.PayWorkingGroupVO;
@@ -43,16 +46,21 @@ import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.shiro.SecurityUtils;
import org.jetbrains.annotations.NotNull;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
/**
@@ -89,6 +97,12 @@ public class PayWorkingGroupServiceImpl extends ServiceImpl<PayWorkingGroupMappe
private PayWorkingGroupSubItemContactsMapper payWorkingGroupSubItemContactsMapper;
@Resource
private IPayMeetingService payMeetingService;
@Resource
private IPayWorkingGroupDataFileService payWorkingGroupDataFileService;
@Resource
private IOSSFileService iossFileService;
@Value("${jero.path.upload}")
private String upLoadPath;
@Override
public IPage<PayWorkingGroupVO> queryPageList(Page<PayWorkingGroupVO> page, PayWorkingGroup payWorkingGroup) {
@@ -805,5 +819,214 @@ public class PayWorkingGroupServiceImpl extends ServiceImpl<PayWorkingGroupMappe
return yearMonthList;
}
/**
* 全部文件导出
*
* @param request
* @param response
*/
@Override
public void exportAllFile(HttpServletRequest request, HttpServletResponse response) {
String selections = request.getParameter("selections");
if (StringUtils.isEmpty(selections)) {
throw new JeroBootException("至少选择一个工作组项目");
}
List<String> selectionList = Arrays.asList(selections.split(","));
Set<String> addedEntries = new HashSet<>(); // 用于记录已添加的条目路径
try (ZipOutputStream zipos = new ZipOutputStream(new BufferedOutputStream(response.getOutputStream()))) {
response.reset();
response.setCharacterEncoding("utf-8");
response.setContentType("multipart/form-data");
// 设置压缩包的名字
String downloadName = UUID.randomUUID().toString() + ".zip";
String agent = request.getHeader("USER-AGENT");
try {
if (agent.contains("MSIE") || agent.contains("Trident")) {
downloadName = java.net.URLEncoder.encode(downloadName, "UTF-8");
} else {
downloadName = new String(downloadName.getBytes("UTF-8"), "ISO-8859-1");
}
} catch (Exception e) {
e.printStackTrace();
}
response.setHeader("Content-Disposition", "attachment;fileName=\"" + downloadName + "\"");
// 处理工作组资料
for (String projectId : selectionList) {
PayWorkingGroupDataFile payWorkingGroupDataFile = new PayWorkingGroupDataFile();
payWorkingGroupDataFile.setProjectId(projectId);
payWorkingGroupDataFile.setProjectType(PayProjectCommon.PROJECT_TYPE_FILE1);
List<PayWorkingGroupDataFile> fileList = payWorkingGroupDataFileService.queryListExport(payWorkingGroupDataFile);
PayWorkingGroup payWorkingGroup = payWorkingGroupService.getById(projectId);
if (payWorkingGroup == null) {
throw new JeroBootException("工作组不存在！");
}
String groupName = payWorkingGroup.getWorkingGroupProject(); // 获取工作组名称
String groupFolderPath = "工作组资料/" + groupName + "/";
String researchProjectFolderPath = groupFolderPath + "在研项目/";
String otherFilesFolderPath = groupFolderPath + "其他分发资料/";
// 创建工作组文件夹
zipos.putNextEntry(new ZipEntry(groupFolderPath));
zipos.closeEntry();
// 创建在研项目文件夹
zipos.putNextEntry(new ZipEntry(researchProjectFolderPath));
zipos.closeEntry();
// 创建分发资料文件夹
zipos.putNextEntry(new ZipEntry(otherFilesFolderPath));
zipos.closeEntry();
// 在研项目的文件
LambdaQueryWrapper<PayResearchProject> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(PayResearchProject::getWorkingGroupId, projectId);
List<PayResearchProject> payResearchProjectList = payResearchProjectService.list(queryWrapper);
for (PayResearchProject payResearchProject : payResearchProjectList) {
String projectName = payResearchProject.getProjectName();
String projectFolderPath = researchProjectFolderPath + projectName + "/";
// 创建在研项目名文件夹
zipos.putNextEntry(new ZipEntry(projectFolderPath));
zipos.closeEntry();
List<String> researchProjectFileIds = new ArrayList<>();
if (!StringUtils.isEmpty(payResearchProject.getFiles())) {
researchProjectFileIds.addAll(Arrays.asList(payResearchProject.getFiles().split(",")));
}
// 添加在研项目的文件到在研项目名文件夹
for (String fileId : researchProjectFileIds) {
OSSFile ossFile = iossFileService.getById(fileId.trim());
if (ossFile == null) {
continue;
}
File file = new File(upLoadPath + File.separator + ossFile.getUrl());
if (!file.exists()) {
continue;
}
String entryPath = projectFolderPath + ossFile.getFileName();
// 检查是否已经添加过相同路径的条目
if (addedEntries.contains(entryPath)) {
// 如果已经存在相同路径的条目，可以修改文件名或者跳过处理
continue;
}
// 添加到压缩包中
zipos.putNextEntry(new ZipEntry(entryPath));
try (InputStream is = new FileInputStream(file)) {
byte[] b = new byte[1024];
int length;
while ((length = is.read(b)) != -1) {
zipos.write(b, 0, length);
}
} catch (IOException e) {
e.printStackTrace();
}
zipos.closeEntry(); // 关闭当前条目
addedEntries.add(entryPath); // 记录已添加的条目路径
}
}
// 添加工作组资料文件到分发资料文件夹
for (PayWorkingGroupDataFile dataFile : fileList) {
OSSFile ossFile = iossFileService.getById(dataFile.getFileId());
if (ossFile == null) {
continue;
}
File file = new File(upLoadPath + File.separator + ossFile.getUrl());
if (!file.exists()) {
continue;
}
String entryPath = otherFilesFolderPath + ossFile.getFileName();
// 检查是否已经添加过相同路径的条目
if (addedEntries.contains(entryPath)) {
// 如果已经存在相同路径的条目，可以修改文件名或者跳过处理
continue;
}
// 添加到压缩包中
zipos.putNextEntry(new ZipEntry(entryPath));
try (InputStream is = new FileInputStream(file)) {
byte[] b = new byte[1024];
int length;
while ((length = is.read(b)) != -1) {
zipos.write(b, 0, length);
}
} catch (IOException e) {
e.printStackTrace();
}
zipos.closeEntry(); // 关闭当前条目
addedEntries.add(entryPath); // 记录已添加的条目路径
}
}
// 处理工作组会议资料
for (String projectId : selectionList) {
LambdaQueryWrapper<PayMeeting> lambdaQueryWrapper = new LambdaQueryWrapper<>();
lambdaQueryWrapper.eq(PayMeeting::getWorkingGroupId, projectId);
lambdaQueryWrapper.eq(PayMeeting::getMeetingType, MeetingCommon.WORKING_GROUP_MEETING_INT);
List<PayMeeting> list = payMeetingService.list(lambdaQueryWrapper);
for (PayMeeting payMeeting : list) {
String meetingName = payMeeting.getMeetingName();
String meetingFolderPath = "会议资料/" + meetingName + "/";
String meetingMinutesFolderPath = meetingFolderPath + "会议纪要/";
String meetingDataFolderPath = meetingFolderPath + "上传文件/";
// 创建会议文件夹
zipos.putNextEntry(new ZipEntry(meetingFolderPath));
zipos.closeEntry();
// 创建会议纪要文件夹
zipos.putNextEntry(new ZipEntry(meetingMinutesFolderPath));
zipos.closeEntry();
// 创建上传文件文件夹
zipos.putNextEntry(new ZipEntry(meetingDataFolderPath));
zipos.closeEntry();
List<String> fileIds = new ArrayList<>();
if (!StringUtils.isEmpty(payMeeting.getMeetingMinute())) {
fileIds.addAll(Arrays.asList(payMeeting.getMeetingMinute().split(",")));
}
if (!StringUtils.isEmpty(payMeeting.getMeetingData())) {
fileIds.addAll(Arrays.asList(payMeeting.getMeetingData().split(",")));
}
for (String fileId : fileIds) {
OSSFile ossFile = iossFileService.getById(fileId.trim());
if (ossFile == null) {
continue;
}
File file = new File(upLoadPath + File.separator + ossFile.getUrl());
if (!file.exists()) {
continue;
}
// 根据文件类型，放入对应的文件夹
String entryPath;
if (payMeeting.getMeetingMinute() != null && Arrays.asList(payMeeting.getMeetingMinute().split(",")).contains(fileId)) {
entryPath = meetingMinutesFolderPath + ossFile.getFileName();
} else {
entryPath = meetingDataFolderPath + ossFile.getFileName();
}
// 检查是否已经添加过相同路径的条目
if (addedEntries.contains(entryPath)) {
// 如果已经存在相同路径的条目，可以修改文件名或者跳过处理
continue;
}
// 添加到压缩包中
zipos.putNextEntry(new ZipEntry(entryPath));
try (InputStream is = new FileInputStream(file)) {
byte[] b = new byte[1024];
int length;
while ((length = is.read(b)) != -1) {
zipos.write(b, 0, length);
}
} catch (IOException e) {
e.printStackTrace();
}
zipos.closeEntry(); // 关闭当前条目
addedEntries.add(entryPath); // 记录已添加的条目路径
}
}
}
zipos.flush();
} catch (IOException e) {
e.printStackTrace();
}
}
}