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.PlanForm;
|
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;
|
import com.ycl.mapper.*;
|
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;
|
|
/**
|
* 项目管理基础信息表 服务实现类
|
*
|
* @author flq
|
* @since 2024-11-22
|
*/
|
@Service
|
@RequiredArgsConstructor
|
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;
|
|
/**
|
* 添加
|
*
|
* @param form
|
* @return
|
*/
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public Result add(ProjectInfoForm form) {
|
//添加基本信息
|
ProjectInfo entity = ProjectInfoForm.getEntityByForm(form, null);
|
Long userId = SecurityUtils.getUserId();
|
entity.setCreateBy(userId);
|
entity.setUpdateBy(userId);
|
baseMapper.insert(entity);
|
addPlan(entity.getId()); //添加计划表
|
//添加文件
|
List<File> fileList = form.getFileList();
|
fileList.forEach(item -> {
|
item.setBusId(entity.getId());
|
item.setType(FileTypeEnum.PROJECT_INFO);
|
});
|
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);
|
}
|
|
/**
|
* 修改
|
*
|
* @param form
|
* @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);
|
//更新项目信息
|
baseMapper.updateById(entity);
|
List<File> fileList = form.getFileList();
|
fileList.forEach(item -> {
|
item.setId(null);
|
item.setBusId(entity.getId());
|
item.setType(FileTypeEnum.PROJECT_INFO);
|
});
|
//删除原有文件
|
QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>();
|
fileQueryWrapper.eq("type", FileTypeEnum.PROJECT_INFO.getType());
|
fileQueryWrapper.eq("bus_id", entity.getId());
|
fileMapper.delete(fileQueryWrapper);
|
//替换成现有
|
fileService.saveBatch(fileList);
|
return Result.ok("修改成功");
|
}
|
|
/**
|
* 批量删除
|
*
|
* @param ids
|
* @return
|
*/
|
@Override
|
public Result remove(List<String> ids) {
|
baseMapper.deleteBatchIds(ids);
|
return Result.ok("删除成功");
|
}
|
|
/**
|
* id删除
|
*
|
* @param id
|
* @return
|
*/
|
@Override
|
@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("删除成功");
|
}
|
|
/**
|
* 分页查询
|
*
|
* @param query
|
* @return
|
*/
|
@Override
|
public Result page(ProjectInfoQuery query) {
|
if (query.getProjectStartTime() != null) {
|
query.setProjectStartTime(DateUtils.getDayStart(query.getProjectStartTime()));
|
}
|
if (query.getProjectEndTime() != null) {
|
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());
|
} else if (ProjectCategoryEnum.PREVIOUS.getType().equals(projectCategory)) {
|
query.setProjectStatus(ProjectCategoryEnum.PREVIOUS.getStatus());
|
query.setReserveOrPrevious(ProjectCategoryEnum.PREVIOUS.getCode());
|
} else if (ProjectCategoryEnum.FINISH.getType().equals(projectCategory)) {
|
query.setProjectStatus(ProjectCategoryEnum.FINISH.getStatus());
|
} else if (ProjectCategoryEnum.EXCEPTION.getType().equals(projectCategory)) {
|
//TODO
|
//先查出异常流程或者异常进度或者异常计划的projectId和异常种类
|
//通过projectId查出项目数据
|
//补充相应的异常数据（异常种类、异常流程节点等）
|
List<ProjectVO> list = new ArrayList<>();
|
return Result.ok().data(list).total(0);
|
}
|
|
IPage<ProjectInfoVO> page = PageUtil.getPage(query, ProjectInfoVO.class);
|
baseMapper.getPage(page, query);
|
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);
|
//翻译项目阶段
|
String phase = ProjectCategoryEnum.getPhaseByProjectStatus(projectVO.getProjectStatus(), projectVO.getProcessId() != null);
|
projectVO.setProjectPhase(phase);
|
list.add(projectVO);
|
});
|
return Result.ok().data(list).total(page.getTotal());
|
}
|
|
private void copyToProjectVO(ProjectInfoVO vo, ProjectVO projectVO) {
|
//忽略null值的复制
|
CopyUtils.copyNoNullProperties(vo, 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);
|
}
|
|
/**
|
* 根据id查找
|
*
|
* @param id
|
* @return
|
*/
|
@Override
|
public Result detail(Integer id) {
|
ProjectInfo entity = baseMapper.getById(id);
|
Assert.notNull(entity, "记录不存在");
|
ProjectInfoVO vo = ProjectInfoVO.getVoByEntity(entity, null);
|
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);
|
return Result.ok().data(vo);
|
}
|
|
/**
|
* 列表
|
*
|
* @return
|
*/
|
@Override
|
public Result all() {
|
List<ProjectInfo> entities = baseMapper.selectList(null);
|
List<ProjectInfoVO> vos = entities.stream()
|
.map(entity -> ProjectInfoVO.getVoByEntity(entity, null))
|
.collect(Collectors.toList());
|
return Result.ok().data(vos);
|
}
|
|
@Override
|
public IndexCountVO getIndexCount(IndexDTO indexDTO) {
|
// {"proPhaseCountVO":[{"type":"储备规划阶段","count":0,"amount":"0.00","text":"储"},
|
// {"type":"项目前期阶段","count":0,"amount":"0.00","text":"新"},
|
// {"type":"实施阶段","count":0,"amount":"0.00","text":"建"},{"type":"竣工投用阶段","count":0,"amount":"0.00","text":"竣"}],
|
// "impTypeCountVO":[{"type":"一般项目","count":0,"amount":"0.00","text":"普"},
|
// {"type":"县重点项目","count":0,"amount":"0.00","text":"县"},{"type":"市重点项目","count":0,"amount":"0.00","text":"市"},
|
// {"type":"省重点项目","count":0,"amount":"0.00","text":"省"}]}}
|
IndexCountVO indexCountVO = new IndexCountVO();
|
List<IndexProPhaseCountVO> proPhaseCountVO = new ArrayList<>();
|
List<ProjectVO> projectVOS = baseMapper.homeCount();
|
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;
|
}
|
|
@Override
|
public Map<String, Integer> countExceptionProject(IndexDTO indexDTO) {
|
Map<String, Integer> map = new HashMap<>();
|
map.put("processExceptionProject", 0);
|
return map;
|
}
|
|
@Override
|
public Result docDetail(Integer id) {
|
DocumentInfoForm documentInfoForm = new DocumentInfoForm();
|
QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>();
|
fileQueryWrapper.eq("type", FileTypeEnum.DOCUMENT_INFO.getType());
|
fileQueryWrapper.eq("bus_id", id);
|
List<File> files = fileMapper.selectList(fileQueryWrapper);
|
documentInfoForm.setFileList(files);
|
return Result.ok().data(documentInfoForm);
|
}
|
|
@Override
|
public Result addDoc(DocumentInfoForm form) {
|
List<File> fileList = form.getFileList();
|
fileList.forEach(item -> {
|
item.setId(null);
|
item.setBusId(form.getProjectId());
|
item.setType(FileTypeEnum.DOCUMENT_INFO);
|
});
|
//删除原有文件
|
QueryWrapper<File> fileQueryWrapper = new QueryWrapper<>();
|
fileQueryWrapper.eq("type", FileTypeEnum.DOCUMENT_INFO.getType());
|
fileQueryWrapper.eq("bus_id", form.getProjectId());
|
fileMapper.delete(fileQueryWrapper);
|
//替换成现有
|
fileService.saveBatch(fileList);
|
return Result.ok();
|
}
|
|
@Override
|
public Result getManagerFlag(Integer recordId) {
|
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();
|
if (list.contains(userId.toString())) {
|
return Result.ok().data(true);
|
} else {
|
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);
|
}
|
}
|
}
|
}
|
