package com.product.print.service;
|
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import com.deepoove.poi.NiceXWPFDocument;
|
import com.deepoove.poi.XWPFTemplate;
|
import com.deepoove.poi.policy.RenderPolicy;
|
import com.deepoove.poi.template.ElementTemplate;
|
import com.deepoove.poi.template.run.RunTemplate;
|
import com.product.common.lang.StringUtils;
|
import org.apache.poi.xwpf.usermodel.*;
|
import org.apache.xmlbeans.XmlCursor;
|
import org.apache.xmlbeans.XmlObject;
|
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTTbl;
|
|
import java.util.Collections;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
|
/**
|
* @author cheng
|
* @desc poi-tl 自定义字表渲染
|
* @date 2022年5月6日17:04:31
|
*/
|
public class HackLoopTableRenderPolicy implements RenderPolicy {
|
|
@Override
|
public void render(ElementTemplate eleTemplate, Object data, XWPFTemplate template) {
|
NiceXWPFDocument doc = template.getXWPFDocument();
|
RunTemplate runTemplate = (RunTemplate) eleTemplate;
|
XWPFRun run = runTemplate.getRun();
|
String text = run.getText(0);
|
try {
|
// w:tbl-w:tr-w:tc-w:p-w:tr
|
XmlCursor newCursor = ((XWPFParagraph) run.getParent()).getCTP().newCursor();
|
newCursor.toParent();
|
newCursor.toParent();
|
newCursor.toParent();
|
XmlObject object = newCursor.getObject();
|
XWPFTable table = doc.getAllTable((CTTbl) object);
|
render(text, table, (List<Map<String, String>>) data, template, text.substring(2).substring(0, text.length() - 4));
|
run.setText("", 0);
|
} catch (Exception e) {
|
logger.error("dynamic table error:" + e.getMessage(), e);
|
}
|
}
|
|
private void render(String text, XWPFTable table, List<Map<String, String>> subListData, XWPFTemplate template, String subTableName) {
|
List<XWPFTableRow> rows = table.getRows();
|
if (rows != null && rows.size() > 0) {
|
//表格渲染开始行（子表字段表达式所在行）
|
int tableRowIndex = -1;
|
for (int i = 0; i < rows.size(); i++) {
|
XWPFTableCell cell = rows.get(i).getCell(0);
|
if (cell != null && cell.getText().indexOf(text) != -1) {
|
tableRowIndex = i + 1;
|
break;
|
}
|
}
|
if (tableRowIndex > -1) {
|
XWPFTableRow xwpfTableRow = rows.get(tableRowIndex);
|
List<XWPFTableCell> cells = xwpfTableRow.getTableCells();
|
if (cells != null && cells.size() > 0) {
|
String[] keys = new String[cells.size()];
|
for (int i = 0; i < cells.size(); i++) {
|
XWPFTableCell xwpfTableCell = cells.get(i);
|
String expression = xwpfTableCell.getText();
|
if (!StringUtils.isEmpty(expression.trim())) {
|
expression = expression.trim();
|
keys[i] = parseString(expression, subTableName);
|
}
|
}
|
table.removeRow(tableRowIndex);
|
if (CollectionUtil.isEmpty(subListData)) {
|
return;
|
}
|
for (int i = 0; i < subListData.size(); i++) {
|
table.insertNewTableRow(tableRowIndex);
|
XWPFTableRow currentRow = table.getRow(tableRowIndex);
|
int cellCount = cells.size() < keys.length ? keys.length : cells.size();
|
for (int j = 0; j < cellCount; j++) {
|
currentRow.addNewTableCell();
|
}
|
setValues(table.getRow(tableRowIndex).getTableCells(), keys, subListData.get(i));
|
tableRowIndex++;
|
}
|
}
|
}
|
}
|
}
|
|
private void setValues(List<XWPFTableCell> cells, String[] keys, Map subData) {
|
if (cells != null && keys != null && cells.size() > 0 && subData != null && subData.size() > 0) {
|
for (int i = 0; i < cells.size(); i++) {
|
String key = keys[i];
|
if (StringUtils.isEmpty(key)) {
|
continue;
|
}
|
Object val = subData.get(key) ;
|
if (val == null) {
|
val = "";
|
}
|
cells.get(i).setText(val+"");
|
}
|
}
|
|
}
|
|
|
/**
|
* 解析字符串
|
*/
|
public static String parseString(String text, String table) {
|
Matcher matcher = Pattern.compile("(\\$\\{)(" + makeQueryStringAllRegExp(table) + "\\.+)([\\w]+)(\\})").matcher(text);
|
StringBuilder sb = new StringBuilder();
|
while (matcher.find()) {
|
System.out.println(matcher.group());
|
sb.append(matcher.group(3));
|
}
|
return sb.length() > 0 ? sb.toString() : null;
|
}
|
|
/**
|
* 转义正则特殊字符 （$()*+.[]?\^{}
|
* \\需要第一个替换，否则replace方法替换时会有逻辑bug
|
*/
|
public static String makeQueryStringAllRegExp(String str) {
|
if (StringUtils.isBlank(str)) {
|
return str;
|
}
|
|
return str.replace("\\", "\\\\").replace("*", "\\*")
|
.replace("+", "\\+").replace("|", "\\|")
|
.replace("{", "\\{").replace("}", "\\}")
|
.replace("(", "\\(").replace(")", "\\)")
|
.replace("^", "\\^").replace("$", "\\$")
|
.replace("[", "\\[").replace("]", "\\]")
|
.replace("?", "\\?").replace(",", "\\,")
|
.replace(".", "\\.").replace("&", "\\&");
|
}
|
|
|
}
|
