简单几个注解,分分钟搞定 Excel 数据导出(大大简化开发)

Posted Java知音_

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了简单几个注解,分分钟搞定 Excel 数据导出(大大简化开发)相关的知识,希望对你有一定的参考价值。

1. 概览

数据导出是日常开发的常见功能,及将数据导出为Excel并提供下载。 Java 生态存在大量的 Excel 操作类库,基于这些类库便可完成相关功能。这样,大量繁杂、无意义的代码耗费着宝贵的人力资源,大家距离 996 又近了几分,那有没有更优解呢?

1.1. 背景

刚刚接到产品需求,准备为众多“列表”功能增加下载支持,简单来说就是:新增一个“导出”按钮,点击时,将列表数据导入到 Excel 并进行下载。

需求很简单,功能也很明确,接到需求后,小伙伴们着手准备:

  1. 使用 poi 可以将数据写入 Excel,但 poi 的 api 过于复杂,存在一定的学习成本;

  2. 工作量巨大,涉及较多的列表页面,每个列表的字段数量也不少;

  3. 期望列表展示和Excel 数据同源,发生变更时只需要修改一次,列表和Excel能同时生效;

不知道你怎么想,面对这些枯燥的体力工作,我极为反感。

1.2. 目标

“懒” 是人类进步的“原动力”。手懒心不懒,让脑力解放体力!!!

设计目标如下:

  1. 屏蔽复杂的 API,降低使用门槛;

  2. 使用声明式替代编码,提升开发效率;

  3. 为一般场景提供通用能力,方便快速的具备以下能力:

  • 自定义标题

  • 自定义顺序

  • 支持嵌套对象

  • 支持自定义格式

  • 支持自定义样式

为特殊场景提供定制能力,以满足个性化需求;

2. 快速入门

2.1. 引入 starter 以及相关依赖

首先,在 pom 文件中增加 excelasbean starter 以及 poi 依赖。

<dependency>
 <groupId>com.geekhalo.lego</groupId>
 <artifactId>lego-starter-excelasbean</artifactId>
 <version>0.0.1-SNAPSHOT</version>
</dependency>
<dependency>
 <groupId>org.apache.poi</groupId>
 <artifactId>poi</artifactId>
</dependency>
2.2. 使用 ExcelAsBeanService 完成 excel 导出

其次,将 ExcelAsBeanService 注入到自己的 Sevice 中,使用 write 等一系列方法完成 数据写入。

示例代码如下:

// 注入 excelAsBeanService
// 由 ExcelAsBeanAutoConfiguration 完成 ExcelAsBeanService 的注册
@Autowired
private ExcelAsBeanService excelAsBeanService;

public <D extends User> HSSFWorkbook downloadUser(Class<D> cls, Supplier<D> supplier)
    // 1. 准备数据
    List<D> users = createUser(100, supplier);
    // 2. 创建 Workbook
    HSSFWorkbook hssfWorkbook = new HSSFWorkbook();
    // 3. 向 Sheet 中写入数据
    this.excelAsBeanService.writHeaderAndDataToSheet(hssfWorkbook,"User", cls, users);
    // 4. 返回 workbook
    return hssfWorkbook;

ExcelAsBeanService 中涉及写操作包括:

方法名含义
writHeaderAndDataToSheet一次性写入 标题 和 数据
writDataToSheet仅写入数据,主要用于数据追加操作
2.3. 在 JavaBean 上增加相关注解

最后,只要在 JavaBean 增加相关注解,便可以导出 Excel。

从简单到复杂,框架提供了多种定制能力,具体如下:

2.3.1. 简单导出

最简单的方式,只需增加 @HSSFHeader 注解,并指定好 Header 的标题即可。

示例代码如下:

@Data
public class UserV1 implements User
    @HSSFHeader(title = "编号")
    private Long id;

    @HSSFHeader(title = "姓名")
    private String name;

    @HSSFHeader(title = "生日")
    private Date birthAt;

    @HSSFHeader(title = "年龄")
    private Integer age;

运行效果如下:

备注:@HSSFHeader 具有标记作用,JavaBean 中没有 @HSSFHeader 标记的字段或方法不会进行导出。

2.3.2. 自定义列顺序

默认情况下,Excel 中的列与 JavaBean 属性定义顺序相关;如有需要,可自定义列的顺序。

增加自定义列顺序,相关代码如下:

@Data
public class UserV2 implements User
    @HSSFHeader(title = "编号", order = 1)
    private Long id;

    @HSSFHeader(title = "姓名", order = 2)
    private String name;

    @HSSFHeader(title = "生日", order = 4)
    private Date birthAt;

    @HSSFHeader(title = "年龄")
    @HSSFShowOrder(3)
    private Integer age;

可以通过 @HSSFHeader 的 order 指定顺序,也可以使用 @HSSFShowOrder 进行指定,具体效果如下:

见图,年龄 和 生日 的顺序发生变化。

2.3.3. 导出嵌入对象

当 JavaBean 属性为另一个对象时,我们需要将关联对象也一并导出。

首先,定义关联对象,示例如下:

@Data
@Builder
public class Address 
    private Long id;
    @HSSFHeader(title = "省")
    private String l1;
    @HSSFHeader(title = "市")
    private String l2;
    @HSSFHeader(title = "区")
    private String l3;
    @HSSFHeader(title = "详细地址")
    private String l4;

关联对象只是在 Address 上增加了 @HSSFHeader 注解。

其次,使用 @HSSFEmbedded 对嵌入对象进行标记。示例如下:

@Data
public class UserV3 implements User
    @HSSFHeader(title = "编号", order = 1)
    private Long id;

    @HSSFHeader(title = "姓名", order = 2)
    private String name;

    @HSSFHeader(title = "生日", order = 4)
    private Date birthAt;

    @HSSFHeader(title = "年龄", order = 3)
    private Integer age;

    @HSSFEmbedded
    @HSSFShowOrder(5)
    private Address address;

运行效果如下:

如图,Address 内部属性已经完全展开并写入到 Excel。

2.3.4. 自定义格式化方法

当直接对属性进行展示不能满足需求时,可以通过自定义方法对输出内容进行格式化。

比如,需要对 Address 对象中的内容进行自定义展示,示例如下:

@Data
public class UserV4 implements User
    @HSSFHeader(title = "编号", order = 1)
    private Long id;

    @HSSFHeader(title = "姓名", order = 2)
    private String name;

    @HSSFHeader(title = "生日", order = 4)
    private Date birthAt;

    @HSSFHeader(title = "年龄", order = 3)
    private Integer age;

    @HSSFEmbedded
    @HSSFShowOrder(5)
    private Address address;

    @HSSFHeader(title = "详细地址", order = 6)
    public String showAddress()
        if (this.address == null)
            return "暂无地址";
        
        return new StringBuilder()
                .append(this.address.getL1()).append(", ")
                .append(this.address.getL2()).append(", ")
                .append(this.address.getL3()).append(", ")
                .append(this.address.getL4())
                .toString();
    

首先,将展示规则封装到方法中,如 showAddress()然后,在方法上增加 @HSSFHeader 注解。

运行效果如下:

如图,Excel 中新增 “详细地址” 列。

2.3.5. 添加样式

如果觉得展示比较枯燥,可以自定义 Header 和 Data 的显示样式。

在类上指定全局 Header 样式,示例如下:

@Data
@HSSFHeaderStyle("header")
public class UserV5 implements User
    @HSSFHeader(title = "编号", order = 1)
    private Long id;

    @HSSFHeader(title = "姓名", order = 2)
    private String name;

    @HSSFHeader(title = "生日", order = 4)
    private Date birthAt;

    @HSSFHeader(title = "年龄", order = 3)
    private Integer age;

    @HSSFEmbedded
    @HSSFShowOrder(5)
    private Address address;

    @HSSFHeader(title = "详细地址", order = 6)
    public String showAddress()
        if (this.address == null)
            return "暂无地址";
        
        return new StringBuilder()
                .append(this.address.getL1()).append(", ")
                .append(this.address.getL2()).append(", ")
                .append(this.address.getL3()).append(", ")
                .append(this.address.getL4())
                .toString();
    

在 UserV5 类上增加 @HSSFHeaderStyle("header"),指定表头使用 header 样式。

运行效果如下:

如图,表头的样式发生了变化。

备注:自定义样式,需要提供 HSSFCellStyleFactory 实现,详见 “设计&扩展” 部分

3. 设计&扩展

3.1. 核心设计

3.1.1. 整体设计

整体设计如下:

按照层级结构,将核心组件拆分为:

  1. 每一个 class 通过解析后,获得一个与之对于的 HSSFSheetWriter

  2. 每一个 HSSFSheetWriter 包含一个 HSSFRowWriter,完成“写标题”和“写数据”操作;

  3. 每一个 HSSFRowWriter 包含若干个 HSSFColumnWriter 对象,共同完成每一行的写;

  4. 每一个 HSSFColumnWriter 包含两个 HSSFCellWriterChain,负责写标题 和 数据;

  5. 每一个 HSSFCellWriterChain 包含 HSSFValueSupplierHSSFCellConfiguratorHSSFCellWriter 共同完成一个 Cell 的写操作;

其中,操作的核心逻辑基本在 HSSFCellWriterChain 中,其他结构主要完成组件组装。

3.1.1. CellWriterChain 设计

CellWriterChain 是写入流程的核心,整体设计如下:

从写流程来看,包括:

  1. HSSFValueSupplier 从 JavaBean 中获取数据,包括从注解中获取标题信息,从字段读取数据,执行方法获取数据等;

  2. HSSFCellConfigurator 对 Cell 进行配置,比如设置自适应宽度、设置显示样式等;

  3. HSSFCellWriter 将数据写入到 Cell 中,包括简单数据 String、Long、Double 等,也包括自定义 Date 格式等;

从装配流程来看,包括:

  1. Spring 容器中注册有大量的 Factory,包括HSSFHeaderValueSupplierFactoryHSSFDataValueSupplierFactoryHSSFHeaderCellConfiguratorFactoryHSSFDataCellConfiguratorFactoryHSSFHeaderCellWriterFactoryHSSFDataCellWriterFactory

  2. 将众 Factory 注入到对于的 Factories,由Factories 对外提供统一入口。Factories 包括 HSSFValueSupplierFactoriesHSSFCellConfiguratorFactoriesHSSFCellWriterFactories 等;

  3. DefaultHSSFCellWriterChainFactory 从 Factories 中获取组件,将其封装为 HSSFCellWriterChain

  4. 最后,由 HSSFCellWriterChain 完成数据写入流程;

3.2. 功能扩展

为了应对个性化需求,框架提供了众多扩展点,其中最重要的扩展点包括,HSSFValueSupplierHSSFCellWriterHSSFCellConfigurator

三者协助组成 HSSFCellWriterChain,完成数据写入操作。

3.2.1. HSSFValueSupplier

从 Bean 中提取数据。

HSSFValueSupplier 定义如下:

@FunctionalInterface
public interface HSSFValueSupplier<T, R> extends Function<T, R> 

直接继承自 Function,输入为 JavaBean 对象,输出为待写入数据。

目前,系统实现如下:

系统实现功能
FixValueSupplier提供固定值作为写入数据,比如在 HSSFHeader 配置的 title 信息
FieldBasedDataValueSupplier从 field 中读取数据
MethodBasedValueSupplier执行 method 获取数据

备注:一般情况下无需进行扩展

3.2.2. HSSFCellConfigurator

在执行写入操作前,对 Cell 进行配置。

HSSFCellConfigurator 定义如下:

public interface HSSFCellConfigurator 
    /**
     * 对 Cell 进行配置
     * @param context
     * @param columnIndex
     * @param cell
     */
    void configFor(HSSFCellWriterContext context, int columnIndex, HSSFCell cell);

目前,系统中实现包括:

系统实现功能
AutoSizeCellConfigurator为列设置自适应大小
HSSFCellStyleConfigurator为列设置显示样式

如需对样式进行定制,需实现 HSSFCellStyleFactory 并根据 name 返回不同的样式。实例代码如下:

@Component
public class DemoHSSFCellStyleFactory implements HSSFCellStyleFactory 
    @Override
    public HSSFCellStyle createStyle(HSSFCellWriterContext context, String name) 
        if ("header".equalsIgnoreCase(name))
            return createForHeader(context.getWorkbook());
        
        if ("data".equalsIgnoreCase(name))
            return createForData(context.getWorkbook());
        
        return null;
    

    private HSSFCellStyle createForData(HSSFWorkbook workbook) 
        // 定义 data 展示样式
        return style;
    

    private HSSFCellStyle createForHeader(HSSFWorkbook workbook) 
        // 定义 header 展示样式
        return style;
    

3.2.3. HSSFCellWriter

负责向 Cell 中写入数据,比如对数据进行格式化。

HSSFCellWriter 接口定义如下:

public interface HSSFCellWriter<D> 
    /**
     * 向 Cell 写入数据
     * @param context
     * @param cell 待写入的 Cell
     * @param data 待写入的 data
     */
    void write(HSSFCellWriterContext context, HSSFCell cell, D data);

目前,系统中的实现包括:

实现类功能
DefaultHSSFCellWriter默认实现,主要用于处理简单属性
DateFormatHSSFCellWriter对 Date、LocalDate、LocalDateTime 等进行自定义格式化

备注:如需复杂的转换,可以通过扩展 HSSFCellWriter 的方式实现。

4. 项目信息

项目仓库地址:

  • https://gitee.com/litao851025/lego

项目文档地址:

  • https://gitee.com/litao851025/lego/wikis/support/ExcelAsBean-数据导出

推荐

Java面试题宝典

技术内卷群,一起来学习!!

PS:因为公众号平台更改了推送规则,如果不想错过内容,记得读完点一下“在看”,加个“星标”,这样每次新文章推送才会第一时间出现在你的订阅列表里。点“在看”支持我们吧!

以上是关于简单几个注解,分分钟搞定 Excel 数据导出(大大简化开发)的主要内容,如果未能解决你的问题,请参考以下文章

配置简单功能强大的excel工具类搞定excel导入导出工具类

【springboot+easypoi】一行代码搞定excel导入导出

2.自定义@Excel注解实现数据Excel形式导入导出

java 导出 excel 最佳实践,大文件 excel 避免OOM(内存溢出) 框架-02-API

zookeeper之场景与架构-《每日五分钟搞定大数据》

zookeeper-1.操作与应用场景-《每日五分钟搞定大数据》