JAVA通过模板生成DOCX文档（2）-蒲公英云

JAVA通过模板生成DOCX文档（2）

叁歲伎倆 2022-06-14 06:16 324阅读 0赞

第一篇链接

看第一篇文章会对原理更理解一些，但是比较繁琐，所以写了这一篇，怎么获取和修改xml在上一篇都有，不赘述了

import freemarker.template.Configuration;
import freemarker.template.Template;
import java.io.*;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;
public class Main {
    public static void main(String[] args) {
        // 路径
        String templatepath = "D:\\Projects\\lib\\generateDocx";
        String docxname = "demo.docx";
        String xmlname = "demo.xml";
        String tmpxmlpath = "D:\\Projects\\lib\\generateDocx\\complete.xml";
        String targetpath = "D:\\Projects\\lib\\generateDocx\\complete.docx";
        // 数据
        Map<String,Object> data = new HashMap();
        data.put("words","这里是文字");
        // 生成文档
        try {
            generate(templatepath, docxname, xmlname, tmpxmlpath, targetpath, data);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    /**
     * @Description 根据参数生成docx合同文档
     * @author belle.wang
     * @param templatepath 模板所在文件夹
     * @param docxname docx格式模板文件名(不带路径)
     * @param xmlname xml格式模板，有freemaker标记(不带路径)
     * @param tmpxmlpath 临时xml文件路径
     * @param targetPath 目标路径
     * @param param 待填充数据
     * @return
     * @throws Exception
     */
    private static boolean generate(String templatepath, String docxname, String xmlname,
                                    String tmpxmlpath, String targetPath, Map<String, Object> param) throws Exception {
        Configuration cfg = new Configuration();
        cfg.setDirectoryForTemplateLoading(new File(templatepath));
        Template template = cfg.getTemplate(xmlname);
        template.setOutputEncoding("UTF-8");
        Writer out = new FileWriter(new File(tmpxmlpath));
        // 数据放到模板xml里面，生成带数据的xml
        template.process(param, out);
        if (out != null) {
            out.close();
        }
        // 带数据的xml生成docx
        File file = new File(tmpxmlpath);
        File docxFile = new File(templatepath + "/" + docxname);
        ZipFile zipFile = new ZipFile(docxFile);
        Enumeration<? extends ZipEntry> zipEntrys = zipFile.entries();
        ZipOutputStream zipout = new ZipOutputStream(new FileOutputStream(targetPath));
        int len = -1;
        byte[] buffer = new byte[1024];
        while (zipEntrys.hasMoreElements()) {
            ZipEntry next = zipEntrys.nextElement();
            InputStream is = zipFile.getInputStream(next);
            // 把输入流的文件传到输出流中 如果是word/document.xml由我们输入
            zipout.putNextEntry(new ZipEntry(next.toString()));
            if ("word/document.xml".equals(next.toString())) {
                InputStream in = new FileInputStream(file);
                while ((len = in.read(buffer)) != -1) {
                    zipout.write(buffer, 0, len);
                }
                in.close();
            } else {
                while ((len = is.read(buffer)) != -1) {
                    zipout.write(buffer, 0, len);
                }
                is.close();
            }
        }
        zipout.close();
        return true;
    }
}