package com.tievd.jyz.service.impl;
|
|
import cn.hutool.core.date.DateUtil;
|
import cn.hutool.core.io.FileUtil;
|
import cn.hutool.core.util.ZipUtil;
|
import cn.hutool.http.HttpUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.tievd.cube.commons.easyexcel.EasyExcel;
|
import com.tievd.cube.commons.utils.SystemContextUtil;
|
import com.tievd.cube.commons.utils.web.HttpServletUtil;
|
import com.tievd.jyz.cache.AlgorithmCache;
|
import com.tievd.jyz.constants.SystemConstant;
|
import com.tievd.jyz.entity.OiloutEvent;
|
import com.tievd.jyz.entity.vo.OiloutRecordVo;
|
import com.tievd.jyz.mapper.OiloutEventMapper;
|
import com.tievd.jyz.service.IOiloutEventService;
|
import fr.opensagres.xdocreport.document.IXDocReport;
|
import fr.opensagres.xdocreport.document.registry.XDocReportRegistry;
|
import fr.opensagres.xdocreport.template.IContext;
|
import fr.opensagres.xdocreport.template.TemplateEngineKind;
|
import fr.opensagres.xdocreport.template.formatter.FieldsMetadata;
|
import lombok.SneakyThrows;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.scheduling.annotation.Async;
|
import org.springframework.stereotype.Service;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.beans.PropertyDescriptor;
|
import java.io.*;
|
import java.lang.reflect.InvocationTargetException;
|
import java.util.ArrayList;
|
import java.util.Comparator;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
/**
|
* OiloutEvent
|
* @author cube
|
* @since 2023-02-27
|
* @version V2.0.0
|
*/
|
@Service
|
@Slf4j
|
public class OiloutEventServiceImpl extends ServiceImpl<OiloutEventMapper, OiloutEvent> implements IOiloutEventService {
|
|
@Autowired
|
OiloutEventMapper oiloutEventMapper;
|
|
@Autowired
|
EasyExcel easyExcel;
|
|
@Value("${cube.resources.download-path:/app/jyz/download/}")
|
String filePath;
|
|
@Value("${cube.resources.templete-path:/app/jyz/templete/}")
|
String templetePath;
|
|
public static ConcurrentHashMap<String, String> fileMap = new ConcurrentHashMap();
|
|
@Override
|
public List<Map> oilOutStatis(Map map) {
|
return oiloutEventMapper.oilOutStatis(map);
|
}
|
|
@Override
|
@SneakyThrows
|
public void exportWord(OiloutRecordVo recordVo, HttpServletResponse response) {
|
//获取卸油流程数据
|
LambdaQueryWrapper<OiloutEvent> wrapper = new LambdaQueryWrapper();
|
wrapper.eq(OiloutEvent::getRecordId, recordVo.getId());
|
List<OiloutEvent> recordEvents = this.list(wrapper);
|
try {
|
recordEvents.sort(Comparator.comparingInt(o -> AlgorithmCache.algorithmMap().get(o.getAlgorithmCode()).getSort()));
|
} catch (Exception e) {
|
log.error("", e);
|
}
|
HttpServletUtil.addDownloadHeader(response, "卸油记录.docx");
|
String tempPath = templetePath + "recordTemp" + recordEvents.size() + SystemConstant.WORD_SUFFIX;
|
try(InputStream in = new FileInputStream(new File(tempPath))) {
|
IXDocReport docReport = XDocReportRegistry.getRegistry().loadReport(in, TemplateEngineKind.Freemarker);
|
IContext context = docReport.createContext();
|
context.put("recordTime", DateUtil.formatDateTime(recordVo.getStartTime()));
|
context.put("spandTime", recordVo.getSpandTime());
|
context.put("complete", recordVo.getComplete());
|
//创建字段元数据，需要表格才加下面这两行，否则不用
|
FieldsMetadata fm = docReport.createFieldsMetadata();
|
//待删除文件
|
List<String> fileList = new ArrayList<>();
|
//填充word数据
|
for (int i = 0; i < recordEvents.size(); i++) {
|
OiloutEvent event = recordEvents.get(i);
|
int num = i + 1;
|
String fileStr = filePath + event.getId() + SystemConstant.IMG_SUFFIX;
|
fileList.add(fileStr);
|
context.put("algorithmName" + num, event.getAlgorithmName());
|
context.put("eventTime" + num, DateUtil.formatDateTime(event.getEventTime()));
|
context.put("eventType" + num, event.getEventType() == 1 ? "规范" : "异常");
|
//增加图片
|
try {
|
File img = HttpUtil.downloadFileFromUrl(event.getImgPath(), fileStr);
|
context.put("img" + num, new FileInputStream(img));
|
fm.addFieldAsImage("img" + num);
|
} catch (Exception e) {
|
log.info("导出word图片获取失败--{}", event.getImgPath());
|
}
|
}
|
|
//// //Word模板中的表格数据对应的集合类型
|
// fm.load("event", OiloutEvent.class, true);
|
//处理word文档并输出
|
docReport.process(context, response.getOutputStream());
|
} catch (Exception e) {
|
log.error("", e);
|
throw new RuntimeException("导出word失败" + e.getMessage());
|
}
|
}
|
|
/**
|
* 鉴于第一种不太灵活，增加第二种导出方法
|
* 未完待续
|
*/
|
@SneakyThrows
|
public void exportWord2(OiloutRecordVo recordVo, HttpServletResponse response) {
|
//获取卸油流程数据
|
LambdaQueryWrapper<OiloutEvent> wrapper = new LambdaQueryWrapper();
|
wrapper.eq(OiloutEvent::getRecordId, recordVo.getId());
|
List<OiloutEvent> recordEvents = this.list(wrapper);
|
HttpServletUtil.addDownloadHeader(response, "卸油记录.docx");
|
try (InputStream in = new FileInputStream(new File("d:/recordTemp.docx"))) {
|
IXDocReport docReport = XDocReportRegistry.getRegistry().loadReport(in, TemplateEngineKind.Freemarker);
|
IContext context = docReport.createContext();
|
context.put("recordTime", recordVo.getStartTime());
|
context.put("spandTime", recordVo.getSpandTime());
|
context.put("complete", recordVo.getComplete());
|
context.put("event",recordEvents);
|
//创建字段元数据，需要表格才加下面这两行，否则不用
|
FieldsMetadata fm = docReport.createFieldsMetadata();
|
//Word模板中的表格数据对应的集合类型
|
fm.load("event", OiloutEvent.class, true);
|
//处理word文档并输出
|
docReport.process(context, response.getOutputStream());
|
|
} catch (Exception e) {
|
log.error("导出word异常", e);
|
}
|
}
|
|
|
/**
|
*
|
* @param dataList 数据
|
* @param fileName filemap存储的key，用以异步下载
|
*/
|
@Override
|
@SneakyThrows
|
@Async
|
public void exportZip(List dataList, String fileName) {
|
String dirPath = filePath + fileName + "/";
|
FileUtil.mkdir(dirPath);
|
String excel = dirPath + fileName + SystemConstant.EXCEL_SUFFIX;
|
String mediaPath = "media/";
|
FileUtil.mkdir(mediaPath);
|
//获取文件
|
for (Object o : dataList) {
|
try {
|
Class clazz = o.getClass();
|
PropertyDescriptor uidField = BeanUtils.getPropertyDescriptor(clazz, "imgUid");
|
String imgUid = (String) uidField.getReadMethod().invoke(o);
|
|
PropertyDescriptor imgField = BeanUtils.getPropertyDescriptor(clazz, "imgPath");
|
String relativePath = mediaPath + imgUid + SystemConstant.IMG_SUFFIX;
|
reSetField(o, imgField, relativePath, dirPath);
|
|
PropertyDescriptor vdoField = BeanUtils.getPropertyDescriptor(clazz, "videoPath");
|
relativePath = mediaPath + imgUid + SystemConstant.VIDEO_SUFFIX;
|
reSetField(o, vdoField, relativePath, dirPath);
|
} catch (Exception e) {
|
log.error("文件或图片下载失败", e);
|
}
|
|
}
|
//excel
|
try(OutputStream out = new FileOutputStream(new File(excel))) {
|
easyExcel.export(dataList, out, SystemContextUtil.dictTranslator());
|
} catch (IOException e) {
|
log.error("导出失败", e);
|
throw new RuntimeException("excel生成失败" + e.getMessage());
|
}
|
|
File zip = ZipUtil.zip(dirPath);
|
fileMap.put(fileName, zip.getAbsolutePath());
|
FileUtil.del(dirPath);
|
}
|
|
//重新设置文件字段值
|
public void reSetField(Object obj, PropertyDescriptor field, String relativePath, String dirPath) throws InvocationTargetException, IllegalAccessException {
|
String url = (String) field.getReadMethod().invoke(obj);
|
if (StringUtils.isBlank(url)) {
|
return;
|
}
|
File file = new File(dirPath + relativePath);
|
HttpUtil.downloadFile(url, file);
|
field.getWriteMethod().invoke(obj, relativePath);
|
}
|
|
public Map<String, String> getFileMap() {
|
return fileMap;
|
}
|
|
|
|
}
|
