基于POI 封装导出工具类

基于POI 封装导出工具类

由于业务需求的变化 又懒得导出的时候创建dto等文件

所以封装了这套代码

v1.0

  • 解决导出因需求改变,样式变换问题.
  • 解决序号,表头,头文件存在与否,生成是否影响问题
  1. 导入依赖

    <!-- https://mvnrepository.com/artifact/org.apache.poi/poi -->
            <dependency>
                <groupId>org.apache.poi</groupId>
                <artifactId>poi</artifactId>
                <version>5.0.0</version>
            </dependency>
            <!-- https://mvnrepository.com/artifact/org.apache.poi/poi-ooxml -->
            <dependency>
                <groupId>org.apache.poi</groupId>
                <artifactId>poi-ooxml</artifactId>
                <version>5.0.0</version>
            </dependency>
            <dependency>
                <groupId>org.apache.poi</groupId>
                <artifactId>poi-scratchpad</artifactId>
                <version>5.0.0</version>
            </dependency>

2.封装工具类

package cn.xuxuweizhi.project.util;

import cn.xuxuweizhi.project.ExeclStyles.DefaultStyleByExecl;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.util.CellRangeAddress;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.List;

/**
 * @author xuyefang
 * @date 2021/7/20 16:59
 */
@Component
@Slf4j
public class ExcelUtils {
    /**
     * 头部样式
     */
    private final static String HEADER_STYLE = "getHeaderStyle";
    /**
     * 表头样式
     */
    private final static String TITLE_STYLE = "getTitleStyle";
    /**
     * 行_列样式
     */
    private final static String ROW_CELL_STYLE = "setRowOrCellStyle";

    /**
     * @param response    响应
     * @param titles      表头
     * @param data        数据  没有传null
     * @param sheetName   表单名
     * @param headerTitle 表头 没有就传null
     * @param excelName   导出的名称
     * @param cls         样式的类 没有重写就传null 使用默认的
     * @param openNumber  是否设置序号 即第一列是否要序号
     * @throws Exception 异常
     */
    public void exportExcel(HttpServletResponse response, Object[] titles, List<Object[]> data, String sheetName, String headerTitle, String excelName, Class<?> cls, boolean openNumber) throws Exception {
        if (cls == null) {
            //获取默认样式---后续可以改成通过枚举 获取对应的样式 从而达到多样式自由切换
            cls = DefaultStyleByExecl.class;
        }
        //反射获取类中的方法
        final Object o = cls.newInstance();
        //创建Excel工作薄对象
        HSSFWorkbook workbook = new HSSFWorkbook();
        //创建Excel工作表对象 // 生成一个表格
        HSSFSheet sheet = workbook.createSheet(sheetName);
        HSSFRow row = sheet.createRow(0);
        HSSFCell cell;
        //如果有大表头 则
        if (headerTitle != null && !headerTitle.trim().isEmpty()) {
            //有表头就设置在第一行 即 0行  此版本用sheet.getLastRowNum() 存在问题 所以直接写0
            cell = row.createCell(0);
            //表头样式 合并单元格 ---这里的合并是默认样式  特别需求是支持在头部样式中重写的
            if (openNumber) {
                sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, titles.length));
            } else {
                sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, titles.length - 1));
            }
            cell.setCellValue(headerTitle);
            //获取头部样式
            final CellStyle headerStyle = (CellStyle) getDeclareMethod(cls,HEADER_STYLE, HSSFWorkbook.class, HSSFSheet.class)
                    .invoke(o, workbook, sheet);
            cell.setCellStyle(headerStyle);
            //创建表头行
            row = sheet.createRow(sheet.getLastRowNum() + 1);
        }
        //-------------------------设置表头----------------------------------
        // 将第一列设为序号
        if (openNumber) {
            for (int i = 0; i <= titles.length; i++) {
                cell = row.createCell(i);
                if (i == 0) {
                    cell.setCellValue("序号");
                } else {
                    cell.setCellValue(titles[i - 1].toString());
                }
                //获取标题样式
                final CellStyle titleStyle = (CellStyle) getDeclareMethod(cls,TITLE_STYLE, HSSFWorkbook.class, HSSFSheet.class)
                        .invoke(o, workbook, sheet);
                cell.setCellStyle(titleStyle);
            }
        } else {
            for (int i = 0; i < titles.length; i++) {
                cell = row.createCell(i);
                cell.setCellValue(titles[i].toString());
                //获取标题样式
                final CellStyle titleStyle = (CellStyle) getDeclareMethod(cls,TITLE_STYLE, HSSFWorkbook.class, HSSFSheet.class)
                        .invoke(o, workbook, sheet);
                cell.setCellStyle(titleStyle);
            }
        }
        //--------------------------------------创建单元格--------------------------------------------------------
        int j = 1;
        if (data != null && data.size() > 0) {
            for (Object[] obj : data) {
                //创建行的单元格,从1开始
                row = sheet.createRow(sheet.getLastRowNum() + 1);
                if (openNumber) {
                    for (int i = 0; i <= obj.length; i++) {
                        cell = row.createCell(i);
                        // 将第一列设为序号
                        if (i == 0) {
                            cell.setCellValue(j++);
                        } else {
                            //由于设置了序号 所以-1
                            cell.setCellValue(obj[i - 1].toString());
                        }
                    }
                } else {
                    for (int i = 0; i < obj.length; i++) {
                        cell = row.createCell(i);
                        //由于设置了序号 所以-1
                        cell.setCellValue(obj[i].toString());
                    }
                }

            }
            //对列样式进行配置
            getDeclareMethod(cls,ROW_CELL_STYLE, HSSFWorkbook.class, HSSFSheet.class).invoke(o, workbook, sheet);
        }
        this.export(response, excelName, workbook);
    }

    /**
     * workbook 写入 response流 给前端
     *
     * @param response  响应
     * @param excelName 文件名
     * @param workbook  excel 对象
     * @throws IOException IO 异常
     */
    private void export(HttpServletResponse response, String excelName, HSSFWorkbook workbook) throws IOException {
        //文档输出
        response.setCharacterEncoding("utf-8");
        response.setContentType("application/msword");
        String fileName = excelName + ".xls";
        response.setHeader("Content-Disposition", "attachment;filename="
                .concat(String.valueOf(URLEncoder.encode(fileName, "UTF-8"))));
        try (OutputStream out = response.getOutputStream()) {
            workbook.write(out);
        } catch (IOException e) {
            log.error("导出发生异常----------------", e);
        }
    }
    
     /**
     * 根据属性名查找对应的属性,子类没有的属性会到父类找
     *
     * @param methodName 方法名
     * @return 方法
     */
    public static Method getDeclareMethod(Class<?> cls, String methodName, Class<?>... var2) {
        Method declaredMethod = null;
        for (; cls != Object.class; cls = cls.getSuperclass()) {
            try {
                declaredMethod = cls.getDeclaredMethod(methodName, var2);
            } catch (NoSuchMethodException e) {
                log.error("获取方法失败");
            }
        }
        return declaredMethod;
    }
}

封装样式类 --默认样式类

package cn.xuxuweizhi.project.ExeclStyles;

import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;

/**
 * 这个是excel样式 默认 excel 导出样式类
 * 如非特别需求 建议不要动这个
 * 如果样式需要修改 只需要重写这个类就行
 * 此版本支持POI 5.0
 * 样式设置可以参考官方文档:http://poi.apache.org/components/spreadsheet/quick-guide.html#FillsAndFrills
 *
 * @author xuyefang
 * @version v1.0
 * @date 2021/7/23 10:46
 */
public class DefaultStyleByExecl {
    /**
     * @param workbook workbook对象
     * @return 表头样式
     */
    public CellStyle getTitleStyle(HSSFWorkbook workbook, HSSFSheet sheet) {
        //灰色背景
        CellStyle style = workbook.createCellStyle();
        // Create a new font and alter it.
        Font font = workbook.createFont();
        font.setFontHeightInPoints((short) 10);
        //设置默认列宽
        sheet.setDefaultColumnWidth(20);
        font.setFontName("Courier New");
        // Fonts are set into a style so create a new one to use.
        style.setFont(font);
        //设置水平对齐的样式为居中对齐;
        style.setAlignment(HorizontalAlignment.CENTER);
        //设置垂直对齐的样式为居中对齐;
        style.setVerticalAlignment(VerticalAlignment.CENTER);
        return style;
    }

    /**
     * 头部样式
     *
     * @param workbook workbook对象
     * @return 头部样式
     */
    public CellStyle getHeaderStyle(HSSFWorkbook workbook, HSSFSheet sheet) {
        CellStyle style = workbook.createCellStyle();
        //设置水平对齐的样式为居中对齐;
        style.setAlignment(HorizontalAlignment.CENTER);
        //设置垂直对齐的样式为居中对齐;
        style.setVerticalAlignment(VerticalAlignment.CENTER);
        // Create a new font and alter it.
        Font font = workbook.createFont();
        font.setFontHeightInPoints((short) 15);
        font.setFontName("Courier New");
        // Fonts are set into a style so create a new one to use.
        style.setFont(font);
        return style;
    }


    /**
     * 单独对行,列进行样式设置
     */
    public void setRowOrCellStyle(HSSFWorkbook workbook, HSSFSheet sheet) {
        HSSFCellStyle cellStyle = workbook.createCellStyle();
        final int rowNum = sheet.getLastRowNum();
        //设置默认列宽
        sheet.setDefaultColumnWidth(20);
        for (int i = 0; i <= rowNum; i++) {
            final short lastCellNum = sheet.getRow(i).getLastCellNum();
            //这里对行样式进行设置
            for (int j = 0; j < lastCellNum; j++) {
                // 这里对列样式进行设置
                //自动换行
                cellStyle.setWrapText(true);
                //设置水平对齐的样式为居中对齐;
                cellStyle.setAlignment(HorizontalAlignment.CENTER);
                //设置垂直对齐的样式为居中对齐;
                cellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
                sheet.getRow(i).getCell(j).setCellStyle(cellStyle);
            }
        }
    }
    

}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值