使用EasyPoi word模板导出循环写段落写表格段落
import cn.afterturn.easypoi.cache.WordCache;
import cn.afterturn.easypoi.entity.ImageEntity;
import cn.afterturn.easypoi.util.PoiElUtil;
import cn.afterturn.easypoi.util.PoiPublicUtil;
import cn.afterturn.easypoi.word.entity.MyXWPFDocument;
import cn.afterturn.easypoi.word.entity.params.ExcelListEntity;
import cn.afterturn.easypoi.word.parse.excel.ExcelMapParse;
import cn.hutool.core.bean.BeanUtil;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.xwpf.usermodel.*;
import org.apache.xmlbeans.XmlCursor;
import org.apache.xmlbeans.XmlObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.File;
import java.io.FileOutputStream;
import java.util.*;
public class WordParagraphHolder {
private static final Logger logger = LoggerFactory.getLogger(WordParagraphHolder1.class);
private XWPFDocument sourceDocument;
private String targetFileName;
private Map<String, Object> data;
public WordParagraphHolder(XWPFDocument sourceDocument, String targetFileName, Map<String, Object> data) {
this.sourceDocument = sourceDocument;
this.targetFileName = targetFileName;
this.data = data;
}
public String execute() throws Exception {
MyXWPFDocument second = null;
FileOutputStream fos2 = null;
try {
// 循环生成段落
Map<String, Object> data = this.data;
this.parseAllParagraphic(sourceDocument.getParagraphs(), data);
File targetFile = new File(targetFileName);
FileOutputStream fos = FileUtils.openOutputStream(targetFile);
sourceDocument.write(fos);
// 关闭流
fos.close();
sourceDocument.close();
// 循环段落赋值
second = WordCache.getXWPFDocument(targetFile.getPath());
this.evalAllParagraphic(second.getParagraphs(), data);
fos2 = FileUtils.openOutputStream(targetFile);
second.write(fos2);
} catch (Exception e) {
logger.error("生成word段落失败:", e);
return "";
} finally {
// 关闭流
fos2.close();
second.close();
}
return targetFileName;
}
private void evalAllParagraphic(List<XWPFParagraph> paragraphs, Map<String, Object> map) throws Exception {
int size = paragraphs.size();
Map<String, Integer> indexMap = new HashMap<>();
for (int i = 0; i < size; ++i) {
XWPFParagraph paragraph = (XWPFParagraph) paragraphs.get(i);
logger.info("段落内容：{}", paragraph.getText());
if (paragraph.getText().indexOf("(") != -1) {
Object obj = checkThisParagraphNeedIterator(paragraph, map);
String listKey = getParagraphListKey(paragraph);
Integer index = indexMap.getOrDefault(listKey, 0);
if (Objects.nonNull(obj) && obj instanceof List && index < ((List) obj).size()) {
Object o = ((List) obj).get(index);
parseParagraph(paragraph, o, listKey);
index++;
indexMap.put(listKey, index);
}
}
}
}
private void parseAllParagraphic(List<XWPFParagraph> paragraphs, Map<String, Object> map) throws Exception {
int size = paragraphs.size();
List<XWPFParagraph> paragraphList = new ArrayList<>();
// 循环找出需要复制的段落
for (int i = 0; i < size; ++i) {
XWPFParagraph paragraph = (XWPFParagraph) paragraphs.get(i);
if (paragraph.getText().indexOf("(") != -1) {
Object obj = checkThisParagraphNeedIterator(paragraph, map);
if (Objects.nonNull(obj) && obj instanceof List) {
paragraphList.add(paragraph);
}
}
}
// 复制段落
for (XWPFParagraph paragraph : paragraphList) {
System.out.println(paragraph.getText());
Object obj = checkThisParagraphNeedIterator(paragraph, map);
addParagraph(paragraph, (List) obj);
// 删除模板段落
XWPFDocument document = paragraph.getDocument();
document.removeBodyElement(document.getPosOfParagraph(paragraph));
}
}
/**
* 复制段落
*
* @param source 原段落
* @param doc
*/
public XWPFParagraph createParagraph(XWPFParagraph source, XWPFDocument doc) {
// 使用游标创建一个新行
XmlCursor cursor = source.getCTP().newCursor();
XWPFParagraph newParagraph = doc.insertNewParagraph(cursor);
newParagraph.getCTP().set(source.getCTP().copy());
return newParagraph;
}
private Object checkThisParagraphNeedIterator(XWPFParagraph paragraph, Map<String, Object> map) throws Exception {
String text = paragraph.getText().trim();
if (text != null && text.contains("fe:") && text.startsWith("(")) {
text = text.replace("!fe:", "").replace("$fe:", "").replace("fe:", "").replace("(", "");
String[] keys = text.replaceAll("\\s{1,}", " ").trim().split(" ");
Object result = PoiPublicUtil.getParamsValue(keys[0], map);
return Objects.nonNull(result) ? result : new ArrayList(0);
} else {
return null;
}
}
private String getParagraphListKey(XWPFParagraph paragraph) throws Exception {
String text = paragraph.getText().trim();
if (text != null && text.contains("fe:") && text.startsWith("(")) {
text = text.replace("!fe:", "").replace("$fe:", "").replace("fe:", "").replace("(", "");
String[] keys = text.replaceAll("\\s{1,}", " ").trim().split(" ");
return keys[0];
} else {
return null;
}
}
/**
* 赋值段落
*
* @param paragraph
* @param obj
* @throws Exception
*/
public void parseParagraph(XWPFParagraph paragraph, Object obj, String listKey) throws Exception {
String listname = paragraph.getText().trim();
boolean contains = listname.contains("fe:");
if (!contains) {
return;
}
Map<String, Object> objectMap = BeanUtil.beanToMap(obj);
parseThisParagraph(paragraph, objectMap, listKey);
}
/**
* 增加段落
*
* @param paragraph
* @param list
* @throws Exception
*/
public void addParagraph(XWPFParagraph paragraph, List<Object> list) throws Exception {
XWPFParagraph currentParagraph = paragraph;
System.out.println("start for each data list :" + list.size());
Iterator var11 = list.iterator();
while (var11.hasNext()) {
Object obj = var11.next();
this.createParagraph(currentParagraph, currentParagraph.getDocument());
}
}
/**
* 遍历段落赋值
*
* @param paragraph
* @param map
* @throws Exception
*/
private void parseThisParagraph(XWPFParagraph paragraph, Map<String, Object> map, String listKey) throws Exception {
XWPFRun currentRun = null;
String currentText = "";
Boolean isfinde = false;
List<Integer> runIndex = new ArrayList();
XWPFRun preRun = null;
for (int i = 0; i < paragraph.getRuns().size(); ++i) {
XWPFRun run = (XWPFRun) paragraph.getRuns().get(i);
String text = run.getText(0);
if (!StringUtils.isEmpty(text)) {
if (isfinde) {
currentText = currentText + text;
if (currentText.indexOf("[") == -1) {
isfinde = false;
runIndex.clear();
} else {
runIndex.add(i);
}
if (currentText.indexOf("]") != -1) {
this.changeValues(paragraph, currentRun, currentText, runIndex, map);
currentText = "";
isfinde = false;
}
} else if (text.indexOf("[") >= 0) {
currentText = text;
isfinde = true;
currentRun = run;
} else {
currentText = "";
}
if (currentText.indexOf("]") != -1) {
this.changeValues(paragraph, currentRun, currentText, runIndex, map);
isfinde = false;
}
}
// 去除多余的字符串
if (!StringUtils.isEmpty(text)) {
if (text.indexOf("(") != -1) {
preRun = run;
} else if (text.indexOf("$fe") != -1) {
// ("", 0);
// 清除第一个(
preRun.setText("", 0);
// 防止自定义前缀未被拆分为多个run,根据自定义关键字删除多余前缀。例如($fe:resultList
if (text.indexOf(listKey) != -1) {
String lastText = run.getText(0);
if (StringUtils.isNotBlank(lastText) && text.indexOf(listKey) != -1) {
String replace = lastText.substring(text.indexOf(listKey) + listKey.length());
run.setText(replace, 0);
}
}
}
// 清除最后一个)
if (i == paragraph.getRuns().size() - 1) {
if (text.indexOf(")") != -1) {
if (text.length() >= 1) {
String lastText = run.getText(0);
String replace = lastText.replace(")", "");
run.setText(replace, 0);
}
}
}
}
}
}
private void changeValues(XWPFParagraph paragraph, XWPFRun currentRun, String currentText, List<Integer> runIndex, Map<String, Object> map) throws Exception {
Object obj = getRealValue(currentText, map);
if (obj instanceof ImageEntity) {
currentRun.setText("", 0);
ExcelMapParse.addAnImage((ImageEntity) obj, currentRun);
} else {
currentText = obj.toString();
PoiPublicUtil.setWordText(currentRun, currentText);
}
for (int k = 0; k < runIndex.size(); ++k) {
((XWPFRun) paragraph.getRuns().get((Integer) runIndex.get(k))).setText("", 0);
}
runIndex.clear();
}
public static Object getRealValue(String currentText, Map<String, Object> map) throws Exception {
String params = "";
while (currentText.indexOf("[") != -1) {
params = currentText.substring(currentText.indexOf("[") + 1, currentText.indexOf("]"));
Object obj = PoiElUtil.eval(params.trim(), map);
if (obj instanceof ImageEntity || obj instanceof List || obj instanceof ExcelListEntity) {
return obj;
}
if (obj != null) {
currentText = currentText.replace("[" + params + "]", obj.toString());
} else {
currentText = currentText.replace("[" + params + "]", "");
}
}
return currentText;
}
/**
* 复制表格段落
*
* @param source 原段落
* @param cell
*/
public XWPFParagraph copyTableParagraph(XWPFParagraph source, XWPFTableCell cell) {
// 使用游标创建一个新行
XmlCursor cursor = source.getCTP().newCursor();
// 在游标位置插入新段落
XWPFParagraph newParagraph = cell.insertNewParagraph(cursor);
// 复制底层 XML 对象
XmlObject copy = source.getCTP().copy();
newParagraph.getCTP().set(copy);
// 关闭游标
cursor.dispose();
return newParagraph;
}
/**
* 表格段落复制
*
* @param document
* @param data
*/
private void createTableParagraph(XWPFDocument document, Map<String, Object> data) {
Object resultList = data.get("resultList");
if (Objects.isNull(resultList)) {
return;
}
List list = (List) resultList;
List<XWPFTable> tables = document.getTables();
for (XWPFTable table : tables) {
for (XWPFTableRow row : table.getRows()) {
for (XWPFTableCell cell : row.getTableCells()) {
XWPFParagraph xwpfParagraph = null;
List<XWPFParagraph> paragraphs = cell.getParagraphs();
for (int i = 0; i < paragraphs.size(); i++) {
XWPFParagraph paragraph = paragraphs.get(i);
String listname = paragraph.getText().trim();
if (listname.startsWith("($fe")) {
xwpfParagraph = paragraph;
}
}
if (Objects.nonNull(xwpfParagraph)) {
for (int i = 0; i < list.size() - 1; i++) {
copyTableParagraph(xwpfParagraph, cell);
}
}
}
}
}
}
/**
* 表格段落赋值
*
* @param document
* @param date
*/
private void evalTableParagraph(XWPFDocument document, Map<String, Object> date) {
List<XWPFTable> tables = document.getTables();
for (XWPFTable table : tables) {
for (XWPFTableRow row : table.getRows()) {
for (XWPFTableCell cell : row.getTableCells()) {
List<XWPFParagraph> paragraphs = cell.getParagraphs();
List<XWPFParagraph> addList = new ArrayList<>();
for (int i = 0; i < paragraphs.size(); i++) {
XWPFParagraph paragraph = paragraphs.get(i);
String listname = paragraph.getText().trim();
if (listname.startsWith("($fe")) {
addList.add(paragraph);
}
}
try {
evalAllParagraphic(addList, date);
} catch (Exception e) {
logger.error("单元格{}段落赋值异常", cell.getText(), e);
}
}
}
}
}
}