diff --git a/wmyun-module-system/wmyun-module-system-biz/pom.xml b/wmyun-module-system/wmyun-module-system-biz/pom.xml index 018b1a0..8510843 100644 --- a/wmyun-module-system/wmyun-module-system-biz/pom.xml +++ b/wmyun-module-system/wmyun-module-system-biz/pom.xml @@ -162,6 +162,17 @@ 3.25.2 compile + + com.wmyun + wmyun-module-infra-api + 2.3.0-SNAPSHOT + compile + + + org.springframework + spring-test + 6.2.2 + diff --git a/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/controller/admin/pdf/PDFController.java b/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/controller/admin/pdf/PDFController.java index 6ef9719..ab2e876 100644 --- a/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/controller/admin/pdf/PDFController.java +++ b/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/controller/admin/pdf/PDFController.java @@ -2,9 +2,10 @@ package com.wmyun.module.system.controller.admin.pdf; import com.wmyun.module.system.controller.admin.pdf.vo.PdfResultVO; import com.wmyun.module.system.service.pdf.PdfSplitService; +import io.swagger.v3.oas.annotations.Operation; import jakarta.annotation.Resource; -import lombok.Value; import org.springframework.http.ResponseEntity; +import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.transaction.annotation.Transactional; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.*; @@ -31,44 +32,64 @@ public class PDFController { @Resource private PdfSplitService pdfSplitService; - private String uploadDir; // 配置文件设置存储路径 @PostMapping("/split_result") + @Operation(summary = "PDF拆分") + @PreAuthorize("@ss.hasPermission('system:pdf:split_result')") public ResponseEntity splitPdfresult(@RequestParam("file") MultipartFile file) { try { + if (!file.getContentType().equals("application/pdf")) { + throw new IllegalArgumentException("仅支持 PDF 文件"); + } // 保存临时文件 - File tempFile = File.createTempFile("split-", ".pdf"); + File tempFile = File.createTempFile("split-", ".pdf"); file.transferTo(tempFile); - // 执行拆分 - List resultList = pdfSplitService.splitPdf( - tempFile, - uploadDir - ); - + List resultList = pdfSplitService.splitPdf(tempFile, uploadDir); + tempFile.delete(); // 返回结果 return ResponseEntity.ok(resultList); } catch (IOException e) { return ResponseEntity.status(500) - .body(Map.of("error", "处理失败: " + e.getMessage())); + .body(Map.of("error", "处理失败: " + e.getMessage())); } } - @PostMapping("/split") - public ResponseEntity splitPdf( - @RequestParam("file") MultipartFile file, - @RequestParam(defaultValue = "1") int startPage, - @RequestParam(defaultValue = "1") int endPage) { + @PostMapping("/split_image") + @Operation(summary = "PDF拆分图片") + @PreAuthorize("@ss.hasPermission('system:pdf:split_image')") + public ResponseEntity PdfToImageConverter(@RequestParam("file") MultipartFile file) { try { - File tempFile = File.createTempFile("temp", ".pdf"); + if (!file.getContentType().equals("application/pdf")) { + throw new IllegalArgumentException("仅支持 PDF 文件"); + } + // 保存临时文件 + File tempFile = File.createTempFile("split-", ".pdf"); file.transferTo(tempFile); - String outputPath = "/output/"; - pdfSplitService.splitPdfByPages(tempFile, outputPath, startPage, endPage); - return ResponseEntity.ok(" 拆分成功"); + // 执行拆分 + List resultList = pdfSplitService.PdfToImageConverter(tempFile, uploadDir); + tempFile.delete(); + // 返回结果 + return ResponseEntity.ok(resultList); } catch (IOException e) { - return ResponseEntity.status(500).body(" 处理失败: " + e.getMessage()); + return ResponseEntity.status(500).body(Map.of("error", "处理失败: " + e.getMessage())); } } +// @PostMapping("/split") +// public ResponseEntity splitPdf( +// @RequestParam("file") MultipartFile file, +// @RequestParam(defaultValue = "1") int startPage, +// @RequestParam(defaultValue = "1") int endPage) { +// try { +// File tempFile = File.createTempFile("temp", ".pdf"); +// file.transferTo(tempFile); +// String outputPath = "/output/"; +// pdfSplitService.splitPdfByPages(tempFile, outputPath, startPage, endPage); +// return ResponseEntity.ok(" 拆分成功"); +// } catch (IOException e) { +// return ResponseEntity.status(500).body(" 处理失败: " + e.getMessage()); +// } +// } } diff --git a/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/controller/admin/pdf/vo/PdfResultVO.java b/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/controller/admin/pdf/vo/PdfResultVO.java index 94944b8..f087d30 100644 --- a/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/controller/admin/pdf/vo/PdfResultVO.java +++ b/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/controller/admin/pdf/vo/PdfResultVO.java @@ -1,5 +1,6 @@ package com.wmyun.module.system.controller.admin.pdf.vo; +import io.swagger.v3.oas.annotations.media.Schema; import lombok.AllArgsConstructor; import lombok.Data; @@ -12,7 +13,10 @@ import lombok.Data; @Data @AllArgsConstructor public class PdfResultVO { + @Schema(description = "拆分后文件名称", example = "page_1.pdf") private String fileName; // 拆分后的文件名 + @Schema(description = "服务武器存储路径") private String filePath; // 服务器存储路径 + @Schema(description = "文件大小（字节））", example = "315063") private Long fileSize; // 文件大小（字节） } diff --git a/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/service/pdf/PdfSplitService.java b/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/service/pdf/PdfSplitService.java index fae64fe..8ee7440 100644 --- a/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/service/pdf/PdfSplitService.java +++ b/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/service/pdf/PdfSplitService.java @@ -1,17 +1,33 @@ package com.wmyun.module.system.service.pdf; import com.wmyun.module.system.controller.admin.pdf.vo.PdfResultVO; -import org.apache.pdfbox.pdmodel.PDDocument; +import org.springframework.web.multipart.MultipartFile; import java.io.File; import java.io.IOException; -import java.util.ArrayList; import java.util.List; public interface PdfSplitService { + /** + * 拆分一个PDF文件为多个pdf（每页一个） + * @param sourceFile + * @param outputDir + * @return + * @throws IOException + */ List splitPdf(File sourceFile, String outputDir) throws IOException; + /** + * 拆分一个PDF文件为多个图片（每页一个） + * @param pdfFile + * @param outputDir + * @return + * @throws IOException + */ + List PdfToImageConverter(File pdfFile, String outputDir) throws IOException; void splitPdfByPages(File sourceFile, String outputDir, int startPage, int endPage) throws IOException; void splitAllPages(File sourceFile, String outputDir) throws IOException; + + MultipartFile convertToMultipartFile(String filePath) throws IOException; } diff --git a/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/service/pdf/PdfSplitServiceImpl.java b/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/service/pdf/PdfSplitServiceImpl.java index c1b9110..c2c05f4 100644 --- a/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/service/pdf/PdfSplitServiceImpl.java +++ b/wmyun-module-system/wmyun-module-system-biz/src/main/java/com/wmyun/module/system/service/pdf/PdfSplitServiceImpl.java @@ -1,13 +1,22 @@ package com.wmyun.module.system.service.pdf; +import cn.hutool.core.io.IoUtil; +import com.wmyun.module.infra.api.file.FileApi; import com.wmyun.module.system.controller.admin.pdf.vo.PdfResultVO; +import jakarta.annotation.Resource; import org.apache.pdfbox.pdmodel.PDDocument; +import org.apache.pdfbox.rendering.PDFRenderer; +import org.springframework.mock.web.MockMultipartFile; import org.springframework.stereotype.Service; +import org.springframework.web.multipart.MultipartFile; -import java.io.File; -import java.io.IOException; +import javax.imageio.ImageIO; +import java.awt.image.BufferedImage; +import java.io.*; +import java.nio.file.Files; import java.util.ArrayList; import java.util.List; +import java.util.UUID; /** @@ -18,26 +27,28 @@ import java.util.List; */ @Service -public class PdfSplitServiceImpl implements PdfSplitService{ +public class PdfSplitServiceImpl implements PdfSplitService { + @Resource + private FileApi fileService; + public List splitPdf(File sourceFile, String outputDir) throws IOException { List results = new ArrayList<>(); try (PDDocument document = PDDocument.load(sourceFile)) { int totalPages = document.getNumberOfPages(); - for (int i = 0; i < totalPages; i++) { // 创建单页文档 PDDocument singlePageDoc = new PDDocument(); singlePageDoc.addPage(document.getPage(i)); - // 生成文件名和路径 String fileName = "page_" + (i + 1) + ".pdf"; File outputFile = new File(outputDir, fileName); singlePageDoc.save(outputFile); - + MultipartFile multipartFile = convertToMultipartFile(outputFile.getAbsolutePath()); + String filePath = fileService.createFile(fileName, null, IoUtil.readBytes(multipartFile.getInputStream())); // 构建返回VO results.add(new PdfResultVO( fileName, - outputFile.getAbsolutePath(), + filePath, outputFile.length() )); singlePageDoc.close(); @@ -46,6 +57,17 @@ public class PdfSplitServiceImpl implements PdfSplitService{ return results; } + public MultipartFile convertToMultipartFile(String filePath) throws IOException { + File file = new File(filePath); + FileInputStream input = new FileInputStream(file); + return new MockMultipartFile( + "file", // 表单字段名 + file.getName(), // 原始文件名 + Files.probeContentType(file.toPath()), // MIME类型（需根据实际文件类型修改） + input // 文件流 + ); + } + public void splitPdfByPages(File sourceFile, String outputDir, int startPage, int endPage) throws IOException { try (PDDocument document = PDDocument.load(sourceFile)) { int totalPages = document.getNumberOfPages(); @@ -77,4 +99,27 @@ public class PdfSplitServiceImpl implements PdfSplitService{ } } } + // PDF转多张图片（每页一张） + public List PdfToImageConverter(File pdfFile, String outputDir) throws IOException { + List results = new ArrayList<>(); + try (PDDocument document = PDDocument.load(pdfFile)) { + PDFRenderer renderer = new PDFRenderer(document); + for (int i = 0; i < document.getNumberOfPages(); i++) { + BufferedImage image = renderer.renderImageWithDPI(i, 300); // 300 DPI + String imageName = "page_" + (i + 1) + ".png"; + File outputImage = new File(outputDir, imageName); + ImageIO.write(image, "PNG", outputImage); +// imagePaths.add(outputImage.getAbsolutePath()); + MultipartFile multipartFile = convertToMultipartFile(outputImage.getAbsolutePath()); + String filePath = fileService.createFile(imageName, null, IoUtil.readBytes(multipartFile.getInputStream())); + // 构建返回VO + results.add(new PdfResultVO( + imageName, + filePath, + outputImage.length() + )); + } + } + return results; + } }