java+freemarker实现根据word模板生成文档

开始,同事用的freemarker实现了生成.doc的文件,但是我想生成.docx的文件,于是网上到处找资料,最终找到了解决方法,JAVA通过模板生成DOCX文档(2)
这篇文章的帮助。还找了一些处理的文章,但是链接忘记了保存。下面附上源代码(只是搬运工)
新建一个处理word的工具类

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

import java.io.*;
import java.util.Enumeration;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipException;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;

public class WordUtil {
    // 初始化配置
    private Configuration configuration;
    // 模板文件目录(doc模板ftl文件,docx模板xml文件都放在此目录)
    private String templateDir = "C:\\Users\\Administrator\\Desktop\\";


    public WordUtil() {
        configuration = new Configuration();
        /** 设置编码 **/
        configuration.setDefaultEncoding("utf-8");
        /** 加载目录 **/
        try {
            configuration.setDirectoryForTemplateLoading(new File(templateDir));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 生成doc文件
     *
     * @param ftlFileName 模板ftl文件的名称
     * @param params      动态传入的数据参数
     * @param outFilePath 生成的最终doc文件的保存完整路径
     */
    public void ftlToDoc(String ftlFileName, Map params, String outFilePath) {
        try {
            /** 加载模板文件 **/
            Template template = configuration.getTemplate(ftlFileName);
            /** 指定输出word文件的路径 **/
            File docFile = new File(outFilePath);
            FileOutputStream fos = new FileOutputStream(docFile);
            Writer bufferedWriter = new BufferedWriter(new OutputStreamWriter(fos, "utf-8"), 10240);
            template.process(params, bufferedWriter);
            if (bufferedWriter != null) {
                bufferedWriter.close();
            }
        } catch (TemplateException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 生成docx文件
     *
     * @param docxTemplate    docx的模板docx文件路径
     * @param docxXmlTemplate docx的模板xml文件名称
     * @param tempDocxXmlPath docx的临时xml文件(docx的模板xml文件填充完数据生成的临时文件)
     * @param params          填充到docx的临时xml文件中的数据
     * @param toFilePath      最终输出的docx文件路径
     */
    public void xmlToDocx(String docxTemplate, String docxXmlTemplate, String tempDocxXmlPath, Map params, String toFilePath) {
        try {
            Template template = configuration.getTemplate(docxXmlTemplate);

            Writer fileWriter = new FileWriter(new File(tempDocxXmlPath));
            template.process(params, fileWriter);
            if (fileWriter != null) {
                fileWriter.close();
            }

            File docxFile = new File(docxTemplate);
            ZipFile zipFile = new ZipFile(docxFile);
            Enumeration zipEntrys = zipFile.entries();
            ZipOutputStream zipout = new ZipOutputStream(new FileOutputStream(toFilePath));
            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(new File(XmlToDocx.class.getClassLoader().getResource("").toURI().getPath()+"template/test.xml"));
                    InputStream in = new FileInputStream(tempDocxXmlPath);
                    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();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (ZipException e) {
            e.printStackTrace();
        } catch (TemplateException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (Exception e) {
            // TODO: handle exception
            e.printStackTrace();
        }
    }
    
    public void createDocx(Map dataMap,OutputStream outputStream,String document,String docxTemplate, String docxXmlTemplate, String tempDocxXmlPath) {
        ZipOutputStream zipout = null;
        try {
            //图片配置文件模板
           // ByteArrayInputStream documentXmlRelsInput =FreeMarkUtils.getFreemarkerContentInputStream(dataMap, documentXmlRels);
            //内容模板
            ByteArrayInputStream documentInput = FreeMarkUtils.getFreemarkerContentInputStream(dataMap, document);
            //最初设计的模板
            File docxFile = new File(docxTemplate);
            if (!docxFile.exists()) {
                docxFile.createNewFile();
            }
            ZipFile zipFile = new ZipFile(docxFile);
            Enumeration zipEntrys = zipFile.entries();
            zipout = new ZipOutputStream(outputStream);
            //开始覆盖文档------------------
            int len = -1;
            byte[] buffer = new byte[1024];
            while (zipEntrys.hasMoreElements()) {
                ZipEntry next = zipEntrys.nextElement();
                InputStream is = zipFile.getInputStream(next);
                if (next.toString().indexOf("media") < 0) {
                    zipout.putNextEntry(new ZipEntry(next.getName()));
                   if ("word/document.xml".equals(next.getName())) {//如果是word/document.xml由我们输入
                        if (documentInput != null) {
                            while ((len = documentInput.read(buffer)) != -1) {
                                zipout.write(buffer, 0, len);
                            }
                            documentInput.close();
                        }
                    } else {
                        while ((len = is.read(buffer)) != -1) {
                            zipout.write(buffer, 0, len);
                        }
                        is.close();
                    }
                }
            }
           
        } catch (Exception e) {
           e.printStackTrace();
        }finally {
            if(zipout!=null){
                try {
                    zipout.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if(outputStream!=null){
                try {
                    outputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }
}

再创建word模板(docx),模板图:


模板内容

将模板复制一份,重命名,内容改成这样,这样主要是为了方便改xml


新的模板

将新模板文档用压缩工具打开,如网上说的,找到word文件夹下的document.xml,复制出来,然后将所有参数替换,都换成第一份模板一样的参数名字,然后保存。并且把原来模板文档打开,将这个改好的xml文件替换模板里的word下的document.xml。
类似这样
替换了xml的模板和修改后的xml

然后开始写测试类

import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class ExportTest {
    public static void main(String[] args) {
        ExportTest test = new ExportTest();
        //test.testDoc();
        test.testDocx();
    }

    public void testDoc() {
        WordUtil wordUtil = new WordUtil();
        Map dataMap = new HashMap<>();
        dataMap.put("name", "四个空格-https://www.4spaces.org");
        wordUtil.ftlToDoc("docTemplete.ftl", dataMap, "E:\\freemarker\\testDoc.doc");
    }

    public void testDocx() {
        // docx模板文件的路径和文件名
        String docxTemplate = "C:\\Users\\Administrator\\Desktop\\freemarker\\template.docx";

        // docx模板文件名称,该文件可以直接使用解压软件打开docx文件,复制word/document.xml文件内容进行修改
        String docxXmlTemplate = "template_docx.xml";

        // docx需要的临时xml文件路径,名称和路径都无所谓,只是中间过程会用到,之后可以删除,文件不需存在,但路径必须存在
        String tempDocxXmlPath = "C:\\Users\\Administrator\\Desktop\\temp\\temp.xml";

        // 目标文件名
        String outputFilePath = "C:\\Users\\Administrator\\Desktop\\freemarker\\testDocx.docx";

        // 需要动态传入的数据
        Map params = new HashMap();
        params.put("name", "四个空格-https://www.4spaces.org");
        params.put("pm_project_name", "test");
        params.put("time", "2020.09.13-2020.09.18");
        params.put("text", "no info");
        List list = new ArrayList();
        User user = new User();
        user.setA("A");
        user.setB("B");
        user.setC("C");
        user.setD("D");
        user.setE("E");
        user.setF("F");
        user.setG("G");
        user.setH("H");
        user.setJ("J");
        user.setK("K");
        user.setM("M");
        user.setN("N");
        user.setZ("Z");
        list.add(user);
        list.add(user);
        list.add(user);
        list.add(user);
        params.put("userList", list);
        params.put("userList1", list);
        params.put("userList2", list);
        FileOutputStream outputStream = null;
        try {
            outputStream = new FileOutputStream(outputFilePath);
        } catch (FileNotFoundException e) {
            // TODO 自动生成的 catch 块
            e.printStackTrace();
        }
        WordUtil xtd = new WordUtil();
        xtd.xmlToDocx(docxTemplate, docxXmlTemplate, tempDocxXmlPath, params, outputFilePath);
        //xtd.createDocx(params, outputStream, "template.docx", docxTemplate, docxXmlTemplate, tempDocxXmlPath);
    }
}

运行,可生成docx文档文件。搬运完成!

你可能感兴趣的:(java+freemarker实现根据word模板生成文档)