poi实现多个word文档的合并测试代码样例
提供一个代码样例实现使用poi工具包合并多个word文档:
工具类 MergeDocUtils
package com.yootii.bdy.util.test;
import org.apache.commons.collections.CollectionUtils;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.xwpf.usermodel.BreakType;
import org.apache.poi.xwpf.usermodel.Document;
import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.apache.poi.xwpf.usermodel.XWPFPictureData;
import org.apache.xmlbeans.XmlOptions;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTBody;
import java.io.File;
import java.io.FileInputStream;
import java.io.OutputStream;
import java.nio.file.Files;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class MergeDocUtils {
public static void mergeDoc(List<File> fileList, File newFile) {
try {
System.out.println("正在合并文件:" + newFile);
OutputStream dest = Files.newOutputStream(newFile.toPath());
if(CollectionUtils.isEmpty(fileList)){
System.out.println("待合并文件为空,合并文件失败:" + newFile.toString());
return;
}
// 判断文件格式是不是以docx或doc结尾
for (int i = 0; i < fileList.size(); i++) {
if (!(fileList.get(i).getName().toLowerCase().endsWith(".doc")
|| fileList.get(i).getName().toLowerCase().endsWith(".docx"))) {
System.out.println("文件格式错误,合并失败:" + newFile.toString());
break;
}
}
// 打开第一个文件
// Path path1 = Paths.get(fileList.get(0));
// boolean exists = Files.exists(path1);
// System.out.println("文件是否存在: " + exists);
ArrayList<XWPFDocument> documentList = new ArrayList<XWPFDocument>();
XWPFDocument doc = null;
for (int i = 0; i < fileList.size(); i++) {
FileInputStream in = new FileInputStream(fileList.get(i).getPath());
OPCPackage open = OPCPackage.open(in);
XWPFDocument document = new XWPFDocument(open);
documentList.add(document);
}
for (int i = 0; i < documentList.size(); i++) {
doc = documentList.get(0);
// 加入分页
if (i != documentList.size()-1){
documentList.get(i).createParagraph().createRun().addBreak(BreakType.PAGE);
}
// 从第二个文件开始追加合并
if (i != 0) {
// 但是会出现在首行为空的情况
// documentList.get(i).createParagraph().setPageBreak(true);
appendBody(doc, documentList.get(i));
}
}
// 输出合并之后的文件
if (doc != null) {
doc.write(dest);
}
}catch (Exception e){
e.printStackTrace();
// System.out.println("合并文件失败:" + newFile.toString());
}
}
public static void appendBody(XWPFDocument src, XWPFDocument append) throws Exception {
CTBody src1Body = src.getDocument().getBody();
CTBody src2Body = append.getDocument().getBody();
List<XWPFPictureData> allPictures = append.getAllPictures();
// 记录图片合并前及合并后的ID
Map<String, String> map = new HashMap();
for (XWPFPictureData picture : allPictures) {
String before = append.getRelationId(picture);
// 将原文档中的图片加入到目标文档中
String after = src.addPictureData(picture.getData(), Document.PICTURE_TYPE_PNG);
map.put(before, after);
}
appendBodyDetail(src1Body, src2Body, map);
}
private static void appendBodyDetail(CTBody src, CTBody append, Map<String, String> map) throws Exception {
XmlOptions optionsOuter = new XmlOptions();
optionsOuter.setSaveOuter();
String appendString = append.xmlText(optionsOuter);
// 去掉追加word内容中的 w:sectPr 标签,确保合成的word中只有一个 w:sectPr 标签对
// 避免合成的word文档打开之后会提示有些内容读不出来,导致文件损坏
String rgex = "<[\\s]*?w:sectPr[^>]*?>[\\s\\S]*?<[\\s]*?\\/[\\s]*?w:sectPr[\\s]*?>";
appendString = appendString.replaceAll(rgex, "");
String srcString = src.xmlText();
// String regex = regexText(srcString, "w:sectPr");
// System.out.println(regex);
String prefix = srcString.substring(0, srcString.indexOf(">") + 1);
String mainPart = srcString.substring(srcString.indexOf(">") + 1, srcString.lastIndexOf("<"));
String sufix = srcString.substring(srcString.lastIndexOf("<"));
String addPart = appendString.substring(appendString.indexOf(">") + 1, appendString.lastIndexOf("<"));
if (map != null && !map.isEmpty()) {
// 对xml字符串中图片ID进行替换
for (Map.Entry<String, String> set : map.entrySet()) {
addPart = addPart.replace(set.getKey(), set.getValue());
}
}
// 将两个文档的xml内容进行拼接
CTBody makeBody = CTBody.Factory.parse(prefix + mainPart + addPart + sufix);
src.set(makeBody);
}
}
测试类 WordMergeTest
package com.yootii.bdy.util.test;
import java.io.File;
import java.time.LocalDate;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
/**
* @BelongsProject: ip_service
* @BelongsPackage: com.yootii.bdy.util.test
* @Author: chuanwei.yang 42624
* @CreateTime: 2024-08-12 11:47
* @Description: TODO
* @Version: 1.0
*/
public class WordMergeTest {
public static void main(String[] args) throws Exception {
// new一个list 模拟要合并的word对象集合
List<File> docFileList = new ArrayList<>();
docFileList.add(new File("D:\\ipdoc\\template\\POA\\2\\LD2021000106\\1723521822027\\所函(二审)LD2021000106.doc"));
docFileList.add(new File("D:\\ipdoc\\template\\POA\\2\\LD2021000106\\1723521822027\\所函(一审)-第三人LD2021000115.doc"));
// docFileList.add(new File("D:\\ipdoc\\template\\POA\\2\\LD2021000335\\1723430865470\\测试文件test.docx"));
// docFileList.add(new File("D:\\ipdoc\\template\\POA\\2\\LD2021000335\\1723430865470\\测试文件test2.docx"));
//合并之后doc存储路径 此处读的配置文件的存储路径 D:/pdfData/
String docPath = "D:\\ipdoc\\template\\POA\\2\\LD2021000335\\1723430865470\\";
//当前日期+UUID 作为文件名防止重复
String fileName = LocalDate.now() + "_" + UUID.randomUUID().toString().replaceAll("-", "");
//合并之后doc存储路径
String mergeDocUrl = docPath+fileName+".docx";
//转成file对象
File mergeDocFile = new File(mergeDocUrl);
//合并doc
MergeDocUtils.mergeDoc(docFileList,mergeDocFile);
System.out.println("合并word成功");
}
}
把测试方法里的文件路径改一下,改成你自己要用的文件就可以,实现效果就是例如把文件A和B合并在一起,文件A有2.5页(也就是第三页没占满),文件B有1页,合并之后的文件有4页,会在每个独立文件之后插入一个分页符再进行新文件内容的追加,保证阅读的连贯性。
好看请赞,养成习惯:) 本文来自博客园,作者:靠谱杨, 转载请注明原文链接:https://www.cnblogs.com/rainbow-1/p/18361082
欢迎来我的51CTO博客主页踩一踩 我的51CTO博客
文章中的公众号名称可能有误,请统一搜索:靠谱杨的秘密基地