Java使用freemarker模板导出pdf文件

package com.phbc.spark.utils;

import java.io.BufferedWriter;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.StringWriter;
import java.io.UnsupportedEncodingException;
import java.io.Writer;
import java.util.Locale;
import java.util.Map;


import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;


import org.xhtmlrenderer.pdf.ITextFontResolver;
import org.xhtmlrenderer.pdf.ITextRenderer;


import com.lowagie.text.pdf.BaseFont;


import freemarker.template.Configuration;
import freemarker.template.Template;


public class PDFUtil {

      /**
* pdf文档导出
* @param request
* @param ftlName 文件名
* @param root 导出数据
* @return
* @throws Exception
*/

    public static ByteArrayOutputStream createPDF(HttpServletRequest request, String ftlName,Map root) throws Exception {    
           String basePath =request.getSession().getServletContext().getRealPath("/");//绝对路径    
           Configuration cfg = new Configuration();    
           try {    
               cfg.setLocale(Locale.CHINA);    
               cfg.setEncoding(Locale.CHINA, "UTF-8");    
               //设置编码    
               cfg.setDefaultEncoding("UTF-8");  
               //设置模板路径    
               cfg.setDirectoryForTemplateLoading(new File(basePath + "/WEB-INF/template"));    
               //获取模板    
               Template template = cfg.getTemplate(ftlName);    
               template.setEncoding("UTF-8"); 
              
              Writer writer = new StringWriter();    
               //数据填充模板    
               template.process(root, writer);    
               String str = writer.toString();    
               //pdf生成  
               DocumentBuilder builder = DocumentBuilderFactory.newInstance().newDocumentBuilder();  
        
               ITextRenderer iTextRenderer = new ITextRenderer();  
               iTextRenderer.setDocumentFromString(str);    
               //设置字体  其他字体需要添加字体库    
               ITextFontResolver fontResolver = iTextRenderer.getFontResolver();    
               fontResolver.addFont(basePath + "/WEB-INF/template/simsun.ttc", BaseFont.IDENTITY_H, BaseFont.NOT_EMBEDDED);
//               System.out.println(basePath);
//               System.out.println(str);
               iTextRenderer.setDocument(builder.parse(new ByteArrayInputStream(str.getBytes("utf-8"))),null);  
               iTextRenderer.layout();   
               //生成PDF    
               ByteArrayOutputStream baos = new ByteArrayOutputStream();    
               iTextRenderer.createPDF(baos);    
               baos.close();    
               return baos;    
           } catch(Exception e) {    
               throw new Exception(e);    
           }    
       }  
      
       /**  
        * 客户端下载pdf  
        * @author  
        * 创建时间
        * @param response  
        * @param bytes  
        * @param filename  
        */  
       public static void renderPdf(HttpServletResponse response, final byte[] bytes, final String filename) {    
         /*  initResponseHeader(response, PDF_TYPE);  */  
           setFileDownloadHeader(response, filename, ".pdf");    
           if (null != bytes) {    
               try {    
                   response.getOutputStream().write(bytes);    
                   response.getOutputStream().flush();    
               } catch (IOException e) {    
                   throw new IllegalArgumentException(e);    
               }    
           }    
       }    
         
       /**   
        * 分析并设置contentType与headers.   
        */  
       /*    
       private HttpServletResponse initResponseHeader(HttpServletResponse response, final String contentType, final String... headers) {    
           // 分析headers参数    
           String encoding = DEFAULT_ENCODING;    
           boolean noCache = DEFAULT_NOCACHE;    
           for (String header : headers) {    
               String headerName = StringUtils.substringBefore(header, ":");    
               String headerValue = StringUtils.substringAfter(header, ":");    
               if (StringUtils.equalsIgnoreCase(headerName, HEADER_ENCODING)) {    
                   encoding = headerValue;    
               } else if (StringUtils.equalsIgnoreCase(headerName, HEADER_NOCACHE)) {    
                   noCache = Boolean.parseBoolean(headerValue);    
               } else {    
                   throw new IllegalArgumentException(headerName + "不是一个合法的header类型");    
               }    
           }    
           // 设置headers参数    
           String fullContentType = contentType + ";charset=" + encoding;    
           response.setContentType(fullContentType);    
           if (noCache) {    
               // Http 1.0 header    
               response.setDateHeader("Expires", 0);    
               response.addHeader("Pragma", "no-cache");    
               // Http 1.1 header    
               response.setHeader("Cache-Control", "no-cache");    
           }    
           return response;    
       }  */  
         
       /**  
        *  设置让浏览器弹出下载对话框的Header  
        * @author  
        * 创建时间 
        * @param response  
        * @param fileName  
        * @param fileType  
        */  
       public static void setFileDownloadHeader(HttpServletResponse response, String fileName, String fileType) {    
           try {    
               // 中文文件名支持    
               String encodedfileName = new String(fileName.getBytes("GBK"), "ISO8859-1");    
               response.setHeader("Content-Disposition", "attachment; filename=\"" + encodedfileName + fileType + "\"");    
           } catch (UnsupportedEncodingException e) {    
           }    
       }  
}

你可能感兴趣的:(freemaker)