手把手教你使用 Java 線上生成 pdf 文件

程式設計師志哥發表於2022-03-13

一、介紹

在實際的業務開發的時候,研發人員往往會碰到很多這樣的一些場景,需要提供相關的電子憑證資訊給使用者,例如網銀/支付寶/微信購物支付的電子發票、訂單的庫存列印單、各種電子簽署合同等等,以方便使用者檢視、列印或者下載。

例如下圖的電子發票!

熟悉這塊業務的童鞋,一定特別清楚,目前最常用的解決方案是:把相關的資料資訊,通過一些技術手段生成對應的 PDF 檔案,然後返回給使用者,以便預覽、下載或者列印。

不太熟悉這項技術的童鞋,也不用著急,今天我們一起來詳細瞭解一下線上生成 PDF 檔案的技術實現手段!

二、案例實現

在介紹這個程式碼實踐之前,我們先來了解一下這個第三方庫:iText,對,沒錯,它就是我們今天的主角。

iText是著名的開放原始碼站點sourceforge一個專案,是用於生成PDF文件的一個java類庫,通過iText不僅可以生成PDFrtf的文件,而且還可以將XMLHtml檔案轉化為PDF檔案。

iText目前有兩套版本,分別是iText5iText7iText5應該是網上用的比較多的一個版本。iText5因為是很多開發者參與貢獻程式碼,因此在一些規範和設計上存在不合理的地方。iText7是後來官方針對iText5的重構,兩個版本差別還是挺大的。不過在實際使用中,一般用到的都比較簡單的 API,所以不用特別拘泥於使用哪個版本。

2.1、新增 iText 依賴包

在使用它之前,我們先引人相關的依賴包!

<dependencies>
    <!-- pdf:start -->
    <dependency>
        <groupId>com.itextpdf</groupId>
        <artifactId>itextpdf</artifactId>
        <version>5.5.11</version>
    </dependency>
    <dependency>
        <groupId>com.itextpdf.tool</groupId>
        <artifactId>xmlworker</artifactId>
        <version>5.5.11</version>
    </dependency>
    <!-- 支援中文 -->
    <dependency>
        <groupId>com.itextpdf</groupId>
        <artifactId>itext-asian</artifactId>
        <version>5.2.0</version>
    </dependency>
    <!-- 支援css樣式渲染 -->
    <dependency>
        <groupId>org.xhtmlrenderer</groupId>
        <artifactId>flying-saucer-pdf-itext5</artifactId>
        <version>9.1.16</version>
    </dependency>
    <!-- 轉換html為標準xhtml包 -->
    <dependency>
        <groupId>net.sf.jtidy</groupId>
        <artifactId>jtidy</artifactId>
        <version>r938</version>
    </dependency>
    <!-- pdf:end -->    
</dependencies>

2.2、簡單實現

老規矩,我們先來一個hello world,程式碼如下:

public class CreatePDFMainTest {

    public static void main(String[] args) throws Exception {
        Document document = new Document(PageSize.A4);
        //第二步,建立Writer例項
        PdfWriter.getInstance(document, new FileOutputStream("hello.pdf"));
        //建立中文字型
        BaseFont bfchinese = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
        Font fontChinese = new Font(bfchinese, 12, Font.NORMAL);
        //第三步,開啟文件
        document.open();
        //第四步,寫入內容
        Paragraph paragraph = new Paragraph("hello world", fontChinese);
        document.add(paragraph);
        //第五步,關閉文件
        document.close();
    }
}

開啟hello.pdf檔案,內容如下!

2.3、複雜實現

在實際的業務開發中,因為業務場景非常複雜,而且變化快,我們往往不會採用上面介紹的寫入內容方式來生成檔案,而是採用HTML檔案轉化為PDF檔案。

例如下面這張入庫單!

我們應該如何快速實現呢?

首先,我們採用html語言編寫一個入庫單頁面,將其命令為printDemo.html,原始碼如下:

<html>
	<head></head>
	<body>
		<meta http-equiv="Content-Type" content="text/html; charset=UTF-8" />
		<title>出庫單</title>
		<div>
			<div>
				<table width="100%" border="0" cellspacing="0" cellpadding="0">
					<tbody>
						<tr>
							<td height="40" colspan="2"><h3 style="font-weight: bold; text-align: center; letter-spacing: 5px; font-size: 24px;">入庫單</h3></td>
							<td width="12%" height="20" rowspan="2">
								<img style="width: 105px;height: 105px;" src="data:image/jpeg;base64,iVBORw0KGgoAAAANSUhEUgAAAH0AAAB9AQAAAACn+1GIAAAAqElEQVR42u3VMQ7DMAwDQP6A//8lx24qKRRw0s1yu8Uw4OQGIaHsBHUfLzzwAxCAInoZg6dI9dUUBIOyHEG56CmodAaxwtfbboLTVWpeU9+EDAH37m9CmkTYxDGUE0agMIakk3y4Ut8G37iom02M4bPniHWAtqFDTjjSGLrZvXAOmTnL1124C73r6Yo8Ane61k6eQeVjIM2h482D1RwScrpNjuH5R/0b3s6ZZNyKlt3iAAAAAElFTkSuQmCC" />
							</td>
						</tr>
						<tr>
							<td width="50%" height="30">操作人:xxx</td>
							<td width="50%" height="30" colspan="2">建立時間:2021-09-14 12:00:00</td>
						</tr>
					</tbody>
				</table>
			</div>
			<div style="margin-top: 5px; margin-bottom: 6px; margin-left: 4px"></div>
			<div>
				<table width="100%"
					style="border-collapse: collapse; border-spacing: 0;border:0px;">
						<tr style="height: 25px;">
							<td style="background: #eaeaea; text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;"
							 width="10%">序號</td>
							<td style="background: #eaeaea; text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;"
							 width="30%">商品</td>
							<td style="background: #eaeaea; text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;"
							 width="30%">單位</td>
							<td style="background: #eaeaea; text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000; border-right: 1px solid #000000;"
							 width="30%">數量</td>
						</tr>
						<tr>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;">1</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;">xxx沐浴露</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;">箱</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000; border-right: 1px solid #000000;">3</td>
						</tr>
						<tr>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;">2</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;">xxx洗髮水</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;">箱</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000; border-right: 1px solid #000000;">4</td>
						</tr>
						<tr>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;">3</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;">xxx洗衣粉</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000;">箱</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000; border-right: 1px solid #000000;">5</td>
						</tr>
						<tr>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000; border-bottom: 1px solid #000000;">4</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000; border-bottom: 1px solid #000000;">xxx洗面奶</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000; border-bottom: 1px solid #000000;">箱</td>
							<td style="text-align: center; border-left: 1px solid #000000; border-top: 1px solid #000000; border-right: 1px solid #000000; border-bottom: 1px solid #000000;">5</td>
						</tr>
				</table>
			</div>
		</div>
	</body>

</html>

接著,我們將html檔案轉成PDF檔案,原始碼如下:

public class CreatePDFMainTest {


    /**
     * 建立PDF檔案
     * @param htmlStr
     * @throws Exception
     */
    private static void writeToOutputStreamAsPDF(String htmlStr) throws Exception {
        String targetFile = "pdfDemo.pdf";
        File targeFile = new File(targetFile);
        if(targeFile.exists()) {
            targeFile.delete();
        }

        //定義pdf檔案尺寸,採用A4橫切
        Document document = new Document(PageSize.A4, 25, 25, 15, 40);// 左、右、上、下間距
        //定義輸出路徑
        PdfWriter writer = PdfWriter.getInstance(document, new FileOutputStream(targetFile));
        PdfReportHeaderFooter header = new PdfReportHeaderFooter("", 8, PageSize.A4);
        writer.setPageEvent(header);
        writer.addViewerPreference(PdfName.PRINTSCALING, PdfName.NONE);
        document.open();

        // CSS
        CSSResolver cssResolver = new StyleAttrCSSResolver();
        CssAppliers cssAppliers = new CssAppliersImpl(new XMLWorkerFontProvider(){

            @Override
            public Font getFont(String fontname, String encoding, boolean embedded, float size, int style, BaseColor color) {
                try {
                    //用於中文顯示的Provider
                    BaseFont bfChinese = BaseFont.createFont("STSongStd-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
                    return new Font(bfChinese, size, style);
                } catch (Exception e) {
                    return super.getFont(fontname, encoding, size, style);
                }
            }
        });

        //html
        HtmlPipelineContext htmlContext = new HtmlPipelineContext(cssAppliers);
        htmlContext.setTagFactory(Tags.getHtmlTagProcessorFactory());
        htmlContext.setImageProvider(new AbstractImageProvider() {
            @Override
            public Image retrieve(String src) {
                //支援圖片顯示
                int pos = src.indexOf("base64,");
                try {
                    if (src.startsWith("data") && pos > 0) {
                        byte[] img = Base64.decode(src.substring(pos + 7));
                        return Image.getInstance(img);
                    } else if (src.startsWith("http")) {
                        return Image.getInstance(src);
                    }
                } catch (BadElementException ex) {
                    return null;
                } catch (IOException ex) {
                    return null;
                }
                return null;
            }

            @Override
            public String getImageRootPath() {
                return null;
            }
        });


        // Pipelines
        PdfWriterPipeline pdf = new PdfWriterPipeline(document, writer);
        HtmlPipeline html = new HtmlPipeline(htmlContext, pdf);
        CssResolverPipeline css = new CssResolverPipeline(cssResolver, html);

        // XML Worker
        XMLWorker worker = new XMLWorker(css, true);
        XMLParser p = new XMLParser(worker);
        p.parse(new ByteArrayInputStream(htmlStr.getBytes()));

        document.close();
    }

    /**
     * 讀取 HTML 檔案
     * @return
     */
    private static String readHtmlFile() {
        StringBuffer textHtml = new StringBuffer();
        try {
            File file = new File("printDemo.html");
            BufferedReader reader = new BufferedReader(new FileReader(file));
            String tempString = null;
            // 一次讀入一行,直到讀入null為檔案結束
            while ((tempString = reader.readLine()) != null) {
                textHtml.append(tempString);
            }
            reader.close();
        } catch (IOException e) {
            return null;
        }
        return textHtml.toString();
    }

    public static void main(String[] args) throws Exception {
        //讀取html檔案
        String htmlStr = readHtmlFile();
        //將html檔案轉成PDF
        writeToOutputStreamAsPDF(htmlStr);
    }
}

執行程式,開啟pdfDemo.pdf,結果如下!

2.4、變數替換方式

上面的html檔案,是我們事先已經編輯好的,才能正常渲染。

但是在實際的業務開發的時候,例如下面的商品內容,完全是動態的,還是xxx-202109入庫單的名稱,以及二維碼,都是動態的。

這個時候,我們可以採用freemarker模板引擎,通過定義變數來動態填充內容,直到轉換出來的結果就是我們想要的html頁面。

當然,還有一種辦法,例如下面這個,我們也可以在html頁面裡面定義${name}變數,然後在讀取完檔案之後,我們將其變數進行替換成我們想填充的任何值,這其實也是模板引擎最核心的一個玩法。

<html>
	<head>
		<meta charset="utf-8">
		<title></title>
	</head>
	<body>
		<div>您好:${name}</div>
		<div>歡迎,登入部落格網站</div>
	</body>
</html>

三、總結

itext框架是一個非常實用的第三方pdf檔案生成庫,尤其是面對比較簡單的pdf檔案內容渲染的時候,它完全滿足我們的需求。

但是對於那種複雜的pdf文件,可能需要我們自己單獨進行適配開發。

如果想要獲取原始碼,關注下方公眾號,並回復【cccc10】即可獲取!

四、參考

1、JAVA使用ItextPDF

相關文章