From ec9a95eb994018cc1ba28a80946165f795001dea Mon Sep 17 00:00:00 2001 From: xiangpei <xiangpei@timesnew.cn> Date: 星期五, 29 十一月 2024 16:49:24 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java | 37 +++++++++++++++++++++++++++++++++++++ 1 files changed, 37 insertions(+), 0 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 060e8cf..b865b20 100644 --- a/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java +++ b/business/src/main/java/com/ycl/service/impl/ProjectInfoServiceImpl.java @@ -1,5 +1,7 @@ package com.ycl.service.impl; +import cn.hutool.core.collection.CollUtil; +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; @@ -7,13 +9,18 @@ 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; @@ -25,8 +32,11 @@ import org.springframework.transaction.annotation.Transactional; import org.springframework.util.Assert; +import javax.servlet.ServletOutputStream; +import javax.servlet.http.HttpServletResponse; import java.math.BigDecimal; import java.math.RoundingMode; +import java.text.SimpleDateFormat; import java.util.*; import java.util.stream.Collectors; @@ -156,6 +166,7 @@ query.setProjectEndTime(DateUtils.getDayEnd(query.getProjectEndTime())); } String projectCategory = query.getProjectCategory(); + //瀹炴柦闃舵鏈変袱涓�兼斁鍦╯ql澶勭悊 if (ProjectCategoryEnum.RESERVE.getType().equals(projectCategory)) { query.setProjectStatus(ProjectCategoryEnum.RESERVE.getStatus()); query.setReserveOrPrevious(ProjectCategoryEnum.RESERVE.getCode()); @@ -375,4 +386,30 @@ return Result.ok().data(false); } } + + @Override + public void export(HttpServletResponse response, ProjectExportQuery query) { + 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()); + } + } + //TODO锛氶渶瑕侀檮浠� + } } -- Gitblit v1.8.0