基于POI 封装导出工具类
由于业务需求的变化 又懒得导出的时候创建dto等文件
所以封装了这套代码
v1.0
- 解决导出因需求改变,样式变换问题.
- 解决序号,表头,头文件存在与否,生成是否影响问题
-
导入依赖
<!-- 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 liulangzhe
* @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 liulangzhe
* @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);
if (sheet.getRow(i)!=null) {
if (sheet.getRow(i).getCell(j) != null) {
sheet.getRow(i).getCell(j).setCellStyle(cellStyle);
}
}
}
}
}
}
v1.1
> 由于业务需求 后面需要对指定列进行合并 更新后如下
package com.iop.operation.util;
import com.iop.common.enums.ApiStatusCode;
import com.iop.common.exception.ResultException;
import com.iop.operation.config.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.http.MediaType;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.lang.reflect.Method;
import java.net.URLEncoder;
import java.util.*;
/**
* @author liulangzhe
* @version 1.1
* @date 2021/7/20 16:59
*/
@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 static 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);
}
export(response, excelName, workbook);
}
/**
* @param response 响应
* @param titles 表头 允许为null
* @param data 数据
* @param excelName 导出的名称
* @param mergeFields 需要合并的字段 会根据相同值进行合并 所以处理的数据必须有序 允许为null
*/
public static void exportExcelForMap(HttpServletResponse response, Map<String, String> titles, List<Map<String, Object>> data, String excelName, String[] mergeFields) {
try {
exportExcelForMap(response, titles, data, null, null, excelName, null, mergeFields, false);
} catch (Exception e) {
log.error("导出失败,错误原因:", e);
throw new ResultException(ApiStatusCode.VAILD_ERROR.getCode(), "导出失败,请联系管理员");
}
}
/**
* @param response 响应
* @param titles 表头 允许为null
* @param data 数据 允许为null
* @param sheetName 表单名 不传默认sheet
* @param headerTitle 表头 没有就传null
* @param excelName 导出的名称
* @param cls 样式的类 没有重写就传null 使用默认的
* @param mergeFields 需要合并的字段 会根据相同值进行合并 所以处理的数据必须有序 允许为null
* @param openNumber 是否设置序号 即第一列是否要序号
* @throws Exception
*/
public static void exportExcelForMap(HttpServletResponse response, Map<String, String> titles, List<Map<String, Object>> data, String sheetName, String headerTitle, String excelName, Class<?> cls, String[] mergeFields, 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 == null ? "sheet" : sheetName);
HSSFRow row = sheet.createRow(0);
HSSFCell cell;
// 用于记录表头的索引 合并单元格会根据这个定位列序号
LinkedList<String> titleIndex = new LinkedList<>();
// 如果有大表头 则
if (headerTitle != null && !headerTitle.trim().isEmpty()) {
// 有表头就设置在第一行 即 0行 此版本用sheet.getLastRowNum() 存在问题 所以直接写0
cell = row.createCell(0);
// 表头样式 合并单元格 ---这里的合并是默认样式 特别需求是支持在头部样式中重写的
if (openNumber) {
sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, titles.size()));
} else {
sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, titles.size() - 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);
}
// 设置表头
int cellNumber = 0;
if (openNumber) {
cell = row.createCell(cellNumber);
cell.setCellValue("序号");
titleIndex.add(cellNumber, "序号");
cellNumber++;
}
Set<String> keys = titles.keySet();
// 2、第二步:遍历每个键,根据键提取值
for (String key : keys) {
Object value = titles.get(key);
cell = row.createCell(cellNumber);
cell.setCellValue(value.toString());
titleIndex.add(cellNumber, key);
cellNumber++;
// 获取标题样式
final CellStyle titleStyle = (CellStyle) getDeclareMethod(cls, TITLE_STYLE, HSSFWorkbook.class, HSSFSheet.class)
.invoke(o, workbook, sheet);
cell.setCellStyle(titleStyle);
}
// 创建单元格
if (data != null && data.size() > 0) {
// 排序号
int sequenceNumber = 1;
for (Map<String, Object> map : data) {
// 创建行的单元格,从1开始
row = sheet.createRow(sheet.getLastRowNum() + 1);
int cellIndex = 0;
if (openNumber) {
cell = row.createCell(cellIndex);
cell.setCellValue(sequenceNumber++);
cellIndex++;
}
// 2、第二步:遍历每个键,根据键提取值 根据title 去赋值
for (String key : keys) {
cell = row.createCell(cellIndex++);
Object value = map.get(key);
// 设置常用的数据类型
if (value instanceof String) {
cell.setCellValue(value.toString());
} else if (value instanceof Double) {
cell.setCellValue((double) value);
} else if (value instanceof Integer) {
cell.setCellValue((Integer) value);
} else if (value instanceof Date) {
cell.setCellValue((Date) value);
} else {
cell.setCellValue(Objects.toString(value,""));
}
}
}
// 合并列
mergeColumn(mergeFields, sheet, titleIndex);
// 对列样式进行配置
getDeclareMethod(cls, ROW_CELL_STYLE, HSSFWorkbook.class, HSSFSheet.class).invoke(o, workbook, sheet);
}
export(response, excelName, workbook);
}
/**
* 合并列
*
* @param mergeFields 需要合并的字段
* @param sheet 表格
* @param titleIndex 列表索引
*/
private static void mergeColumn(String[] mergeFields, HSSFSheet sheet, LinkedList<String> titleIndex) {
// 对指定的字段进行合并
if (mergeFields != null) {
// 用于记录已经合并的区域
Set<CellRangeAddress> mergedRegions = new HashSet<>();
// 对指定字段进行处理合并 合并列
for (String field : mergeFields) {
// 当前字段在excel中的位置 即列的位置
int startIndex = titleIndex.indexOf(field);
// 根据列找对应相同的行 将其合并
Integer startRow = null;
Integer endRow = null;
for (int i1 = 1; i1 < sheet.getLastRowNum(); i1++) {
// 第一行
HSSFCell cell1 = sheet.getRow(i1).getCell(startIndex);
if (startRow == null) {
// 开始的行号只进行一次赋值
startRow = i1;
}
// 下一行
HSSFCell cell2 = sheet.getRow(i1 + 1).getCell(startIndex);
if (cell1 != null && cell2 != null && cell1.getStringCellValue().equals(cell2.getStringCellValue())) {
// 如果列的值相等 name代表是可以合并的
endRow = i1 + 1;
} else {
// 如果遇到不相等的情况,先进行合并操作,然后重置startRow和endRow
if (endRow != null) {
CellRangeAddress region = new CellRangeAddress(startRow, endRow, startIndex, startIndex);
if (isOverlapping(region, mergedRegions)) {
sheet.addMergedRegion(region);
mergedRegions.add(region);
}
}
// 重置startRow和endRow
startRow = i1 + 1;
endRow = null;
}
}
// 最后一组相同的单元格需要在循环结束后处理
if (startRow != null && endRow != null) {
CellRangeAddress region = new CellRangeAddress(startRow, endRow, startIndex, startIndex);
if (isOverlapping(region, mergedRegions)) {
sheet.addMergedRegion(region);
mergedRegions.add(region);
}
}
}
}
}
/**
* 检查新合并区域是否与已存在的合并区域重叠 重叠则不允许合并
*
* @param region 新合并区域
* @param mergedRegions 合并区域集合
* @return boolean true 不重叠 false 重叠
*/
private static boolean isOverlapping(CellRangeAddress region, Set<CellRangeAddress> mergedRegions) {
for (CellRangeAddress existingRegion : mergedRegions) {
if (existingRegion.intersects(region)) {
return false;
}
}
return true;
}
/**
* workbook 写入 response流 给前端
*
* @param response 响应
* @param excelName 文件名
* @param workbook excel 对象
* @throws IOException IO 异常
*/
public static void export(HttpServletResponse response, String excelName, HSSFWorkbook workbook) throws IOException {
// 文档输出
response.setCharacterEncoding("utf-8");
// response.setContentType("application/msword");
response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);
String fileName = excelName + ".xlsx";
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("获取方法失败", e);
}
}
return declaredMethod;
}
}
注意:本文归作者所有,未经作者允许,不得转载