HTML生成下载PDF文件或ZIP压缩包文件

1.工具类（下载到本地）

public class FreemarkerReportWriter {

    /**
     * 字体文件名称
     */
    private final static String DEFAULT_FONT = "yahei.ttf";

 /** 
   * templateContent ftl模版内容 
   * data ftl模版数据 
   * file 下载本地PDF文件
   * classPath 文件路径
 */
    public static void createPdf(String templateContent, Map<String, Object> data, File file, String classPath) {
        FileOutputStream outputStream = null;
        ITextRenderer renderer = new ITextRenderer();
        try {
            Configuration cfg = new Configuration(Configuration.DEFAULT_INCOMPATIBLE_IMPROVEMENTS);
            StringTemplateLoader stringLoader = new StringTemplateLoader();
            stringLoader.putTemplate("myTemplate", templateContent);
            cfg.setTemplateLoader(stringLoader);
            Template template = cfg.getTemplate("myTemplate", "utf-8");
            String htmlData = FreeMarkerTemplateUtils.processTemplateIntoString(template, data);
            outputStream = new FileOutputStream(file);
            ITextFontResolver fontResolver = renderer.getFontResolver();
            // 解决中文乱码问题，fontPath为中文字体地址
            fontResolver.addFont(classPath + DEFAULT_FONT, BaseFont.IDENTITY_H, BaseFont.NOT_EMBEDDED);
            renderer.setDocumentFromString(htmlData);
            renderer.layout();
            renderer.createPDF(outputStream);
        } catch (Exception e) {
            log.error("生成失败", e);
        } finally {
            renderer.finishPDF();
            IOUtils.closeQuietly(outputStream);
        }
    }

    public static void exportPdf(String templateContent, Map<String, Object> data, HttpServletResponse response) {
        File file = null;
        try {
            String classPath = PdfExportUtil.class.getProtectionDomain().getCodeSource().getLocation().getPath();
            log.info("classPath路径:{}", classPath);
            String fileName = System.currentTimeMillis() + ".pdf";
            createPdf(templateContent, data, new File(classPath + fileName), classPath);
            log.info("pdf长度:{}", new File(classPath + fileName).length());
            file = new File(classPath + fileName);
            ServletOutputStream outputStream = response.getOutputStream();
            FileInputStream fileInputStream = new FileInputStream(file);
            byte[] buffer = new byte[1024];
            int len;
            while ((len = fileInputStream.read(buffer)) != -1) {
                outputStream.write(buffer, 0, len);
            }
            fileInputStream.close();
            outputStream.flush();
            outputStream.close();
        } catch (Exception e) {
            log.error("生成PDF失败", e);
        } finally {
            // 清理文件
            file.delete();
        }
    }

    public static void exportZip(String templateContent, List<Map<String, Object>> mapList, HttpServletResponse response) {
        try {
            String classPath = PdfExportUtil.class.getProtectionDomain().getCodeSource().getLocation().getPath();
            log.info("classPath路径:{}", classPath);
            ZipOutputStream zips = new ZipOutputStream(new BufferedOutputStream(response.getOutputStream()));
            zips.setMethod(ZipOutputStream.DEFLATED);
            DataOutputStream os = null;
            for (Map<String, Object> map : mapList) {
                File file = null;
                try {
                    String fileName = System.currentTimeMillis() + ".pdf";
                    createPdf(templateContent, map, new File(classPath + fileName), classPath);
                    log.info("pdf长度:{}", new File(classPath + fileName).length());
                    file = new File(classPath + fileName);
                    zips.putNextEntry(new ZipEntry(fileName));
                    os = new DataOutputStream(zips);
                    InputStream is = new FileInputStream(file);
                    byte[] b = new byte[1024];
                    int length;
                    while ((length = is.read(b)) != -1) {
                        os.write(b, 0, length);
                    }
                    is.close();
                    zips.closeEntry();
                } catch (Exception e) {
                    log.error("生成PDF失败", e);
                } finally {
                    // 清理文件
                    file.delete();
                }
            }
            os.flush();
            os.close();
            zips.close();
        } catch (Exception e) {
            log.error("生成ZIP失败", e);
        }
    }
}

2.工具类（浏览器预览）

public class FreemarkerReportWriter {

    /**
     * 字体文件名称
     */
    private final static String DEFAULT_FONT_PATH = "fonts/simsun.ttf";
    private final static String DEFAULT_FONT = "SimSun";
    private final static String DEFAULT_PDF = ".pdf";
    private final static String TEMPLATE_NAME = "myTemplate";

    /**
     * fileName 文件名称
     * templateContent ftl模版内容
     * data ftl模版数据
     * attachment 是否以下载附件的方式查看报表，否则以在线方式查看报表
     * response HttpServletResponse
     */
    public static void exportPdf(String fileName, String templateContent, Map<String, Object> data, boolean attachment, HttpServletResponse response) {
        Assert.hasLength(fileName, "The parameter `fileName` can not be empty.");
        Assert.hasLength(templateContent, "The parameter `templateContent` can not be empty.");
        if (!fileName.toLowerCase().endsWith(DEFAULT_PDF)) {
            fileName += ".pdf";
        }
        ServletOutputStream outputStream = null;
        // 设置响应头
        response.setContentType("application/pdf");
        MimeDispositionType mimeDispositionType = attachment ? MimeDispositionType.attachment : MimeDispositionType.inline;
        response.setHeader("Content-Disposition", FileUtil.getContentDisposition(fileName, mimeDispositionType));
        try {
            byte[] bytes = createPdf(templateContent, data);
            log.debug("pdf file length:{}", bytes.length);
            outputStream = response.getOutputStream();
            outputStream.write(bytes);
        } catch (Exception e) {
            log.error("create pdf file failed:", e);
        } finally {
            IoUtil.close(outputStream);
        }
    }

    /**
     * templateContent ftl模版内容
     * mapList ftl模版数据
     * response HttpServletResponse
     */
    public static void exportZip(String templateContent, List<Map<String, Object>> mapList, HttpServletResponse response) {
        Assert.hasLength(templateContent, "The parameter `templateContent` can not be empty.");
        try {
            ZipOutputStream zips = new ZipOutputStream(new BufferedOutputStream(response.getOutputStream()));
            zips.setMethod(ZipOutputStream.DEFLATED);
            DataOutputStream os = null;
            for (Map<String, Object> map : mapList) {
                String fileName = ObjectUtil.isEmpty(map.get("fileName")) ? System.currentTimeMillis() + ".pdf" : String.valueOf(map.get("fileName"));
                try {
                    byte[] bytes = createPdf(templateContent, map);
                    log.debug("pdf file length:{}", bytes.length);
                    zips.putNextEntry(new ZipEntry(fileName));
                    os = new DataOutputStream(zips);
                    os.write(bytes);
                } catch (Exception e) {
                    log.error("create pdf failed：", e);
                } finally {
                    zips.closeEntry();
                }
            }
            IoUtil.close(os);
            IoUtil.close(zips);
        } catch (Exception e) {
            log.error("create zip failed：", e);
        }
    }

    public static byte[] createPdf(String templateContent, Map<String, Object> data) {
        try {
            long startTime = System.currentTimeMillis();
            Configuration cfg = new Configuration(Configuration.DEFAULT_INCOMPATIBLE_IMPROVEMENTS);
            StringTemplateLoader stringLoader = new StringTemplateLoader();
            stringLoader.putTemplate(TEMPLATE_NAME, templateContent);
            cfg.setTemplateLoader(stringLoader);
            Template template = cfg.getTemplate(TEMPLATE_NAME, StandardCharsets.UTF_8.name());
            String htmlString = FreeMarkerTemplateUtils.processTemplateIntoString(template, data);
            log.info("htmlString Execution time :{}", System.currentTimeMillis() - startTime);
            ByteArrayOutputStream os = new ByteArrayOutputStream();
            PdfRendererBuilder builder = new PdfRendererBuilder();
            builder.useFont(() -> {
                try {
                    return FreemarkerReportWriter.class.getClassLoader().getResourceAsStream(DEFAULT_FONT_PATH);
                } catch (Exception e) {
                    log.error("createPdf error, load {} file failed.", DEFAULT_FONT_PATH);
                    throw new RuntimeException(e);
                }
            }, DEFAULT_FONT, 400, BaseRendererBuilder.FontStyle.NORMAL, true);

            builder.withHtmlContent(htmlString, "");
            builder.toStream(os);
            builder.run();
            log.info("createPdf Execution time :{}", System.currentTimeMillis() - startTime);
            return os.toByteArray();
        } catch (Exception e) {
            log.error("create pdf file failed:", e);
            return new byte[1024];
        }
    }
}

3.controller接口

@GetMapping("/exportPdf")
@ApiOperation("导出PDF")
public UnifyResponse<Void> exportPdf(@RequestParam("principalId") String principalId, HttpServletResponse response) {
    PrincipalInstrumentVO instrumentVO = principalInstrumentService.detail(principalId);
    if (BeanUtil.isNotEmpty(instrumentVO)) {
        // 设置响应头
        response.setContentType("application/pdf");
        response.setCharacterEncoding("utf-8");
        response.setHeader("Content-Disposition", "inline; filename=\"pdf.pdf\"");
        PdfExportUtil.exportPdf(instrumentVO.getHtmlContent(), new HashMap<>(8), response);
    }
    return null;
}

@GetMapping("/exportZip")
@ApiOperation("导出ZIP")
public UnifyResponse<Void> exportZip(@RequestParam("principalId") String principalId, HttpServletResponse response) {
    List<Map<String, Object>> mapList = new ArrayList<>();
    Map<String, Object> map1 = new HashMap<>(16);
    map1.put("invoice", "发票1");
    map1.put("serialNumber", "123");
    map1.put("orderNumber", "123");
    mapList.add(map1);
    Map<String, Object> map2 = new HashMap<>(16);
    map2.put("invoice", "发票2");
    map2.put("serialNumber", "456");
    map2.put("orderNumber", "456");
    mapList.add(map2);
    PrincipalInstrumentVO instrumentVO = principalInstrumentService.detail(principalId);
    // 设置响应头
    response.setContentType("multipart/form-data");
    response.setCharacterEncoding("utf-8");
    response.setHeader("Content-Disposition", "attachment; filename=\"zip.zip\"");
    PdfExportUtil.exportZip(instrumentVO.getHtmlContent(), mapList, response);
    return null;
}

4.说明

templateContent为FreeMarker模板的HTML内容