@Slf4j
@Controller
public class ExportPPTController {
private static final String REQ_FILES = "files";
private static final String REQ_DATA = "data";
private static final String REQ_TEMPLATE_ID = "templateId";
private static final String REQ_DOWNLOAD_FILE_NAME = "fileName";
@Autowired
private PdfHelper pdf2Helper;
@Autowired
private PPTConf pptConf;
@Autowired
private MessageSender messageSender;
@Autowired
private ViewOptionConvertReportService viewOptionConvertReportService;
@PostMapping("/ppt/handler")
public void handler(HttpServletRequest request, HttpServletResponse httpResponse) {
String downloadFileName = "" + System.currentTimeMillis();
try (XMLSlideShow ppt = genPPT(request)) {
downloadFileName = ((MultipartHttpServletRequest) request).getParameter(REQ_DOWNLOAD_FILE_NAME);
downloadFileName =
SafeString.filterFileName(StringUtil.isNotBlank(downloadFileName) ? downloadFileName : "" + System.currentTimeMillis());
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
ppt.write(outputStream);
httpResponse.addHeader(HttpHeaders.CONTENT_LENGTH, "" + outputStream.size());
httpResponse.setContentType("application/force-download");// 设置强制下载不打开
httpResponse.addHeader(HttpHeaders.CONTENT_DISPOSITION,
"attachment;filename=" + new String((downloadFileName + ".pptx").getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
ppt.write(httpResponse.getOutputStream());
} catch (Exception e) {
log.error("生成PPT[" + downloadFileName + "]时出错 " + e.getMessage(), e);
}
}
@PostMapping(value = "/pdf/handler", produces = MediaType.APPLICATION_JSON_VALUE)
public void pdfHandlerWithOffice(HttpServletRequest request, HttpServletResponse httpResponse) {
String downloadFileName = "" + System.currentTimeMillis();
try (XMLSlideShow ppt = genPPT(request)) {
downloadFileName = ((MultipartHttpServletRequest) request).getParameter(REQ_DOWNLOAD_FILE_NAME);
downloadFileName =
SafeString.filterFileName(StringUtil.isNotBlank(downloadFileName) ? downloadFileName : "" + System.currentTimeMillis());
pdf2Helper.ppt2PdfWithJacob(ppt, httpResponse.getOutputStream(), pptConf.getTemplateFolder());
// httpResponse.addHeader(HttpHeaders.CONTENT_LENGTH, "" + outputStream2.size());
httpResponse.setContentType("application/force-download");// 设置强制下载不打开
httpResponse.addHeader(HttpHeaders.CONTENT_DISPOSITION,
"attachment;filename=" + new String((downloadFileName + ".pdf").getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
} catch (Exception e) {
log.error("生成PPT[" + downloadFileName + "]时出错 " + e.getMessage(), e);
}
}
@PostMapping("/autoReport/export")
public void autoReport(HttpServletRequest request, HttpServletResponse httpResponse) {
var data = request.getParameter(REQ_DATA);
log.info("自动报告导出request===>{}", data);
ReportExportRequest reportExportRequest = JSONObject.parseObject(data, ReportExportRequest.class);
if (null == reportExportRequest) {
return;
}
StopWatch stopWatch = new StopWatch();
stopWatch.start("export ppt");
String reportNum = reportExportRequest.getReportNum();
RiflemanConfig riflemanConfig = reportExportRequest.getRiflemanConfig();
String fileType = reportExportRequest.getFileType();
String fileName = reportExportRequest.getFileName();
GlobalContext globalContext = reportExportRequest.getGlobalContext();
ExportPPTMessage exportPPTMessage = new ExportPPTMessage();
exportPPTMessage.setReportNum(reportNum);
try {
Map<String, Object> context = ObjectUtils.convert2map(globalContext, true);
fileName = SafeString.filterFileName(StringUtils.isNotEmpty(fileName) ? fileName : System.currentTimeMillis() + "");
PPTReport pptReport = viewOptionConvertReportService.convert(context, reportNum, riflemanConfig);
XMLSlideShow ppt = AutoReportGenerator.generatorPPT(pptReport, pptConf.getTemplateFolder(), pptConf.getTempFileFolder());
FileOutputStream fileO = new FileOutputStream(pptConf.getTempFileFolder() + "自动报告.pptx");
ppt.write(fileO);
fileO.close();
httpResponse.setContentType("application/force-download");// 设置强制下载不打开
if ("pptx".equals(fileType)) {//ppt
// ByteArrayOutputStream outputStream = null;
// try {
// outputStream = new ByteArrayOutputStream();
// ppt.write(outputStream);
// outputStream.flush();
// } finally {
// outputStream.close();
// ppt.close();
// }
// outputStream.writeTo(httpResponse.getOutputStream());
fileName += ".pptx";
ppt.write(httpResponse.getOutputStream());
httpResponse.getOutputStream().flush();
} else if ("pdf".equals(fileType)) {//pdf
fileName += ".pdf";
pdf2Helper.ppt2PdfWithJacob(ppt, httpResponse.getOutputStream(), pptConf.getTemplateFolder());
}
exportPPTMessage.setStatus("T");
httpResponse.addHeader(HttpHeaders.CONTENT_DISPOSITION,
"attachment;filename=" + new String((fileName).getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
stopWatch.stop();
long time = stopWatch.getTotalTimeMillis();
log.info("export ppt success! use {} ms.", time);
} catch (Exception e) {
log.error("生成[" + fileName + "]时出错" + e.getMessage(), e);
exportPPTMessage.setStatus("F");
}
log.info("ppt生成完成，发送mq消息；reportNum={}", reportNum);
messageSender.send(exportPPTMessage);
}
private XMLSlideShow genPPT(HttpServletRequest request) throws IOException {
val multiReq = (MultipartHttpServletRequest) request;
val files = multiReq.getFiles(REQ_FILES);
val data = multiReq.getParameter(REQ_DATA);
var templateId = multiReq.getParameter(REQ_TEMPLATE_ID);
log.info(data);
MultipartFile file = null;
Map<String, byte[]> fileMap = Maps.newHashMap();
for (var i = 0; i < files.size(); ++i) {
file = files.get(i);
if (!file.isEmpty()) {
fileMap.put(file.getOriginalFilename(), file.getBytes());
} else {
// does nothing
}
}
List<SlideData> slideDataList = JSON.parseArray(data, SlideData.class);
slideDataList.forEach(slide -> {
if (CollectionUtils.isNotEmpty(slide.getImageDataList())) {
slide.getImageDataList().forEach(img -> {
if (fileMap.containsKey(img.getPicId())) {
img.setPicBytes(fileMap.get(img.getPicId()));
}
});
}
});
return PowerPointGenerator.generatorPowerPoint(pptConf.getTemplateFolder() + SafeString.filterFileName(templateId) + ".pptx", slideDataList);
}
f
@Controller
@Slf4j
public class ExportWordController {
@Autowired
private PPTConf pptConf;
@Autowired
private ReportGenerateService reportGenerateService;
@PostMapping(value = "/word/export", produces = MediaType.APPLICATION_JSON_VALUE)
public void exportWord(HttpServletRequest request, HttpServletResponse httpResponse) {
var data = request.getParameter("data");
List<WordExportRequest> wordExportRequests = JSON.parseArray(data, WordExportRequest.class);
try {
reportGenerateService.htmlToWord(httpResponse.getOutputStream(), wordExportRequests);
httpResponse.setContentType("application/force-download");// 设置强制下载不打开
httpResponse.addHeader(HttpHeaders.CONTENT_DISPOSITION,
"attachment;filename=" + new String((System.currentTimeMillis() + ".doc").getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
} catch (IOException e) {
log.error("htmlToWord失败，e==>{}", e.getMessage(), e);
}
}
}
@Slf4j
@RestController
public class RiflemanController {
@Autowired
private PPTConf pptConf;
@Autowired
private ScreenService screenService;
@PostMapping("/rifleman/shotChart")
public List<ScreenResultDTO> shotChart(HttpServletRequest request, HttpServletResponse httpResponse) {
List<ScreenResultDTO> screenResultDTOS = Lists.newArrayList();
try {
String data = request.getParameter("data");
RiflemanConfig riflemanConfig = JSON.parseObject(data, RiflemanConfig.class);
List<ScreenConfig> screenConfigs = riflemanConfig.getScreenConfigs();
// ScreenUtils.screen(riflemanConfig.getJsName(), riflemanConfig.getUrl(), riflemanConfig.getPageViewSize()
// , screenConfigs, riflemanConfig.getParam());
ScreenUtils.chromeDriverScreen(riflemanConfig, pptConf.getTempFileFolder(), pptConf.getChromeDriverPath());
List<String> tempFiles = Lists.newArrayList();
for (int i = 0; i < screenConfigs.size(); i++) {
ScreenResultDTO screenResultDto = new ScreenResultDTO();
String picName = screenConfigs.get(i).getPicName();
tempFiles.add(picName);
byte[] picBytes = ScreenUtils.getScreenPicBytes(picName, pptConf.getTempFileFolder());
screenResultDto.setFileName(picName);
screenResultDto.setFileBytes(picBytes);
screenResultDTOS.add(screenResultDto);
}
FileUtils.removeTempFile(pptConf.getTempFileFolder(), tempFiles);
} catch (Exception e) {
log.error("截屏时出错 " + e.getMessage(), e);
}
return screenResultDTOS;
}
@PostMapping("/rifleman/scrollScreenShot")
public List<ScreenResultDTO> scrollScreenShot(HttpServletRequest request, HttpServletResponse httpResponse) {
List<ScreenResultDTO> screenResultDTOS = Lists.newArrayList();
try {
String data = request.getParameter("data");
RiflemanConfig riflemanConfig = JSON.parseObject(data, RiflemanConfig.class);
List<ScreenConfig> screenConfigs = riflemanConfig.getScreenConfigs();
screenService.scrollScreenShot(riflemanConfig);
List<String> tempFiles = Lists.newArrayList();
for (int i = 0; i < screenConfigs.size(); i++) {
ScreenResultDTO screenResultDto = new ScreenResultDTO();
String picName = screenConfigs.get(i).getPicName();
tempFiles.add(picName);
byte[] picBytes = ScreenUtils.getScreenPicBytes(picName, pptConf.getTempFileFolder());
screenResultDto.setFileBytes(picBytes);
screenResultDto.setFileName(picName);
screenResultDTOS.add(screenResultDto);
}
FileUtils.removeTempFile(pptConf.getTempFileFolder(), tempFiles);
} catch (Exception e) {
log.error("截屏时出错 " + e.getMessage(), e);
}
return screenResultDTOS;
}
}
}