3. import com.itextpdf.text.Document;
    4. import com.itextpdf.text.PageSize;
    5. import com.itextpdf.text.pdf.PdfWriter;
    6. import com.itextpdf.tool.xml.XMLWorkerHelper;
    7. import freemarker.template.Configuration;
    8. import freemarker.template.Template;
    10. import javax.servlet.http.HttpServletResponse;
    11. import java.io.*;
    12. import java.net.URLEncoder;
    13. import java.nio.charset.StandardCharsets;
    14. import java.util.Map;
    16. /**
    17.  * @author X-MD
    18.  * @version 1.0.0
    19.  * @Description 根据HTML魔板导出PDF
    20.  * @createTime 2022年01月17日 16:47:00
    21.  */
    22. public class PdfHtmlUtil {
    23.     private static Configuration freemarkerCfg;
    24.     static {
    25.         //freemarker的模板目录
    26.         try {
    27.             freemarkerCfg = new Configuration(Configuration.VERSION_2_3_28);
    28.             String templatePath = PdfHtmlUtil.class.getResource("/").getPath() + "/templates";
    29.             freemarkerCfg.setDirectoryForTemplateLoading(new File(templatePath));
    30.         } catch (IOException e) {
    31.             e.printStackTrace();
    32.         }
    33.     }
    35.     /**
    36.      * 预览PDF
    37.      *
    38.      * @param response
    39.      * @param data     数据
    40.      * @param template templates文件下的HTML文件名称（test.html）
    41.      */
    42.     public static void createHtmlPdfPreview(HttpServletResponse response, String filename, Map<String, Object> data, String template) {
    43.         try (
    44.                 ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
    45.         ) {
    46.             createHtmlPdf(outputStream, data, template);
    47.             // inline(默认)当前页面打开;attachment 下载到本地=
    48.             response.setContentType("application/pdf");
    49.             response.addHeader("Content-Disposition", "inline;filename=" + URLEncoder.encode(filename, "UTF-8") + ".pdf");
    50.             OutputStream stream = response.getOutputStream();
    51.             outputStream.writeTo(stream);
    52.             outputStream.flush();
    53.         } catch (Exception e) {
    54.             e.printStackTrace();
    55.         }
    56.     }
    58.     /**
    59.      * 下载PDF
    60.      *
    61.      * @param response
    62.      * @param data     数据
    63.      * @param template templates文件下的HTML文件名称（test.html）
    64.      */
    65.     public static void createHtmlPdfDownload(HttpServletResponse response, String filename, Map<String, Object> data, String template) {
    66.         try (
    67.                 ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
    68.         ) {
    69.             createHtmlPdf(outputStream, data, template);
    70.             response.setContentType("application/x-download");
    71.             response.addHeader("Content-Disposition", "inline;filename=" + URLEncoder.encode(filename, "UTF-8") + ".pdf");
    72.             OutputStream stream = response.getOutputStream();
    73.             outputStream.writeTo(stream);
    74.             outputStream.flush();
    75.         } catch (Exception e) {
    76.             e.printStackTrace();
    77.         }
    78.     }
    80.     private static void createHtmlPdf(ByteArrayOutputStream outputStream, Map<String, Object> data, String template) {
    81.         String content = freeMarkerRender(data, template);
    82.         Document document = new Document(PageSize.A4);
    83.         try (
    84.                 ByteArrayInputStream stream = new ByteArrayInputStream(content.getBytes(StandardCharsets.UTF_8))
    85.         ) {
    86.             PdfWriter pdfWriter = PdfWriter.getInstance(document, outputStream);
    87.             document.open();
    88.             XMLWorkerHelper xmlWorkerHelper = XMLWorkerHelper.getInstance();
    89.             xmlWorkerHelper.parseXHtml(pdfWriter, document, stream, StandardCharsets.UTF_8, new AsianFontProvider());
    90.         } catch (Exception e) {
    91.             e.printStackTrace();
    92.         } finally {
    93.             document.close();
    94.         }
    95.     }
    98.     /**
    99.      * freemarker渲染html
    100.      */
    101.     private static <T> String freeMarkerRender(T data, String htmlTmp) {
    102.         try (
    103.                 Writer out = new StringWriter()
    104.         ) {
    105.             // 获取模板,并设置编码方式
    106.             Template template = freemarkerCfg.getTemplate(htmlTmp);
    107.             // 合并数据模型与模板，将合并后的数据和模板写入到流中，这里使用的字符流
    108.             template.process(data, out);
    109.             out.flush();
    110.             return out.toString();
    111.         } catch (Exception e) {
    112.             e.printStackTrace();
    113.         }
    114.         return null;
    115.     }
    117. }