XMLWorkerHelper生成pdf文件添加页眉页脚

一、Controller代码

import java.io.ByteArrayInputStream;import java.io.FileOutputStream;import java.io.StringWriter;import java.nio.charset.Charset;import com.itextpdf.text.Document;import com.itextpdf.text.pdf.PdfWriter;import com.itextpdf.tool.xml.XMLWorkerHelper;import com.jfinal.log.Logger;import com.jfinal.render.FreeMarkerRender;import freemarker.template.Template;public class PdfKit {private static final Logger log = Logger.getLogger(PdfKit.class);/*** 导出pdf** @param templateView-freemarker模板相对路径* @param os-输出流* @param model-模板所需要的数据* @throws Exception*/public static void export(String templateView,Object model,String temp) throws Exception {log.info("生成的合同文件地址："+temp);FileOutputStream tempOs = new FileOutputStream(temp); Template template = FreeMarkerRender.getConfiguration().getTemplate(templateView);StringWriter result = new StringWriter();template.process(model, result);String htmlData = result.toString();//模版变量Document document = new Document();PdfWriter writer = PdfWriter.getInstance(document, tempOs);//添加水印和页码PDFBuilder builder = new PDFBuilder();writer.setPageEvent(builder);document.open();XMLWorkerHelper.getInstance().parseXHtml(writer, document,new ByteArrayInputStream(htmlData.getBytes()), Charset.forName("UTF-8"));document.close();}}

二、PDFBuilder代码

package com.sign.system.kit;import java.io.IOException;import com.itextpdf.text.Document;import com.itextpdf.text.DocumentException;import com.itextpdf.text.Element;import com.itextpdf.text.Font;import com.itextpdf.text.Image;import com.itextpdf.text.PageSize;import com.itextpdf.text.Phrase;import com.itextpdf.text.Rectangle;import com.itextpdf.text.pdf.BaseFont;import com.itextpdf.text.pdf.ColumnText;import com.itextpdf.text.pdf.PdfContentByte;import com.itextpdf.text.pdf.PdfPageEventHelper;import com.itextpdf.text.pdf.PdfTemplate;import com.itextpdf.text.pdf.PdfWriter;import com.sign.system.PathCons;import com.sign.util.PathUtil;/*** 设置页面附加属性**/public class PDFBuilder extends PdfPageEventHelper {/*** 页眉*/public String header = "";/*** 文档字体大小，页脚页眉最好和文本大小一致*/public int presentFontSize = 12;/*** 文档页面大小，最好前面传入，否则默认为A4纸张*/public Rectangle pageSize = PageSize.A4;// 模板public PdfTemplate total;// 基础字体对象public BaseFont bf = null;// 利用基础字体生成的字体对象，一般用于生成中文文字public Font fontDetail = null;/**** Creates a new instance of PdfReportM1HeaderFooter 无参构造方法.**/public PDFBuilder() {}/**** Creates a new instance of PdfReportM1HeaderFooter 构造方法.** @param yeMei*页眉字符串* @param presentFontSize*数据体字体大小* @param pageSize*页面文档大小，A4，A5，A6横转翻转等Rectangle对象*/public PDFBuilder(String yeMei, int presentFontSize, Rectangle pageSize) {this.header = yeMei;this.presentFontSize = presentFontSize;this.pageSize = pageSize;}public void setHeader(String header) {this.header = header;}public void setPresentFontSize(int presentFontSize) {this.presentFontSize = presentFontSize;}/**** TODO 文档打开时创建模板** @see com.itextpdf.text.pdf.PdfPageEventHelper#onOpenDocument(com.itextpdf.text.pdf.PdfWriter,* com.itextpdf.text.Document)*/public void onOpenDocument(PdfWriter writer, Document document) {total = writer.getDirectContent().createTemplate(50, 50);// 共 页 的矩形的长宽高}/**** TODO 关闭每页的时候，写入页眉，写入'第几页共'这几个字。** @see com.itextpdf.text.pdf.PdfPageEventHelper#onEndPage(com.itextpdf.text.pdf.PdfWriter,* com.itextpdf.text.Document)*/public void onEndPage(PdfWriter writer, Document document) {this.addPage(writer, document);this.addWatermark(writer);}//加分页public void addPage(PdfWriter writer, Document document){try {if (bf == null) {bf = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", false);}if (fontDetail == null) {fontDetail = new Font(bf, presentFontSize, Font.NORMAL);// 数据体字体}} catch (DocumentException e) {e.printStackTrace();} catch (IOException e) {e.printStackTrace();}// 1.写入页眉ColumnText.showTextAligned(writer.getDirectContent(),Element.ALIGN_LEFT, new Phrase(header, fontDetail),document.left(), document.top() + 20, 0);// 2.写入前半部分的 第 X页/共int pageS = writer.getPageNumber();String foot1 = "第 " + pageS + " 页 /共";Phrase footer = new Phrase(foot1, fontDetail);// 3.计算前半部分的foot1的长度，后面好定位最后一部分的'Y页'这俩字的x轴坐标，字体长度也要计算进去 = lenfloat len = bf.getWidthPoint(foot1, presentFontSize);// 4.拿到当前的PdfContentBytePdfContentByte cb = writer.getDirectContent();// 5.写入页脚1，x轴就是(右margin+左margin + right() -left()- len)/2.0F// 再给偏移20F适合人类视觉感受，否则肉眼看上去就太偏左了// ,y轴就是底边界-20,否则就贴边重叠到数据体里了就不是页脚了；注意Y轴是从下往上累加的，最上方的Top值是大于Bottom好几百开外的。ColumnText.showTextAligned(cb,Element.ALIGN_CENTER,footer,(document.rightMargin() + document.right()+ document.leftMargin() - document.left() - len) / 2.0F + 20F,document.bottom() - 20, 0);// 6.写入页脚2的模板（就是页脚的Y页这俩字）添加到文档中，计算模板的和Y轴,X=(右边界-左边界 - 前半部分的len值)/2.0F +// len ， y 轴和之前的保持一致，底边界-20cb.addTemplate(total, (document.rightMargin() + document.right()+ document.leftMargin() - document.left()) / 2.0F + 20F,document.bottom() - 20); // 调节模版显示的位置}//加水印public void addWatermark(PdfWriter writer){// 水印图片Image image;try {image = Image.getInstance(PathUtil.getRootClassPath(PathCons.FILE_PDF_IMG));PdfContentByte content = writer.getDirectContentUnder();content.beginText();// 开始写入水印for(int k=0;k<5;k++){for (int j = 0; j <4; j++) {image.setAbsolutePosition(150*j,170*k);content.addImage(image);}}content.endText();} catch (IOException | DocumentException e) {// TODO Auto-generated catch blocke.printStackTrace();}}/**** TODO 关闭文档时，替换模板，完成整个页眉页脚组件** @see com.itextpdf.text.pdf.PdfPageEventHelper#onCloseDocument(com.itextpdf.text.pdf.PdfWriter,* com.itextpdf.text.Document)*/public void onCloseDocument(PdfWriter writer, Document document) {// 7.最后一步了，就是关闭文档的时候，将模板替换成实际的 Y 值,至此，page x of y 制作完毕，完美兼容各种文档size。total.beginText();total.setFontAndSize(bf, presentFontSize);// 生成的模版的字体、颜色String foot2 = " " + (writer.getPageNumber()-1) + " 页";total.showText(foot2);// 模版显示的内容total.endText();total.closePath();}}