检验报告 - PDF

quality-review/src/main/java/com/ruoyi/service/impl/PdfTemplateManagementServiceImpl.java

@@ -17,6 +17,7 @@ import com.ruoyi.system.mapper.SysDictDataMapper;
 import io.minio.MinioClient;
 import io.minio.ObjectWriteArgs;
 import io.minio.PutObjectArgs;
+import lombok.SneakyThrows;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
@@ -24,8 +25,11 @@ import org.springframework.transaction.annotation.Transactional;
 
 import java.io.ByteArrayInputStream;
 import java.io.ByteArrayOutputStream;
+import java.io.FileOutputStream;
 import java.io.IOException;
 import java.lang.reflect.Field;
+import java.nio.file.Files;
+import java.nio.file.Paths;
 import java.util.*;
 import java.util.List;
 
@@ -64,51 +68,72 @@ public class PdfTemplateManagementServiceImpl implements PdfTemplateManagementSe
      * @param taskId
      * @return
      */
+    @SneakyThrows
     @Override
-    public String generateInspectionReport(Long taskId) throws Exception {
+    public String generateInspectionReport(Long taskId) {
         PdfReader reader = null;
         PdfStamper ps = null;
         ByteArrayOutputStream bos = null;
-        try {
-            // 模板绝对路径--服务器
-            String fileName = "/template/receipt_template_03_27.pdf";
-            // 读取现有模板内容
-            reader = new PdfReader(fileName);
-            // 创建输出流
-            bos = new ByteArrayOutputStream();
-            // 实例化PdfStamper准备编辑pdf内容
-            ps = new PdfStamper(reader, bos);
-            // 获取表单所有元素
-            AcroFields fields = ps.getAcroFields();
-
-            // 设置具体字体格式的编码, 不设置的话中文可能不会显示
-            BaseFont bf = BaseFont.createFont("STSongStd-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
-            fields.addSubstitutionFont(bf);
-            // 测试数据
-            InspectionReportPdfVO inspectionReportPdfVO = requireData(taskId);
-
-            // 渲染数据
-            renderData(fields, inspectionReportPdfVO, ps);
-
-            //必须要调用这个，否则文档不会生成的
-            ps.setFormFlattening(true);
-            ps.close();
-
-        }catch (Exception e){
-            e.printStackTrace();
-        }finally {
-            if (bos != null){
-                bos.close();
-            }
-            if(reader != null){
-                reader.close();
-            }
-        }
-        assert bos != null;
-        // 名字
+        // 模板绝对路径--服务器
+        String fileName = "/template/receipt_template_03_27.pdf";
+        // 读取现有模板内容
+        reader = new PdfReader(fileName);
+        // 创建输出流
+        bos = new ByteArrayOutputStream();
+        // 实例化PdfStamper准备编辑pdf内容
+        ps = new PdfStamper(reader, bos);
+        // 获取表单所有元素
+        AcroFields fields = ps.getAcroFields();
+        // 设置具体字体格式的编码, 不设置的话中文可能不会显示
+        BaseFont bf = BaseFont.createFont("STSongStd-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
+        fields.addSubstitutionFont(bf);
+        // 测试数据
+        InspectionReportPdfVO inspectionReportPdfVO = requireData(taskId);
+        // 渲染数据
+        renderData(fields, inspectionReportPdfVO, ps);
+        // 必须要调用这个，否则文档不会生成的
+        ps.setFormFlattening(true);
+        // 关闭
+        ps.close();
+        bos.close();
+        reader.close();
+//        // 把第一个模板填入数据后生成一个oss
+//        String templateName = uploadMinio(bos, "整车信息安全检验临时模板");
+//        // 创建一个空PDF, 融合两个PDF
+//        Document document = new Document();
+//        ByteArrayOutputStream out = new ByteArrayOutputStream();
+//        PdfCopy copy = new PdfCopy(document, out);
+//        document.open();
+//        // 第一个PDF
+//        PdfReader readerFirst = new PdfReader(minioUrl() + templateName);
+//        int numberPage = readerFirst.getNumberOfPages();
+//        for (int i = 1;i < numberPage;i++) {
+//            PdfImportedPage importedPage = copy.getImportedPage(readerFirst, i);
+//            copy.addPage(importedPage);
+//        }
+//        // 第二个PDF
+//        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
+//        PdfBaseWriter writer = new PdfBaseWriter(outputStream);
+//        writer.open();
+//        // 代码渲染
+//        writer.setParagraph("企业留档文件", 30f, Element.ALIGN_CENTER, 0,15f, 180f);
+//        writer.close();
+//        String codeName = uploadMinio(outputStream, "整车信息安全检验临时代码");
+//        PdfReader pr = new PdfReader(minioUrl() + codeName);
+//        copy.addPage(copy.getImportedPage(pr, 1));
+//        document.close();
+
         return uploadMinio(bos, "整车信息安全检验报告-" + getReportName());
     }
 
+    /**
+     * minio对象存储查看图片地址
+     * @return
+     */
+    private String minioUrl() {
+        return minioEndpoint + "/" + bucketName + "/";
+    }
+
     private String getReportName() {
         return DateUtils.parseDateToStr("yyyyMMddHHmm", new Date());
     }