java

关注公众号 jb51net

关闭
首页 > 软件编程 > java > java替换文件内容

java如何替换word/doc文件中的内容

作者:五官一体即忢

docx格式的文件本质上是一个XML文件,只要用占位符在指定的地方标记,然后替换掉标记出的内容,这篇文章主要介绍了java替换word/doc文件中的内容,需要的朋友可以参考下

docx格式的文件本质上是一个XML文件,只要用占位符在指定的地方标记,然后替换掉标记出的内容,就能达到我们的目的

封装成工具类

import java.io.*;
import java.nio.charset.StandardCharsets;
import java.nio.file.*;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;
/**
 * Author ht
 * Date: 2023-06-30
 */
//替换.docx文件的工具类
public class DocxUtil {
    //    保存随机生的解压出.docx文件的保存目录
    private static final String cachePath = System.getProperty("user.dir") + File.separator + UUID.randomUUID().toString().replaceAll("-", "");
    //    保存模板文件的输入流和模板压缩输入流
    private static final List<InputStream> iss = new ArrayList<>();
    //    替换docx文件中的内容
    public static void compile(FileInputStream is, Map<String, String> data, FileOutputStream os) throws IOException {
//        获取模板保存的目录
        Path path = Paths.get(cachePath);
//        如果不存在或者当前的模板输入流不在输入流集合中就解压模板
        if (!Files.exists(path) || !iss.contains(is)) {
//        解压docx文件
            unDocx(is);
        }
//        替换document.xml的内容
        replaceXmlContent(data);
//        把解压的文件重新压缩成docx文件
        buildDocx(os);
    }
    //    解压docx文件
    private static void unDocx(FileInputStream template) throws IOException {
//        解压出来文件保存的目录
        File cacheDir = new File(cachePath);
        if (!cacheDir.exists()) {
//            不存在就先创建该文件夹
            Files.createDirectory(Paths.get(cacheDir.getPath()));
        }
//        创建压缩包输入流
        ZipInputStream zis = new ZipInputStream(template);
//        把需要关闭的模板输入流存到集合中
        iss.add(template);
        iss.add(zis);
//        获取首个压缩文件中的文件
        ZipEntry entry = zis.getNextEntry();
//        内容缓冲区
        byte[] buff = new byte[1024 * 10];
//        循环获取压缩包中的压缩文件
        while (entry != null) {
//            判断当前的压缩文件是文件夹还是文件
            if (entry.isDirectory()) {
//                如果是文件夹就要创建对应的文件夹
                Path tempDir = Paths.get(cacheDir.getPath() + File.separator + entry.getName());
//                不存在就创建
                if (!Files.exists(tempDir)) {
                    Files.createDirectory(tempDir);
                }
            } else {
//                如果是文件就创建文件写入内容
                File file = new File(cacheDir, entry.getName());
//                创建文件的输出流
                FileOutputStream fos = new FileOutputStream(file);
//                写入内容到输出流
                int len;
                while ((len = zis.read(buff)) != -1) {
                    fos.write(buff, 0, len);
                }
//                zis.transferTo(fos); // jdk9及更高版本可以用这个
//                关闭当前文件的输出流
                fos.close();
            }
//            获取下一个压缩文件
            entry = zis.getNextEntry();
        }
        System.out.println("===== 模板docx文件解压完成 =====");
    }
    //    替换document.xml的内容
    private static void replaceXmlContent(Map<String, String> data) throws IOException {
//        document.xml的位置
        String documentXmlPath = cachePath + File.separator + "word" + File.separator + "document.xml";
//        获取document.xml文件的输入流
        FileInputStream fis = new FileInputStream(documentXmlPath);
//        读取xml中的内容
        byte[] buff = new byte[fis.available()];
        fis.read(buff);
//        获取xml文件中的内容
        String content = new String(buff);
//        拿到Map集合中的数据和值,替换掉对应位置的内容
        for (String key : data.keySet()) {
            String value = data.get(key);
            System.out.println("替换 [ {{" + key + "}} ] 为 => [ " + value + " ]");
            content = content.replaceAll("\\{\\{" + key + "\\}\\}", value);
        }
//        把替换好的内容写入原来的document.xml文件中
        FileOutputStream fos = new FileOutputStream(documentXmlPath);
//        写入内容
        fos.write(content.getBytes(StandardCharsets.UTF_8));
//        关闭流
        fos.close();
        fis.close();
        System.out.println("===== 替换完成 =====");
    }
    //    把替换好document.xml内容的文件夹重新压缩成docx文件
    private static void buildDocx(FileOutputStream template) throws IOException {
//        创建压缩文件输出流
        ZipOutputStream zos = new ZipOutputStream(new DataOutputStream(template));
//        获取要压缩文件的路径
        Path path = Paths.get(cachePath);
//        遍历当前压缩文件路径下的所有文件
        Files.walkFileTree(path, new SimpleFileVisitor<>() {
            @Override
            public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
//                获取当前文件的父目录的名字
                String parent = file.getParent().getFileName().toString();
//                压缩文件的名字
                String zipFileName;
//                如果父目录是根目录名字就是当前的名字否则就是夫目录加上当前的名字
                if (parent.equals(path.toFile().getName())) {
                    zipFileName = file.toFile().getName();
                } else {
//                    如果父目录的夫目录是word就需要word+当前文件的父目录+当前文件的名字
                    if ("word".equals(file.getParent().getParent().toFile().getName())) {
                        zipFileName = "word" + File.separator + parent + File.separator + file.toFile().getName();
                    } else {
//                        如果不是就是当前文件的父目录+当前文件的名字
                        zipFileName = parent + File.separator + file.toFile().getName();
                    }
                }
//                把文件添加到压缩包中
                zos.putNextEntry(new ZipEntry(zipFileName));
//                获取当前文件的输入流
                DataInputStream zis = new DataInputStream(new FileInputStream(file.toFile()));
//                写入内容到当前的压缩文件
                int len;
                byte[] buff = new byte[1024 * 10];
                while ((len = zis.read(buff)) != -1) {
                    zos.write(buff, 0, len);
                }
                zis.close();
//                zis.transferTo(zos); // jdk9及更高版本可以用这个
                return super.visitFile(file, attrs);
            }
        });
//        关闭流
        zos.close();
        template.close();
        System.out.println("===== 把解压的文件重新压缩成.docx ====");
    }
    //    关闭模板输入流和清除缓存文件
    public static void closeTemplateAndClearCache() {
//        关闭集合中的所有流
        for (InputStream inputStream : iss) {
            try {
                inputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        //            删除缓存文件
        deleteCache(new File(cachePath));
        System.out.println("===== 关闭了模板,清理了缓存 =====");
    }
    //    删除不用的缓存解压文件目录
    private static void deleteCache(File cachePath) {
//        先删文件夹里面的文件
        for (File file : Objects.requireNonNull(cachePath.listFiles())) {
//            是文件夹就递归调用
            if (file.isDirectory()) {
                deleteCache(file);
            } else {
//                是文件就直接删
                file.delete();
            }
        }
//        删掉自己
        cachePath.delete();
    }
}

使用

import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
/**
 * Author ht
 * Date: 2023-06-30
 */
public class App {
    public static void main(String[] args) throws IOException {
//        填充的数据
        Map<String, String> data = new HashMap<>();
        data.put("name", "hello world");
        data.put("time", "2022年10月1日");
//        模板输入流
        FileInputStream is = new FileInputStream("src/main/resources/one.docx");
//        模板输出流
        FileOutputStream os = new FileOutputStream("src/main/resources/test1.docx");
        FileOutputStream os1 = new FileOutputStream("src/main/resources/test2.docx");
//        生成模板
        DocxUtil.compile(is, data, os);
        DocxUtil.compile(is, data, os1);
        //    关闭模板输入流和清除缓存文件
        DocxUtil.closeTemplateAndClearCache();
    }
}

记录一些代码和问题处理方式,需要参考的请谨慎。

到此这篇关于java如何替换word/doc文件中的内容的文章就介绍到这了,更多相关java替换文件内容内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文