chenxudong
/
electromagnetic-data-new
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化代码

This commit is contained in:
chenxudong 2025-04-09 16:57:42 +08:00
parent d1dead8802
commit c525e30372
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
electrmangnetic/src/main/java/com/electromagnetic/industry/software/manage/service/serviceimpl/ChatService.java
View File

@ -136,7 +136,6 @@ public class ChatService {
String srcPath = elePropertyConfig.getEleTmpPath() + File.separator + fileMd5 + "." + fileType; String srcPath = elePropertyConfig.getEleTmpPath() + File.separator + fileMd5 + "." + fileType;
if (StrUtil.equals(fileType, "pdf")) { if (StrUtil.equals(fileType, "pdf")) {
FileUtil.writeFromStream(file.getInputStream(), pdfPath); FileUtil.writeFromStream(file.getInputStream(), pdfPath);
pageFiles = OfficeFileUtil.parsePdfByPage(pdfPath);
} else { } else {
FileUtil.writeFromStream(file.getInputStream(), srcPath); FileUtil.writeFromStream(file.getInputStream(), srcPath);
if (StrUtil.equals(fileType, "doc") || StrUtil.equals(fileType, "docx")) { if (StrUtil.equals(fileType, "doc") || StrUtil.equals(fileType, "docx")) {
@ -144,8 +143,8 @@ public class ChatService {
} else { } else {
OfficeFileUtil.ppt2pdf(srcPath, pdfPath); OfficeFileUtil.ppt2pdf(srcPath, pdfPath);
} }
pageFiles = OfficeFileUtil.parsePdfByPage(pdfPath);
} }
pageFiles = OfficeFileUtil.parsePdfByPage(pdfPath);
StringBuilder stringBuilder = new StringBuilder(); StringBuilder stringBuilder = new StringBuilder();
pageFiles.forEach(page -> stringBuilder.append(page.getContent())); pageFiles.forEach(page -> stringBuilder.append(page.getContent()));
Document document = new Document(stringBuilder.toString(), Map.of("file_name", file.getOriginalFilename())); Document document = new Document(stringBuilder.toString(), Map.of("file_name", file.getOriginalFilename()));

1
electromagnetic-common/src/main/java/com/electromagnetic/industry/software/common/util/OfficeFileUtil.java
View File

@ -292,6 +292,7 @@ public class OfficeFileUtil {
PDFRenderer renderer = new PDFRenderer(document); PDFRenderer renderer = new PDFRenderer(document);
// 遍历每一页 // 遍历每一页
for (int page = 0; page < document.getNumberOfPages(); page++) { for (int page = 0; page < document.getNumberOfPages(); page++) {
log.info("当前解析 {} / {} 页", page + 1, document.getNumberOfPages());
// 渲染页面为图像（设置缩放比例，300 dpi） // 渲染页面为图像（设置缩放比例，300 dpi）
BufferedImage image = renderer.renderImageWithDPI(page, 300); BufferedImage image = renderer.renderImageWithDPI(page, 300);
// 生成输出路径 // 生成输出路径