前言
仅支持docx格式,支持word文档表格中的占位符替换
代码如下(示例):
/**
* @param srcPath word模板数据源路径
* @param destPath word导出路径
* @param map 关键字键值对映射
* @param tableMap 表格数据映射
* @param needAddRow 是否需要追加行
* @param rowSize 需要追加的总行数
* @param rowNum 从第几行开始追加
* @return
*/
public static File replaceWord(String srcPath, String destPath, Map<String, String> map, Map<Integer, Map<String, String>> tableMap, boolean needAddRow, int rowSize, int rowNum) {
FileOutputStream out = null;
try {
XWPFDocument document = new XWPFDocument(POIXMLDocument.openPackage(srcPath));
List<XWPFParagraph> paragraphList = document.getParagraphs();
// 替换段落中的占位符
replaceTextInParagragh(map, paragraphList);
if (needAddRow) {
replaceAndAddRowTextInTables(tableMap, document, rowSize, rowNum);
} else {
// 替换表格中的占位符
replaceTextInTables(map, document);
}
ByteArrayOutputStream ostream = new ByteArrayOutputStream();
out = new FileOutputStream(new File(destPath));
document.write(out);
out.write(ostream.toByteArray());
out.flush();
return new File(destPath);
} catch (Exception e) {
e.printStackTrace();
log.info("模板替换失败");
} finally {
try {
if (out != null) {
out.close();
}
} catch (IOException e) {
}
}
return null;
}
/**
* 替换段落中的占位符
*
* @param map map
* @param paragraphList 段落
*/
private static void replaceTextInParagragh(Map<String, String> map, List<XWPFParagraph> paragraphList) {
for (XWPFParagraph paragraph : paragraphList) {
// 遍历获取段落中所有的runs
List<XWPFRun> runs = paragraph.getRuns();
for (int i = 0; i < runs.size(); i++) {
//获取字符
String text0 = runs.get(i).getText(runs.get(i).getTextPosition());
if (text0 != null && text0.contains("$")) {
//包含占位符的字符缓存
StringBuilder cache = new StringBuilder(text0);
int endIndex = 0;
boolean contains = text0.contains("}");
//同一个run中是否包含占位符
if (!contains) {
int j = i + 1;
for (; j < runs.size(); j++) {
String text1 = runs.get(j).getText(runs.get(j).getTextPosition());
if (text1 == null) {
continue;
}
cache.append(text1);
if (text1.contains("}")) {
endIndex = j;
break;
}
}
}
if (contains || endIndex != 0) {
//处理替换
String s = cache.toString();
for (Map.Entry<String, String> entry : map.entrySet()) {
String k = entry.getKey();
String v = entry.getValue();
if (s.contains(k)) {
String replace = s.replace(k, v);
runs.get(i).setText(replace, 0);
for (int j = endIndex; j > i; j--) {
paragraph.removeRun(j);
}
break;
}
}
}
}
}
}
}
/**
* 替换表格中的文本
*
* @param map map
* @param document 文档
*/
private static void replaceTextInTables(Map<String, String> map, XWPFDocument document) {
//获取所有的表格
List<XWPFTable> tables = document.getTables();
for (XWPFTable table : tables) {
//获取每个表格的总行数
int rcount = table.getNumberOfRows();
for (int i = 0; i < rcount; i++) {
//获取表格的第i行
XWPFTableRow row = table.getRow(i);
//获取一行的所有单元格
List<XWPFTableCell> cells = row.getTableCells();
for (XWPFTableCell cell : cells) {
replaceTextInParagragh(map, cell.getParagraphs());
}
}
}
}
/**
* @param map 待替换的map集合
* @param document 文本对象
* @param rowSize 需要追加的总行数
* @param rowNum 从第几行开始追加
* @throws XmlException
* @throws IOException
*/
private static void replaceAndAddRowTextInTables(Map<Integer, Map<String, String>> map, XWPFDocument document, int rowSize, int rowNum) {
//获取所有的表格
List<XWPFTable> tables = document.getTables();
for (XWPFTable table : tables) {
// 复制第rowNum行数据
CTRow ctrow = null;
try {
ctrow = CTRow.Factory.parse(table.getRow(rowNum).getCtRow().newInputStream());
} catch (Exception e) {
e.printStackTrace();
log.error("替换表格内容时,获取行失败!");
}
// 获取复制好的行对象
XWPFTableRow rowCopy = new XWPFTableRow(ctrow, table);
//获取每个表格的总行数
int rcount = table.getNumberOfRows();
// 先将原表格中的占位符替换完成
for (int i = 0; i < rcount; i++) {
//获取表格的第i行
XWPFTableRow row = table.getRow(i);
//获取一行的所有单元格
List<XWPFTableCell> cells = row.getTableCells();
for (XWPFTableCell cell : cells) {
replaceTextInParagragh(map.get(0), cell.getParagraphs());
}
}
// 替换复制行的占位符,并追加到第rowNum行之后
for (int i = 0; i < rowSize; i++) {
List<XWPFTableCell> cells = rowCopy.getTableCells();
for (XWPFTableCell cell : cells) {
replaceTextInParagragh(map.get(i + 1), cell.getParagraphs());
}
table.addRow(rowCopy, rowNum + i);
}
}
}
注:
- map参数为word正文的占位符键值对映射,tableMap参数为word表格中的占位符键值对映射,注意区分。
- needAddRow用于控制是否仅做替换,还是需要给word中的现有表格增加行
- rowSize表示需要给现有表格追加几行数据
- rowNum表示从现有表格的第几行开始追加
- word模板中的占位符内容建议手敲,复制很容易出错,注意不要有空格
总结
原文地址:https://blog.csdn.net/wxz980927155/article/details/84945979