若依Excel导出单元格合并功能拓展

zigeweiwu

已于 2023-07-20 15:57:39 修改

阅读量758

点赞数

分类专栏： RouYi 文章标签： excel spring spring boot

于 2023-07-19 12:08:04 首次发布

本文链接：https://blog.csdn.net/zigeweeiwu/article/details/131805356

版权

这篇博客介绍了如何在若依框架中扩展Excel导出功能，特别是单元格合并的实现。通过在@Excel注解中添加新方法，并创建单元格合并工具类，实现了与原有ExcelUtil类似的使用方式，展示了导出数据的合并效果。主要处理单元格合并的代码可自定义，以满足特定需求。

摘要由CSDN通过智能技术生成

修改 `@Excel`注解

在@Excel 中添加 mergeRows() 方法

在这里插入图片描述

在注解中使用添加的方法

在这里插入图片描述

添加单元格合并工具类代码如下

package com.szst.common.utils.poi;

import com.szst.common.annotation.Excel;
import com.szst.common.annotation.Excel.ColumnType;
import com.szst.common.annotation.Excel.Type;
import com.szst.common.annotation.Excels;
import com.szst.common.config.SzstConfig;
import com.szst.common.core.domain.AjaxResult;
import com.szst.common.core.text.Convert;
import com.szst.common.exception.ServiceException;
import com.szst.common.exception.UtilException;
import com.szst.common.utils.DateUtils;
import com.szst.common.utils.DictUtils;
import com.szst.common.utils.StringUtils;
import com.szst.common.utils.file.FileTypeUtils;
import com.szst.common.utils.file.ImageUtils;
import com.szst.common.utils.reflect.ReflectUtils;
import lombok.Getter;
import lombok.Setter;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.ss.util.CellRangeAddressList;
import org.apache.poi.util.IOUtils;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import org.apache.poi.xssf.usermodel.XSSFClientAnchor;
import org.apache.poi.xssf.usermodel.XSSFDataValidation;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.lang.reflect.Field;
import java.math.BigDecimal;
import java.text.DecimalFormat;
import java.util.*;
import java.util.stream.Collectors;


/**
 * Excel相关处理
 *
 * @author ruoyi
 */
public class ExcelUtilMerge<T> {
   
    private static final Logger log = LoggerFactory.getLogger(ExcelUtilMerge.class);

    /**
     * Excel sheet最大行数，默认65536
     */
    public static final int sheetSize = 65536;

    /**
     * 工作表名称
     */
    private String sheetName;

    /**
     * 导出类型（EXPORT:导出数据；IMPORT：导入模板）
     */
    private Type type;

    /**
     * 工作薄对象
     */
    private Workbook wb;

    /**
     * 工作表对象
     */
    private Sheet sheet;

    /**
     * 样式列表
     */
    private Map<String, CellStyle> styles;

    /**
     * 导入导出数据列表
     */
    private List<T> list;

    /**
     * 注解列表
     */
    private List<Object[]> fields;

    /**
     * 最大高度
     */
    private short maxHeight;

    /**
     * 统计列表
     */
    private Map<Integer, Double> statistics = new HashMap<Integer, Double>();

    /**
     * 合并单元格数值记录
     */
    private Map<Integer, mergeNumP> mapNumber = new HashMap<>(16);
    /**
     * 数字格式
     */
    private static final DecimalFormat DOUBLE_FORMAT = new DecimalFormat("######0.00");

    /**
     * 实体对象
     */
    public Class<T> clazz;

    public ExcelUtilMerge(Class<T> clazz) {
   
        this.clazz = clazz;
    }

    public void init(List<T> list, String sheetName, Type type) {
   
        if (list == null) {
   
            list = new ArrayList<T>();
        }
        this.list = list;
        this.sheetName = sheetName;
        this.type = type;
        createExcelField();
        createWorkbook();
    }

    /**
     * 对excel表单默认第一个索引名转换成list
     *
     * @param is 输入流
     * @return 转换后集合
     */
    public List<T> importExcel(InputStream is) throws Exception {
   
        return importExcel(StringUtils.EMPTY, is);
    }

    /**
     * 对excel表单指定表格索引名转换成list
     *
     * @param sheetName 表格索引名
     * @param is        输入流
     * @return 转换后集合
     */
    public List<T> importExcel(String sheetName, InputStream is) throws Exception {
   
        this.type = Type.IMPORT;
        this.wb = WorkbookFactory.create(is);
        List<T> list = new ArrayList<T>();
        Sheet sheet = null;
        if (StringUtils.isNotEmpty(sheetName)) {
   
            // 如果指定sheet名,则取指定sheet中的内容.
            sheet = wb.getSheet(sheetName);
        } else {
   
            // 如果传入的sheet名不存在则默认指向第1个sheet.
            sheet = wb.getSheetAt(0);
        }

        if (sheet == null) {
   
            throw new IOException("文件sheet不存在");
        }

        int rows = sheet.getPhysicalNumberOfRows();

        if (rows > 0) {
   
            // 定义一个map用于存放excel列的序号和field.
            Map<String, Integer> cellMap = new HashMap<String, Integer>();
            // 获取表头
            Row heard = sheet.getRow(0);
            for (int i = 0; i < heard.getPhysicalNumberOfCells(); i++) {
   
                Cell cell = heard.getCell(i);
                if (StringUtils.isNotNull(cell)) {
   
                    String value = this.getCellValue(heard, i).toString();
                    cellMap.put(value, i);
                } else {
   
                    cellMap.put(null, i);
                }
            }
            // 有数据时才处理 得到类的所有field.
            Field[] allFields = clazz.getDeclaredFields();
            // 定义一个map用于存放列的序号和field.
            Map<Integer, Field> fieldsMap = new HashMap<Integer, Field>();
            for (int col = 0; col < allFields.length; col++) {
   
                Field field = allFields[col];
                Excel attr = field.getAnnotation(Excel.class);
                if (attr != null && (attr.type() == Type.ALL || attr.type() == type)) {
   
                    // 设置类的私有字段属性可访问.
                    field.setAccessible(true);
                    Integer column = cellMap.get(attr.name());
                    if (column != null) {
   
                        fieldsMap.put(column, field);
                    }
                }
            }
            for (int i = 1; i < rows; i++) {
   
                // 从第2行开始取数据,默认第一行是表头.
                Row row = sheet.getRow(i);
                if (row == null) {
   
                    continue;
                }
                T entity = null;
                for (Map.Entry<Integer, Field> entry : fieldsMap.entrySet()) {
   
                    Object val = this.getCellValue(row, entry.getKey());

                    // 如果不存在实例则新建.
                    entity = (entity == null ? clazz.newInstance() : entity);
                    // 从map中得到对应列的field.
                    Field field = fieldsMap.get(entry.getKey());
                    // 取得类型,并根据对象类型设置值.
                    Class<?> fieldType = field.getType();
                    if (String.class == fieldType) {
   
                        String s = Convert.toStr(val);
                        if (StringUtils.endsWith(s, ".0")) {
   
                            val = StringUtils.substringBefore(s, ".0");
                        } else {
   
                            String dateFormat = field.getAnnotation(Excel.class).dateFormat();
                            if (StringUtils.isNotEmpty(dateFormat)) {
   
                                val = DateUtils.parseDateToStr(dateFormat, (Date