EasyExcel 導出合并層級單元格
一、案例
案例一
- 1.相同訂單號單元格進行合并
合并結果
案例二
- 1.相同訂單號的單元格進行合并
- 2.相同訂單號的總數和總金額進行合并
合并結果
案例三
- 1.相同訂單號的單元格進行合并
- 2.相同訂單號的商品分類進行合并
- 3.相同訂單號的總數和總金額進行合并
- 4.相同訂單號和相同商品分類的分類總數、分類總金額進行合并
合并結果
二、代碼實現
相關依賴
<dependency><groupId>com.alibaba</groupId><artifactId>easyexcel</artifactId><version>3.2.1</version>
</dependency><dependency><groupId>cn.hutool</groupId><artifactId>hutool-all</artifactId><version>5.8.16</version>
</dependency><dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId><version>1.18.26</version>
</dependency>
2.1 AbstractMergeStrategy
import com.alibaba.excel.write.handler.CellWriteHandler;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.Sheet;
public abstract class AbstractMergeStrategy implements CellWriteHandler {/*** 最大行索引*/public final static int EXCEL_LAST_INDEX = 1048575;/*** 默認合并起始行*/public final static int DEFAULT_START_ROW_INDEX = 1;/*** 合并抽象方法* @param sheet* @param cell*/public abstract void merge(Sheet sheet, Cell cell);/*** 獲取單元格值* @param cell */public Object getCellValue(Cell cell) {return cell.getCellType() == CellType.STRING ? cell.getStringCellValue() : cell.getNumericCellValue();}
}
2.2 ColumnMergeStrategy
import cn.hutool.core.collection.CollUtil;
import com.alibaba.excel.metadata.Head;
import com.alibaba.excel.metadata.data.WriteCellData;
import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
import com.alibaba.excel.write.metadata.holder.WriteTableHolder;
import com.easy.excel.demo.model.MergeRowColumn;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.util.CellRangeAddress;import java.util.*;
import java.util.stream.Collectors;/*** 合并單元格策略:適用于列合并*/
@Slf4j
public class ColumnMergeStrategy extends AbstractMergeStrategy {/*** 合并起始行索引*/private int mergeStartRowIndex;/*** 合并結束行索引*/private int mergeEndRowIndex;/*** 待合并的列(如果沒有指定,則所有的列都會進行合并)*/private List<Integer> mergeColumnIndexList;/*** 待合并的列父級依賴關系 <需要合并的列索引, 依賴的父級列索引>* key 需要合并的列, value 所依賴的父級列的列表 */private Map<Integer, List<Integer>> mergeColumnIndexMap = new HashMap<>();/*** 合并的行列索引數據(存儲每列的數據合并的行列索引范圍)*/private Map<Integer, MergeRowColumn> mergeRowColumnMap = new HashMap<>();private Sheet sheet;public ColumnMergeStrategy() {this(DEFAULT_START_ROW_INDEX, EXCEL_LAST_INDEX);}public ColumnMergeStrategy(List<Integer> mergeColumnIndexList) {this(DEFAULT_START_ROW_INDEX, EXCEL_LAST_INDEX, mergeColumnIndexList);}public ColumnMergeStrategy(Map<Integer, List<Integer>> mergeColumnIndexMap) {this.mergeColumnIndexMap = mergeColumnIndexMap;this.mergeColumnIndexList = mergeColumnIndexMap.keySet().stream().collect(Collectors.toList());this.mergeStartRowIndex = DEFAULT_START_ROW_INDEX;this.mergeEndRowIndex = EXCEL_LAST_INDEX;}public ColumnMergeStrategy(int mergeStartRowIndex) {this(mergeStartRowIndex, EXCEL_LAST_INDEX);}public ColumnMergeStrategy(int mergeStartRowIndex, int mergeEndRowIndex) {this(mergeStartRowIndex, mergeEndRowIndex, new ArrayList<>());}public ColumnMergeStrategy(int mergeStartRowIndex, int mergeEndRowIndex, List<Integer> mergeColumnIndexList) {this.mergeStartRowIndex = mergeStartRowIndex;this.mergeEndRowIndex = mergeEndRowIndex;this.mergeColumnIndexList = mergeColumnIndexList;}@Overridepublic void afterCellDispose(WriteSheetHolder writeSheetHolder, WriteTableHolder writeTableHolder, List<WriteCellData<?>> list, Cell cell, Head head, Integer integer, Boolean isHead) {// 頭不參與合并if (isHead) {return;}// 初始化 sheetif (sheet == null) {this.sheet = writeSheetHolder.getSheet();}// 如果當前行大于合并起始行則進行合并if (cell.getRowIndex() >= mergeStartRowIndex && cell.getRowIndex() <= mergeEndRowIndex) {// 判斷是否是全列合并或者當前列在需要合并列中if (CollUtil.isEmpty(mergeColumnIndexList) || (CollUtil.isNotEmpty(mergeColumnIndexList) && mergeColumnIndexList.contains(cell.getColumnIndex()))) {// 合并單元格this.merge(writeSheetHolder.getSheet(), cell);}}}@Overridepublic void merge(Sheet sheet, Cell cell) {// 當前單元格行、列索引int curRowIndex = cell.getRowIndex();int curColumnIndex = cell.getColumnIndex();// 當前單元格的值為Object curCellValue = this.getCellValue(cell);// 上一行的行索引int aboveRowIndex = curRowIndex - 1;if (aboveRowIndex < 0 || aboveRowIndex < mergeStartRowIndex) {// 初始化當前列的 合并區域范圍MergeRowColumn mergeRowColumn = new MergeRowColumn(curRowIndex, curRowIndex, curColumnIndex, curColumnIndex);mergeRowColumnMap.put(curColumnIndex, mergeRowColumn);return;}// 獲取上一個單元格Cell aboveCell = sheet.getRow(aboveRowIndex).getCell(curColumnIndex);// 上一個單元格的值Object aboveCellValue = this.getCellValue(aboveCell);// 判斷上一個單元格是否能合并if (Objects.equals(curCellValue, aboveCellValue)) {boolean needMerge = true;// 父級列 列表List<Integer> parentColumnIndexList = mergeColumnIndexMap.get(curColumnIndex);if (parentColumnIndexList != null && !parentColumnIndexList.isEmpty()) {for (Integer parentColumnIndex : parentColumnIndexList) {Cell mainCell = sheet.getRow(curRowIndex).getCell(parentColumnIndex);Cell aboveMainCell = sheet.getRow(aboveRowIndex).getCell(parentColumnIndex);Object mainCellValue = this.getCellValue(mainCell);Object aboveMainCellValue = this.getCellValue(aboveMainCell);// 所有主列都需要滿足合并條件才能合并副列if (!Objects.equals(mainCellValue, aboveMainCellValue)) {needMerge = false;break;}}}// 允許合并if (needMerge){// 修改當前列的行合并索引范圍MergeRowColumn mergeRowColumn = mergeRowColumnMap.get(curColumnIndex);mergeRowColumn.setEndRowIndex(curRowIndex);} else {// 合并已有的單元格,修改行索引指向mergeRowColumnCell(sheet, curRowIndex,curColumnIndex);}} else {// 合并已有的單元格,修改行索引指向mergeRowColumnCell(sheet, curRowIndex,curColumnIndex);}}/*** 檢查給定的單元格是否在一個或多個合并區域中。** @return 如果指定單元格是合并區域的一部分,則返回 true;否則返回 false。*/private boolean isMergedRegion(Sheet sheet, Integer rowIndex, Integer columnIndex) {// 獲取當前工作表中的所有合并區域數量int numMergedRegions = sheet.getNumMergedRegions();// 遍歷所有合并區域for (int i = 0; i < numMergedRegions; i++) {CellRangeAddress region = sheet.getMergedRegion(i);// 檢查指定的單元格是否在當前合并區域內if (region.isInRange(rowIndex, columnIndex)) {return true;}}return false;}/*** 合并區域單元格** @param sheet* @param curRowIndex* @param curColumnIndex*/private void mergeRowColumnCell(Sheet sheet, Integer curRowIndex, Integer curColumnIndex) {// 獲取當前的列的合并區域索引對象MergeRowColumn mergeRowColumn = mergeRowColumnMap.get(curColumnIndex);// 合并單元格mergeCell(sheet, mergeRowColumn, curRowIndex, curColumnIndex);}/*** 手動合并最后的單元格* (最后一段單元格需要手動合并)*/public void finalMergeCell() {// 遍歷所有列的合并索引,合并最后的單元格for (Map.Entry<Integer, MergeRowColumn> entry : mergeRowColumnMap.entrySet()) {Integer columnIndex = entry.getKey();MergeRowColumn mergeRowColumn = entry.getValue();Integer endRowIndex = mergeRowColumn.getEndRowIndex();mergeCell(sheet, mergeRowColumn, endRowIndex, columnIndex);}}/*** 合并單元格** @param sheet* @param mergeRowColumn* @param curRowIndex* @param curColumnIndex*/private void mergeCell(Sheet sheet, MergeRowColumn mergeRowColumn,Integer curRowIndex, Integer curColumnIndex) {// 獲取合并的行起始索引Integer startRowIndex = mergeRowColumn.getStartRowIndex();// 獲取合并的行結束索引Integer endRowIndex = mergeRowColumn.getEndRowIndex();// 合并單元格(至少有兩個單元格以上才能進行合并)if (startRowIndex < endRowIndex) {CellRangeAddress cellAddresses = new CellRangeAddress(startRowIndex, endRowIndex, curColumnIndex, curColumnIndex);// 判斷起始單元格是否已經合并過了boolean mergedRegion = isMergedRegion(sheet, startRowIndex, curColumnIndex);if (!mergedRegion) {// 合并指定區域的單元格sheet.addMergedRegion(cellAddresses);}}// 重置合并索引(當前列的行指針下移)mergeRowColumn.setStartRowIndex(curRowIndex);mergeRowColumn.setEndRowIndex(curRowIndex);}
}
源碼分析:
- mergeRowColumnMap 是進行合并的關鍵所在,存儲了所有需要合并的列的行合并區域索引,在遍歷數據過程中,根據情況進行單元格合并然后偏移指針,或者只修改指針,這樣就不需要頻繁的進行合并操作。
2.3 MergeRowColumn
@NoArgsConstructor
@AllArgsConstructor
@Data
public class MergeRowColumn {/*** 開始行索引*/private Integer startRowIndex;/*** 結束行索引*/private Integer endRowIndex;/*** 開始列索引*/private Integer startColumnIndex;/*** 結束列索引*/private Integer endColumnIndex;
}
三、測試
案例測試代碼
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.util.DateUtils;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.easy.excel.demo.handler.ColumnMergeStrategy;
import com.easy.excel.demo.model.OrderDetailEntity;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;import java.io.File;
import java.math.BigDecimal;
import java.util.*;
import java.util.stream.Collectors;@Slf4j
@SpringBootTest
public class EasyExcelDemoTest2 {/*** 案例一*/@Testpublic void testMerge1(){File file = new File("order1.xlsx");WriteSheet writeSheet = EasyExcel.writerSheet("sheet1").head(OrderDetailEntity.class).registerWriteHandler(new ColumnMergeStrategy(Arrays.asList(0)))// 訂單號.build();ExcelWriter excelWriter = EasyExcel.write(file).build();// 寫入數據excelWriter.write(data(), writeSheet);// 手動合并最后的單元格(最后的單元格沒有辦法合并,需要手動進行合并)writeSheet.getCustomWriteHandlerList().stream().filter(handler -> handler instanceof ColumnMergeStrategy).map(handler -> (ColumnMergeStrategy) handler).forEach(handler -> handler.finalMergeCell());excelWriter.finish();}/*** 案例二*/@Testpublic void testMerge2(){// 輸出文件路徑File file = new File("order2.xlsx");// 初始化列合并列父級依賴關系Map<Integer, List<Integer>> mergeColumnIndexMap = new HashMap<>();mergeColumnIndexMap.put(0, new ArrayList<>());//訂單號mergeColumnIndexMap.put(10, Arrays.asList(0));//總數 ==> 訂單號mergeColumnIndexMap.put(11, Arrays.asList(0));//總金額 ==> 訂單號WriteSheet writeSheet = EasyExcel.writerSheet("sheet1").head(OrderDetailEntity.class).registerWriteHandler(new ColumnMergeStrategy(mergeColumnIndexMap)).build();ExcelWriter excelWriter = EasyExcel.write(file).build();excelWriter.write(data(), writeSheet);// 手動合并最后的單元格(分段數據注入時最后的單元格沒有辦法合并,需要手動進行合并)writeSheet.getCustomWriteHandlerList().stream().filter(handler -> handler instanceof ColumnMergeStrategy).map(handler -> (ColumnMergeStrategy) handler).forEach(handler -> handler.finalMergeCell());excelWriter.finish();}/*** 案例三*/@Testpublic void testMerge3(){// 輸出文件路徑File file = new File("order3.xlsx");// 初始化列合并上級依賴關系Map<Integer, List<Integer>> mergeColumnIndexMap = new HashMap<>();mergeColumnIndexMap.put(0, new ArrayList<>());//訂單號mergeColumnIndexMap.put(2, Arrays.asList(0));//商品分類 ==> 訂單號mergeColumnIndexMap.put(8, Arrays.asList(0, 2));//分類總數 ==> 訂單號,商品分類mergeColumnIndexMap.put(9, Arrays.asList(0, 2));//分類總金額 ==> 訂單號,商品分類mergeColumnIndexMap.put(10, Arrays.asList(0));//總數 ==> 訂單號mergeColumnIndexMap.put(11, Arrays.asList(0));//總金額 ==> 訂單號WriteSheet writeSheet = EasyExcel.writerSheet("sheet1").head(OrderDetailEntity.class).registerWriteHandler(new ColumnMergeStrategy(mergeColumnIndexMap)).build();ExcelWriter excelWriter = EasyExcel.write(file).build();// 模擬分頁查詢數據for (int i = 0; i < 3; i++) {excelWriter.write(data(), writeSheet);}// 手動合并最后的單元格(分段數據注入時最后的單元格沒有辦法合并,需要手動進行合并)writeSheet.getCustomWriteHandlerList().stream().filter(handler -> handler instanceof ColumnMergeStrategy).map(handler -> (ColumnMergeStrategy) handler).forEach(handler -> handler.finalMergeCell());excelWriter.finish();}/*** 隨機生成測試數據* @return*/private Collection<?> data() {Map<String, List<String>> productMap = getProductMap();List<String> statusList = Arrays.asList("待發貨", "已發貨", "運輸中", "待取貨", "已完成");List<OrderDetailEntity> dataList = new ArrayList<>();Random random = new Random();int orderCount = random.nextInt(2) + 5;for (int i = 0; i < orderCount; i++) {String orderCode = "PL" + DateUtils.format(new Date(), "yyyyMMddHHmm") + "000" + i;int orderDetailCount = random.nextInt(10) + 1;List<OrderDetailEntity> detailEntities = new ArrayList<>();Map<String, BigDecimal> categoryTotalQuantityMap = new HashMap<>();Map<String, BigDecimal> categoryTotalPriceMap = new HashMap<>();BigDecimal totalQuantity = BigDecimal.ZERO;BigDecimal totalPrice = BigDecimal.ZERO;for (int j = 0; j < orderDetailCount; j++) {String orderDetailCode = UUID.randomUUID().toString();String productCategory = new ArrayList<String>(productMap.keySet()).get(random.nextInt(productMap.size()));List<String> productList = productMap.get(productCategory);String productCode = "SKU" + (random.nextInt(1000)+1000);String productName = productList.get(random.nextInt(productList.size())) + "-A" + random.nextInt(50);BigDecimal price = new BigDecimal(random.nextInt(2000) + 800);BigDecimal quantity = new BigDecimal(random.nextInt(5) + 1);String status = statusList.get(random.nextInt(statusList.size()));String key = orderCode + "-" + productCategory;BigDecimal categoryTotalQuantity = categoryTotalQuantityMap.get(key);if (categoryTotalQuantity == null) {categoryTotalQuantity = quantity;} else {categoryTotalQuantity = categoryTotalQuantity.add(quantity);}categoryTotalQuantityMap.put(key, categoryTotalQuantity);BigDecimal categoryTotalPrice = categoryTotalPriceMap.get(key);if (categoryTotalPrice == null) {categoryTotalPrice = price.multiply(quantity);} else {categoryTotalPrice = categoryTotalPrice.add(price.multiply(quantity));}categoryTotalPriceMap.put(key, categoryTotalPrice);totalQuantity = totalQuantity.add(quantity);totalPrice = totalPrice.add(price.multiply(quantity));detailEntities.add(OrderDetailEntity.builder().orderCode(orderCode).orderDetailCode(orderDetailCode).productCategory(productCategory).productCode(productCode).productName(productName).price(price).quantity(quantity).status(status).build());}for (OrderDetailEntity item : detailEntities) {String key = item.getOrderCode() + "-" + item.getProductCategory();item.setCategoryTotalQuantity(categoryTotalQuantityMap.get(key));item.setCategoryTotalPrice(categoryTotalPriceMap.get(key));item.setTotalQuantity(totalQuantity);item.setTotalPrice(totalPrice);}detailEntities = detailEntities.stream().sorted(Comparator.comparing(OrderDetailEntity::getOrderCode).thenComparing(OrderDetailEntity::getProductCategory)).collect(Collectors.toList());dataList.addAll(detailEntities);}return dataList;}private Map<String, List<String>> getProductMap() {Map<String, List<String>> productMap = new HashMap<>();// 家電List<String> householdList = new ArrayList<>();householdList.add("電視機");householdList.add("冰箱");householdList.add("洗衣機");householdList.add("空調");productMap.put("家電", householdList);// 數碼產品List<String> digitalList = new ArrayList<>();digitalList.add("手機");digitalList.add("攝影機");digitalList.add("電腦");digitalList.add("照相機");digitalList.add("投影儀");digitalList.add("智能手表");productMap.put("數碼產品", digitalList);// 健身器材List<String> gymEquipmentList = new ArrayList<>();gymEquipmentList.add("動感單車");gymEquipmentList.add("健身椅");gymEquipmentList.add("跑步機");productMap.put("健身器材", gymEquipmentList);return productMap;}
}
OrderDetailEntity
import com.alibaba.excel.annotation.ExcelProperty;
import com.alibaba.excel.annotation.write.style.*;
import com.alibaba.excel.enums.poi.BorderStyleEnum;
import com.alibaba.excel.enums.poi.FillPatternTypeEnum;
import com.alibaba.excel.enums.poi.HorizontalAlignmentEnum;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;import java.math.BigDecimal;@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
// 頭背景設置
@HeadStyle(fillPatternType = FillPatternTypeEnum.SOLID_FOREGROUND, horizontalAlignment = HorizontalAlignmentEnum.CENTER, borderLeft = BorderStyleEnum.THIN, borderTop = BorderStyleEnum.THIN, borderRight = BorderStyleEnum.THIN, borderBottom = BorderStyleEnum.THIN)
//標題高度
@HeadRowHeight(30)
//內容高度
@ContentRowHeight(20)
//內容居中,左、上、右、下的邊框顯示
@ContentStyle(horizontalAlignment = HorizontalAlignmentEnum.CENTER, borderLeft = BorderStyleEnum.THIN, borderTop = BorderStyleEnum.THIN, borderRight = BorderStyleEnum.THIN, borderBottom = BorderStyleEnum.THIN)
public class OrderDetailEntity {@ExcelProperty(value = "訂單號")@ColumnWidth(25)private String orderCode;@ExcelProperty(value = "訂單明細")@ColumnWidth(40)private String orderDetailCode;@ExcelProperty(value = "商品分類")@ColumnWidth(20)private String productCategory;@ExcelProperty(value = "商品編碼")@ColumnWidth(20)private String productCode;@ExcelProperty(value = "商品名稱")@ColumnWidth(20)private String productName;@ExcelProperty(value = "單價")@ColumnWidth(10)private BigDecimal price;@ExcelProperty(value = "數量")@ColumnWidth(10)private BigDecimal quantity;@ExcelProperty(value = "狀態")@ColumnWidth(10)private String status;@ExcelProperty(value = "分類總數")@ColumnWidth(20)private BigDecimal categoryTotalQuantity;@ExcelProperty(value = "分類總金額")@ColumnWidth(20)private BigDecimal categoryTotalPrice;@ExcelProperty(value = "總數")@ColumnWidth(10)private BigDecimal totalQuantity;@ExcelProperty(value = "總金額")@ColumnWidth(10)private BigDecimal totalPrice;
}
參考文章
https://blog.csdn.net/xhmico/article/details/141814528