package com.yuanchu.mom.service.impl;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.alibaba.excel.ExcelWriter;
|
import com.alibaba.excel.context.AnalysisContext;
|
import com.alibaba.excel.event.AnalysisEventListener;
|
import com.alibaba.excel.util.FileUtils;
|
import com.alibaba.excel.util.ListUtils;
|
import com.alibaba.excel.write.metadata.WriteSheet;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.yuanchu.mom.dto.ProcessMethodSearchNewDto;
|
import com.yuanchu.mom.mapper.ProcessMethodSearchNewMapper;
|
import com.yuanchu.mom.pojo.ProcessMethodSearchNew;
|
import com.yuanchu.mom.service.ProcessMethodSearchNewService;
|
import com.yuanchu.mom.utils.QueryWrappers;
|
import com.yuanchu.mom.utils.UserUtils;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.core.io.ClassPathResource;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.File;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.io.OutputStream;
|
import java.net.URLEncoder;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* 标准查新
|
*
|
* @author zhuo
|
* @since 2024-11-04
|
*/
|
@Service
|
public class ProcessMethodSearchNewServiceImpl extends ServiceImpl<ProcessMethodSearchNewMapper, ProcessMethodSearchNew> implements ProcessMethodSearchNewService {
|
|
/**
|
* 新增标准查新
|
*
|
* @param processMethodSearchNewList
|
* @return
|
*/
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public boolean addMethodSearchNew(List<ProcessMethodSearchNew> processMethodSearchNewList) {
|
for (ProcessMethodSearchNew processMethodSearchNew : processMethodSearchNewList) {
|
processMethodSearchNew.setIsNewStandard(1);
|
}
|
// 修改之前所有没有变新的
|
List<ProcessMethodSearchNew> processMethodSearchNews = baseMapper.selectList(Wrappers.<ProcessMethodSearchNew>lambdaQuery()
|
.eq(ProcessMethodSearchNew::getIsNewStandard, 1));
|
for (ProcessMethodSearchNew processMethodSearchNew : processMethodSearchNews) {
|
baseMapper.update(null, Wrappers.<ProcessMethodSearchNew>lambdaUpdate()
|
.set(ProcessMethodSearchNew::getMethodName, processMethodSearchNew.getNewMethodName())
|
.set(ProcessMethodSearchNew::getStandardNo, processMethodSearchNew.getNewStandardNo())
|
.set(ProcessMethodSearchNew::getFileNo, processMethodSearchNew.getNewStandardNo())
|
.set(ProcessMethodSearchNew::getIsNewStandard, 0)
|
.set(ProcessMethodSearchNew::getNewMethodName, null)
|
.set(ProcessMethodSearchNew::getNewStandardNo, null)
|
.eq(ProcessMethodSearchNew::getMethodSearchNewId, processMethodSearchNew.getMethodSearchNewId()));
|
}
|
// 新增
|
this.saveBatch(processMethodSearchNewList);
|
return true;
|
}
|
|
/**
|
* 标准查新列表
|
* @param processMethodSearchNew
|
* @return
|
*/
|
@Override
|
public IPage<ProcessMethodSearchNew> pageMethodSearchNew(Page page, ProcessMethodSearchNewDto processMethodSearchNew) {
|
String beginDate = processMethodSearchNew.getBeginDate();
|
String endDate = processMethodSearchNew.getEndDate();
|
processMethodSearchNew.setBeginDate(null);
|
processMethodSearchNew.setEndDate(null);
|
return baseMapper.pageMethodSearchNew(page, QueryWrappers.queryWrappers(processMethodSearchNew), beginDate, endDate);
|
}
|
|
/**
|
* 标准查新导出
|
* @param processMethodSearchNew
|
* @param response
|
*/
|
@Override
|
public void exportMethodSearchNew(ProcessMethodSearchNewDto processMethodSearchNew, HttpServletResponse response) {
|
// 查询标准查新
|
String beginDate = processMethodSearchNew.getBeginDate();
|
String endDate = processMethodSearchNew.getEndDate();
|
String writeUserId = processMethodSearchNew.getWriteUserId().toString();
|
String ratifyUserId = processMethodSearchNew.getRatifyUserId().toString();
|
processMethodSearchNew.setBeginDate(null);
|
processMethodSearchNew.setEndDate(null);
|
processMethodSearchNew.setWriteUserId(null);
|
processMethodSearchNew.setRatifyUserId(null);
|
List<ProcessMethodSearchNewDto> methodSearchNews = baseMapper.selectMethodSearchNew(QueryWrappers.queryWrappers(processMethodSearchNew), beginDate, endDate);
|
|
int index = 1;
|
// 格式化参数
|
for (ProcessMethodSearchNewDto methodSearchNew : methodSearchNews) {
|
// 是否是跟新标准
|
if (methodSearchNew.getIsNewStandard().equals(1)) {
|
methodSearchNew.setIsNewStandardString("是");
|
} else {
|
methodSearchNew.setIsNewStandardString("否");
|
}
|
|
// 备注
|
if (methodSearchNew.getRemark() != null) {
|
if (methodSearchNew.getRemark().equals(1)) {
|
methodSearchNew.setIsNewStandardString("替换");
|
} else {
|
methodSearchNew.setIsNewStandardString("作废");
|
}
|
}
|
|
if (methodSearchNew.getRemark() != null) {
|
if (methodSearchNew.getRemark().equals(1)) {
|
methodSearchNew.setRemarkString("替换");
|
} else {
|
methodSearchNew.setRemarkString("作废");
|
}
|
}
|
|
switch (methodSearchNew.getSearchNewSource()) {
|
case 0:
|
methodSearchNew.setStandardNet("√");
|
break;
|
case 1:
|
methodSearchNew.setInformationOffices("√");
|
break;
|
case 2:
|
methodSearchNew.setStandardBookstore("√");
|
break;
|
case 3:
|
methodSearchNew.setOther("√");
|
break;
|
}
|
methodSearchNew.setIndex(index);
|
index++;
|
|
}
|
// 查询签名地址
|
String writeUserUrl = UserUtils.getUserSignatureUrl(Integer.valueOf(writeUserId));
|
String ratifyUserUrl = UserUtils.getUserSignatureUrl(Integer.valueOf(ratifyUserId));
|
|
//创建ExcelWriter 可以自动关流但还是手动关一次
|
ExcelWriter excelWriter = null;
|
|
try {
|
// outputStream：要导出的文件的输出流
|
OutputStream outputStream = response.getOutputStream();
|
// 获取模版文件
|
ClassPathResource classPathResource = new ClassPathResource("/static/excel/check-records.xlsx");
|
// 使用模版文件的两种方式：
|
// 1、文件路径：.withTemplate(templateFileName)
|
// 2、输入流：.withTemplate(inputStream)
|
// String templateFileName = classPathResource.getFile().getPath();
|
InputStream inputStream = classPathResource.getInputStream();
|
// 创建ExcelWriter
|
excelWriter = EasyExcel.write(outputStream).withTemplate(inputStream).build();
|
// 获取第一个sheet页
|
WriteSheet writeSheet = EasyExcel.writerSheet(0, "标准查新导出").build();
|
//excelWriter.fill() 这地方就是填充属性。
|
// excelWriter.fill(methodSearchNews, fillConfig, writeSheet);
|
excelWriter.fill(methodSearchNews, writeSheet);
|
|
// 这里easy excel模板导出问题 会删除后面内容再进行进行填充
|
// 所以list 后面还有数据 想办法手动写入
|
// 手动创建一个list 存放数据 然后再进行填充
|
// 这里List<Object>可以用对象代替 偷懒用list
|
List<List<Object>> totalListList = ListUtils.newArrayList();
|
List<Object> totalList = ListUtils.newArrayList();
|
totalListList.add(totalList);
|
// 第一列
|
totalList.add("查新人：");
|
// 第二列 设置签名
|
totalList.add(FileUtils.readFileToByteArray(new File(writeUserUrl)));
|
// 第三列
|
totalList.add("日期：");
|
// 第五列
|
totalList.add("2024-12-23"); // 日期写死
|
// 第六列
|
totalList.add("审核：");
|
// 第七列
|
totalList.add(FileUtils.readFileToByteArray(new File(ratifyUserUrl)));
|
// 第八列
|
totalList.add("日期：");
|
// 第九列
|
totalList.add("2024-12-23"); // 日期写死
|
// 这里是write 别和fill 搞错了
|
excelWriter.write(totalListList, writeSheet);
|
|
// 设置输出流格式以及文件名：
|
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
|
response.setCharacterEncoding("utf-8");
|
String fileName = URLEncoder.encode(
|
"原材料检测信息导出", "UTF-8");
|
response.setHeader("Content-disposition",
|
"attachment;filename=" + fileName + ".xlsx");
|
} catch (Exception e) {
|
throw new RuntimeException(e);
|
} finally {
|
// 千万别忘记close关闭流
|
if (excelWriter != null) {
|
excelWriter.close();
|
|
}
|
}
|
}
|
|
|
|
|
|
|
|
/**
|
* 导入标准查新
|
* @param file
|
* @return
|
*/
|
@Override
|
public boolean importMethodSearchNew(MultipartFile file) {
|
|
List<ProcessMethodSearchNewDto> searchNewDtoList = new ArrayList<>();
|
|
try {
|
// excel解析
|
EasyExcel.read(file.getInputStream(), ProcessMethodSearchNewDto.class, new AnalysisEventListener<ProcessMethodSearchNewDto>() {
|
@Override
|
public void invoke(ProcessMethodSearchNewDto searchNewDto, AnalysisContext analysisContext) {
|
searchNewDtoList.add(searchNewDto);
|
}
|
|
@Override
|
public void doAfterAllAnalysed(AnalysisContext analysisContext) {
|
|
}
|
}).sheet().doRead();
|
// 格式化查新对象
|
List<ProcessMethodSearchNew> collect = searchNewDtoList.stream().map(searchNewDto -> {
|
ProcessMethodSearchNew searchNew = new ProcessMethodSearchNew();
|
BeanUtils.copyProperties(searchNewDto, searchNew);
|
// 是否是更新
|
if (StringUtils.isNotBlank(searchNewDto.getIsNewStandardString())) {
|
if (searchNewDto.getIsNewStandardString().equals("是")) {
|
searchNew.setIsNewStandard(1);
|
} else {
|
searchNew.setIsNewStandard(0);
|
}
|
}
|
// 查新记录
|
if (StringUtils.isNotBlank(searchNewDto.getStandardNet())) {
|
searchNew.setSearchNewSource(0);
|
} else if (StringUtils.isNotBlank(searchNewDto.getInformationOffices())) {
|
searchNew.setSearchNewSource(1);
|
} else if (StringUtils.isNotBlank(searchNewDto.getIsNewStandardString())) {
|
searchNew.setSearchNewSource(2);
|
} else if (StringUtils.isNotBlank(searchNewDto.getOther())) {
|
searchNew.setSearchNewSource(3);
|
}
|
|
// 备注
|
if (StringUtils.isNotBlank(searchNewDto.getRemarkString())) {
|
if (searchNewDto.getRemarkString().equals("替换")) {
|
searchNew.setRemark(1);
|
} else if (searchNewDto.getRemarkString().equals("作废")) {
|
searchNew.setRemark(0);
|
}
|
}
|
|
return searchNew;
|
}).collect(Collectors.toList());
|
this.saveBatch(collect);
|
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
return true;
|
}
|
}
|
