Java工具类（二） 模版导出工具类

根据模版导出Word或Excel

工作中，有时需要导出Word一些操作，最常见的方法就是根据模版导出。主要技术有：强大的POI和较为流行的模版引擎FreeMarker，个人偏爱FreeMarker。

package com.cnblog.util;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.net.URLEncoder;
import java.util.Map;
import java.util.Set;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;
import org.apache.poi.hwpf.HWPFDocument;
import org.apache.poi.hwpf.usermodel.Range;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;

import freemarker.template.Configuration;
import freemarker.template.DefaultObjectWrapper;
import freemarker.template.Template;
import freemarker.template.TemplateException;

public class ExportUtil {
    /**
     * 通过字段替换 导出word
     */
    public static void exportWord(String fileName, Map<String, String> params, HttpServletResponse response) {
        InputStream is = null;
        ServletOutputStream out = null;
        try {
            is = ExportUtil.class.getResourceAsStream("/" + fileName);
            POIFSFileSystem pfs = new POIFSFileSystem(is);
            HWPFDocument hwpf = new HWPFDocument(pfs);
            Range range = hwpf.getRange();
            Set<String> keys = params.keySet();
            for (String key : keys) {
                range.replaceText(key, params.get(key));
            }
            response.setCharacterEncoding("UTF-8");
            response.setContentType("application/octet-stream; charset=UTF-8");
            response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8"));
            out = response.getOutputStream();
            hwpf.write(out);
            out.flush();
        } catch (FileNotFoundException e) {
            throw new RuntimeException("未找到模板文件!");
        } catch (IOException e) {
            throw new RuntimeException("读取文件错误!");
        } finally {
            IOUtils.closeQuietly(out);
            IOUtils.closeQuietly(is);
        }
    }

    /**
     * 通过 FreeMarker 动态生成word/Excel
     * 一般用于导出循环数据
     * tip:图片请替换base64
     */
    public static void createWE(String fileName, String templateName, Map<String, Object> dataMap, HttpServletResponse response) {
        OutputStream out = null;
        try {
            // 创建配置实例
            Configuration cfg = new Configuration();
            cfg.setDefaultEncoding("utf-8");
            String path = ExportUtil.class.getResource("/").toString();
            cfg.setDirectoryForTemplateLoading(new File(path.substring(6, path.length())));
            cfg.setObjectWrapper(new DefaultObjectWrapper());
            // 获取模板
            Template template = cfg.getTemplate(templateName);
            // 将模板和数据模型合并生成文件
            response.setCharacterEncoding("UTF-8");
            response.setContentType("application/octet-stream; charset=UTF-8");
            response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8"));
            out = response.getOutputStream();
            Writer docout = new BufferedWriter(new OutputStreamWriter(out, "UTF-8"));
            template.process(dataMap, docout);
        } catch (FileNotFoundException e) {
            throw new RuntimeException("未找到模板文件!");
        } catch (IOException e) {
            throw new RuntimeException("读取文件错误!");
        } catch (TemplateException e) {
            throw new RuntimeException("模版文件错误!");
        } finally {
            IOUtils.closeQuietly(out);
        }
    }

}

转载于:https://www.cnblogs.com/onedir/p/4842745.html