admin/lx-admin-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update 优化 excel导出合并 在初始化类时进行数据的处理

Browse Source
This commit is contained in:
疯狂的狮子li 2023-05-23 13:45:11 +08:00
parent f687bde93b
commit b8e389bb7e
1 changed files with 65 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

131
ruoyi-common/ruoyi-common-excel/src/main/java/org/dromara/common/excel/core/CellMergeStrategy.java
View File

@ -28,95 +28,94 @@ import java.util.Map;
@Slf4j @Slf4j
public class CellMergeStrategy extends AbstractMergeStrategy { public class CellMergeStrategy extends AbstractMergeStrategy {
private final List<?> list; private final List<CellRangeAddress> cellList;
private final boolean hasTitle; private final boolean hasTitle;
private int rowIndex; private int rowIndex;
public CellMergeStrategy(List<?> list, boolean hasTitle) { public CellMergeStrategy(List<?> list, boolean hasTitle) {
this.list = list;
this.hasTitle = hasTitle; this.hasTitle = hasTitle;
// 行合并开始下标 // 行合并开始下标
this.rowIndex = hasTitle ? 1 : 0; this.rowIndex = hasTitle ? 1 : 0;
this.cellList = handle(list, hasTitle);
} }
@Override @Override
protected void merge(Sheet sheet, Cell cell, Head head, Integer relativeRowIndex) { protected void merge(Sheet sheet, Cell cell, Head head, Integer relativeRowIndex) {
List<CellRangeAddress> cellList = handle(list, hasTitle); // judge the list is not null
// judge the list is not null if (CollUtil.isNotEmpty(cellList)) {
if (CollUtil.isNotEmpty(cellList)) { // the judge is necessary
// the judge is necessary if (cell.getRowIndex() == rowIndex && cell.getColumnIndex() == 0) {
if (cell.getRowIndex() == rowIndex && cell.getColumnIndex() == 0) { for (CellRangeAddress item : cellList) {
for (CellRangeAddress item : cellList) { sheet.addMergedRegion(item);
sheet.addMergedRegion(item); }
} }
} }
} }
}
@SneakyThrows @SneakyThrows
private List<CellRangeAddress> handle(List<?> list, boolean hasTitle) { private List<CellRangeAddress> handle(List<?> list, boolean hasTitle) {
List<CellRangeAddress> cellList = new ArrayList<>(); List<CellRangeAddress> cellList = new ArrayList<>();
if (CollUtil.isEmpty(list)) { if (CollUtil.isEmpty(list)) {
return cellList; return cellList;
} }
Field[] fields = ReflectUtils.getFields(list.get(0).getClass(), field -> !"serialVersionUID".equals(field.getName())); Field[] fields = ReflectUtils.getFields(list.get(0).getClass(), field -> !"serialVersionUID".equals(field.getName()));
// 有注解的字段 // 有注解的字段
List<Field> mergeFields = new ArrayList<>(); List<Field> mergeFields = new ArrayList<>();
List<Integer> mergeFieldsIndex = new ArrayList<>(); List<Integer> mergeFieldsIndex = new ArrayList<>();
for (int i = 0; i < fields.length; i++) { for (int i = 0; i < fields.length; i++) {
Field field = fields[i]; Field field = fields[i];
if (field.isAnnotationPresent(CellMerge.class)) { if (field.isAnnotationPresent(CellMerge.class)) {
CellMerge cm = field.getAnnotation(CellMerge.class); CellMerge cm = field.getAnnotation(CellMerge.class);
mergeFields.add(field); mergeFields.add(field);
mergeFieldsIndex.add(cm.index() == -1 ? i : cm.index()); mergeFieldsIndex.add(cm.index() == -1 ? i : cm.index());
if (hasTitle) { if (hasTitle) {
ExcelProperty property = field.getAnnotation(ExcelProperty.class); ExcelProperty property = field.getAnnotation(ExcelProperty.class);
rowIndex = Math.max(rowIndex, property.value().length); rowIndex = Math.max(rowIndex, property.value().length);
} }
} }
} }
Map<Field, RepeatCell> map = new HashMap<>(); Map<Field, RepeatCell> map = new HashMap<>();
// 生成两两合并单元格 // 生成两两合并单元格
for (int i = 0; i < list.size(); i++) { for (int i = 0; i < list.size(); i++) {
for (int j = 0; j < mergeFields.size(); j++) { for (int j = 0; j < mergeFields.size(); j++) {
Field field = mergeFields.get(j); Field field = mergeFields.get(j);
Object val = ReflectUtils.invokeGetter(list.get(i), field.getName()); Object val = ReflectUtils.invokeGetter(list.get(i), field.getName());
int colNum = mergeFieldsIndex.get(j); int colNum = mergeFieldsIndex.get(j);
if (!map.containsKey(field)) { if (!map.containsKey(field)) {
map.put(field, new RepeatCell(val, i)); map.put(field, new RepeatCell(val, i));
} else { } else {
RepeatCell repeatCell = map.get(field); RepeatCell repeatCell = map.get(field);
Object cellValue = repeatCell.getValue(); Object cellValue = repeatCell.getValue();
if (cellValue == null || "".equals(cellValue)) { if (cellValue == null || "".equals(cellValue)) {
// 空值跳过不合并 // 空值跳过不合并
continue; continue;
} }
if (!cellValue.equals(val)) { if (!cellValue.equals(val)) {
if (i - repeatCell.getCurrent() > 1) { if (i - repeatCell.getCurrent() > 1) {
cellList.add(new CellRangeAddress(repeatCell.getCurrent() + rowIndex, i + rowIndex - 1, colNum, colNum)); cellList.add(new CellRangeAddress(repeatCell.getCurrent() + rowIndex, i + rowIndex - 1, colNum, colNum));
} }
map.put(field, new RepeatCell(val, i)); map.put(field, new RepeatCell(val, i));
} else if (i == list.size() - 1) { } else if (i == list.size() - 1) {
if (i > repeatCell.getCurrent()) { if (i > repeatCell.getCurrent()) {
cellList.add(new CellRangeAddress(repeatCell.getCurrent() + rowIndex, i + rowIndex, colNum, colNum)); cellList.add(new CellRangeAddress(repeatCell.getCurrent() + rowIndex, i + rowIndex, colNum, colNum));
} }
} }
} }
} }
} }
return cellList; return cellList;
} }
@Data @Data
@AllArgsConstructor @AllArgsConstructor
static class RepeatCell { static class RepeatCell {
private Object value; private Object value;
private int current; private int current;
} }
} }