aixan/DevTTL-Service-Template
1
0
Fork 3
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

支持Excel导出对象的多个子列表

Browse Source
This commit is contained in:
Hsy
2025-12-08 09:11:06 +08:00
parent 888b4e51b3
commit f94fc8c80d
1 changed files with 36 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
devttl-common/src/main/java/com/devttl/common/utils/poi/ExcelUtil.java
View File

@@ -21,6 +21,7 @@ import org.apache.commons.lang3.RegExUtils;
import org.apache.commons.lang3.reflect.FieldUtils; import org.apache.commons.lang3.reflect.FieldUtils;
import org.apache.poi.hssf.usermodel.*; import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ooxml.POIXMLDocumentPart; import org.apache.poi.ooxml.POIXMLDocumentPart;
import org.apache.poi.ss.formula.functions.T;
import org.apache.poi.ss.usermodel.*; import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddress; import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.ss.util.CellRangeAddressList; import org.apache.poi.ss.util.CellRangeAddressList;
@@ -129,12 +130,12 @@ public class ExcelUtil<T> {
/** /**
* 对象的子列表方法 * 对象的子列表方法
*/ */
private Method subMethod; private Map<String, Method> subMethods = new HashMap<>();
/** /**
* 对象的子列表属性 * 对象的子列表属性
*/ */
private List<Field> subFields; private Map<String, List<Field>> subFieldsMap = new HashMap<>();
/** /**
* 统计列表 * 统计列表
@@ -199,7 +200,7 @@ public class ExcelUtil<T> {
if (StringUtils.isNotEmpty(title)) { if (StringUtils.isNotEmpty(title)) {
int titleLastCol = this.fields.size() - 1; int titleLastCol = this.fields.size() - 1;
if (isSubList()) { if (isSubList()) {
titleLastCol = titleLastCol + subFields.size() - 1; titleLastCol = titleLastCol + subFieldsMap.values().size() - 1;
} }
Row titleRow = sheet.createRow(rownum == 0 ? rownum++ : 0); Row titleRow = sheet.createRow(rownum == 0 ? rownum++ : 0);
titleRow.setHeightInPoints(30); titleRow.setHeightInPoints(30);
@@ -217,14 +218,15 @@ public class ExcelUtil<T> {
if (isSubList()) { if (isSubList()) {
Row subRow = sheet.createRow(rownum); Row subRow = sheet.createRow(rownum);
int column = 0; int column = 0;
int subFieldSize = subFields != null ? subFields.size() : 0;
for (Object[] objects : fields) { for (Object[] objects : fields) {
Field field = (Field) objects[0]; Field field = (Field) objects[0];
Excel attr = (Excel) objects[1]; Excel attr = (Excel) objects[1];
CellStyle cellStyle = styles.get(StringUtils.format("header_{}_{}", attr.headerColor(), attr.headerBackgroundColor()));
if (Collection.class.isAssignableFrom(field.getType())) { if (Collection.class.isAssignableFrom(field.getType())) {
Cell cell = subRow.createCell(column); Cell cell = subRow.createCell(column);
cell.setCellValue(attr.name()); cell.setCellValue(attr.name());
cell.setCellStyle(styles.get(StringUtils.format("header_{}_{}", attr.headerColor(), attr.headerBackgroundColor()))); cell.setCellStyle(cellStyle);
int subFieldSize = subFieldsMap != null ? subFieldsMap.get(field.getName()).size() : 0;
if (subFieldSize > 1) { if (subFieldSize > 1) {
CellRangeAddress cellAddress = new CellRangeAddress(rownum, rownum, column, column + subFieldSize - 1); CellRangeAddress cellAddress = new CellRangeAddress(rownum, rownum, column, column + subFieldSize - 1);
sheet.addMergedRegion(cellAddress); sheet.addMergedRegion(cellAddress);
@@ -233,7 +235,7 @@ public class ExcelUtil<T> {
} else { } else {
Cell cell = subRow.createCell(column++); Cell cell = subRow.createCell(column++);
cell.setCellValue(attr.name()); cell.setCellValue(attr.name());
cell.setCellStyle(styles.get(StringUtils.format("header_{}_{}", attr.headerColor(), attr.headerBackgroundColor()))); cell.setCellStyle(cellStyle);
} }
} }
rownum++; rownum++;
@@ -301,13 +303,14 @@ public class ExcelUtil<T> {
Map<String, Integer> cellMap = new HashMap<String, Integer>(); Map<String, Integer> cellMap = new HashMap<String, Integer>();
// 获取表头 // 获取表头
Row heard = sheet.getRow(titleNum); Row heard = sheet.getRow(titleNum);
for (int i = 0; i < heard.getPhysicalNumberOfCells(); i++) { if (heard == null) {
throw new UtilException("文件标题行为空，请检查Excel文件格式");
}
for (int i = 0; i < heard.getLastCellNum(); i++) {
Cell cell = heard.getCell(i); Cell cell = heard.getCell(i);
if (StringUtils.isNotNull(cell)) { if (StringUtils.isNotNull(cell)) {
String value = this.getCellValue(heard, i).toString(); String value = this.getCellValue(heard, i).toString();
cellMap.put(value, i); cellMap.put(value, i);
} else {
cellMap.put(null, i);
} }
} }
// 有数据时才处理 得到类的所有field. // 有数据时才处理 得到类的所有field.
@@ -556,7 +559,8 @@ public class ExcelUtil<T> {
Field field = (Field) os[0]; Field field = (Field) os[0];
Excel excel = (Excel) os[1]; Excel excel = (Excel) os[1];
if (Collection.class.isAssignableFrom(field.getType())) { if (Collection.class.isAssignableFrom(field.getType())) {
for (Field subField : subFields) { List<Field> currentSubFields = subFieldsMap.get(field.getName());
for (Field subField : currentSubFields) {
Excel subExcel = subField.getAnnotation(Excel.class); Excel subExcel = subField.getAnnotation(Excel.class);
this.createHeadCell(subExcel, row, column++); this.createHeadCell(subExcel, row, column++);
} }
@@ -565,7 +569,7 @@ public class ExcelUtil<T> {
} }
} }
if (Type.EXPORT.equals(type)) { if (Type.EXPORT.equals(type)) {
fillExcelData(index, row); fillExcelData(index);
addStatisticsRow(); addStatisticsRow();
} }
} }
@@ -578,13 +582,13 @@ public class ExcelUtil<T> {
* @param row 单元格行 * @param row 单元格行
*/ */
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
public void fillExcelData(int index, Row row) { public void fillExcelData(int index) {
int startNo = index * sheetSize; int startNo = index * sheetSize;
int endNo = Math.min(startNo + sheetSize, list.size()); int endNo = Math.min(startNo + sheetSize, list.size());
int currentRowNum = rownum + 1; // 从标题行后开始 int currentRowNum = rownum + 1; // 从标题行后开始
for (int i = startNo; i < endNo; i++) { for (int i = startNo; i < endNo; i++) {
row = sheet.createRow(currentRowNum); Row row = sheet.createRow(currentRowNum);
T vo = (T) list.get(i); T vo = (T) list.get(i);
int column = 0; int column = 0;
int maxSubListSize = getCurrentMaxSubListSize(vo); int maxSubListSize = getCurrentMaxSubListSize(vo);
@@ -596,6 +600,7 @@ public class ExcelUtil<T> {
Collection<?> subList = (Collection<?>) getTargetValue(vo, field, excel); Collection<?> subList = (Collection<?>) getTargetValue(vo, field, excel);
if (subList != null && !subList.isEmpty()) { if (subList != null && !subList.isEmpty()) {
int subIndex = 0; int subIndex = 0;
List<Field> currentSubFields = subFieldsMap.get(field.getName());
for (Object subVo : subList) { for (Object subVo : subList) {
Row subRow = sheet.getRow(currentRowNum + subIndex); Row subRow = sheet.getRow(currentRowNum + subIndex);
if (subRow == null) { if (subRow == null) {
@@ -603,13 +608,13 @@ public class ExcelUtil<T> {
} }
int subColumn = column; int subColumn = column;
for (Field subField : subFields) { for (Field subField : currentSubFields) {
Excel subExcel = subField.getAnnotation(Excel.class); Excel subExcel = subField.getAnnotation(Excel.class);
addCell(subExcel, subRow, (T) subVo, subField, subColumn++); addCell(subExcel, subRow, (T) subVo, subField, subColumn++);
} }
subIndex++; subIndex++;
} }
column += subFields.size(); column += currentSubFields.size();
} }
} catch (Exception e) { } catch (Exception e) {
log.error("填充集合数据失败", e); log.error("填充集合数据失败", e);
@@ -920,7 +925,7 @@ public class ExcelUtil<T> {
if (attr.isExport()) { if (attr.isExport()) {
// 创建cell // 创建cell
cell = row.createCell(column); cell = row.createCell(column);
if (isSubListValue(vo) && getListCellValue(vo).size() > 1 && attr.needMerge()) { if (isSubListValue(vo) && getListCellValue(vo) > 1 && attr.needMerge()) {
if (subMergedLastRowNum >= subMergedFirstRowNum) { if (subMergedLastRowNum >= subMergedFirstRowNum) {
sheet.addMergedRegion(new CellRangeAddress(subMergedFirstRowNum, subMergedLastRowNum, column, column)); sheet.addMergedRegion(new CellRangeAddress(subMergedFirstRowNum, subMergedLastRowNum, column, column));
} }
@@ -1303,10 +1308,10 @@ public class ExcelUtil<T> {
fields.add(new Object[]{field, attr}); fields.add(new Object[]{field, attr});
} }
if (Collection.class.isAssignableFrom(field.getType())) { if (Collection.class.isAssignableFrom(field.getType())) {
subMethod = getSubMethod(field.getName(), clazz); String fieldName = field.getName();
ParameterizedType pt = (ParameterizedType) field.getGenericType(); ParameterizedType pt = (ParameterizedType) field.getGenericType();
Class<?> subClass = (Class<?>) pt.getActualTypeArguments()[0]; Class<?> subClass = (Class<?>) pt.getActualTypeArguments()[0];
this.subFields = FieldUtils.getFieldsListWithAnnotation(subClass, Excel.class); subFieldsMap.put(fieldName, FieldUtils.getFieldsListWithAnnotation(subClass, Excel.class));
} }
} }
@@ -1505,27 +1510,33 @@ public class ExcelUtil<T> {
* 是否有对象的子列表 * 是否有对象的子列表
*/ */
public boolean isSubList() { public boolean isSubList() {
return StringUtils.isNotNull(subFields) && subFields.size() > 0; return !StringUtils.isEmpty(subFieldsMap);
} }
/** /**
* 是否有对象的子列表，集合不为空 * 是否有对象的子列表，集合不为空
*/ */
public boolean isSubListValue(T vo) { public boolean isSubListValue(T vo) {
return StringUtils.isNotNull(subFields) && subFields.size() > 0 && StringUtils.isNotNull(getListCellValue(vo)) && getListCellValue(vo).size() > 0; return !StringUtils.isEmpty(subFieldsMap) && getListCellValue(vo) > 0;
} }
/** /**
* 获取集合的值 * 获取集合的值
*/ */
public Collection<?> getListCellValue(Object obj) { public int getListCellValue(Object obj) {
Object value; Collection<?> value;
int max = 0;
try { try {
value = subMethod.invoke(obj, new Object[]{}); for (String s : subMethods.keySet()) {
value = (Collection<?>) subMethods.get(s).invoke(obj);
if (value.size() > max) {
max = value.size();
}
}
} catch (Exception e) { } catch (Exception e) {
return new ArrayList<Object>(); return 0;
} }
return (Collection<?>) value; return max;
} }
/** /**