java POI替换word模板中的占位符,支持表格扩展


前言

仅支持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);
			}
		}
	}

注:

  1. map参数为word正文的占位符键值对映射,tableMap参数为word表格中的占位符键值对映射,注意区分。
  2. needAddRow用于控制是否仅做替换,还是需要给word中的现有表格增加行
  3. rowSize表示需要给现有表格追加几行数据
  4. rowNum表示从现有表格的第几行开始追加
  5. word模板中的占位符内容建议手敲,复制很容易出错,注意不要有空格

总结

原文地址:https://blog.csdn.net/wxz980927155/article/details/84945979

  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值