由需求而产生的一款db导出excel的工具

代码地址如下：
http://www.demodashi.com/demo/12062.html

程序员最大的毛病可能就是懒,因为懒所以做出了许许多多提高自己工作效率的工具.
起因于我是商业开发,既然是商业项目避免不了各种数据统计,虽然公司有专门的数据平台,但是应对一些临时性需求还是免不了开发人员去导出一些数据.每一次有需求来我都是写一个复杂的sql,然后放到DataGrip中执行,利用其功能导出cvs,然而越来越多的需求该功能无法满足,比如导出组合表,也就是一个excel中有多个sheet表.那么应该这个需求我写了一个为自己的工具.

我理想中的工具

1.简单模式使用sql查询直接导出
2.复杂模式可以定义一些复杂的bean,然后通过组合代码中自定义实现导出逻辑
3.可以自己定义表头,以及对应的数据处理,比如把时间戳转换为yyy-MM-dd hh:MM:ss这样的形式
4.支持一个excel中含有多个sheet
5.不需要很复杂的配置,因为自用,所以能约定俗成的地方就约定俗成.

语言的选择

这个很随意了,我是选择自己最熟悉的语言,也就是Java.
同事听说我用Java写这种工具,强烈推荐我用py,但天生动态语言无感,可以说是反感,所以放弃.

实现

环境: JDK8 Maven
IDE: IntelliJ IDEA / Eclipse

具体过程很简单,代码逻辑也很清晰,这里只说下主要流程,详细的可以参考源码Github地址,另外由于个人使用,所以没有太多的校验和异常考虑.

数据的获取

数据的获取其实就是DBUtils的使用方法,很简单执行Sql,然后使用对应的Handler转换成对应形式参数

//查询出结果,使用Map存储,当然dbUtils也支持bean存储final List<Map<String, Object>> result = queryRunner.query(conn, "select * from user", new MapListHandler());

ResultSetHandler接口提供了一个单独的方法：Object handle(java.sql.ResultSet.rs)。
ResultSetHandler接口的实现类
1. ArrayHandler：把结果集中的第一行数据转换成对象数组。
2. ArrayListHandler：把结果集中的每一行数据都转换成一个对象数组，再存放到List中。
3. BeanHandler：将结果集中的第一行数据封装到一个对应的JavaBean实例中。
4. BeanListHandler：将结果集中的每一行数据都封装到一个对应的JavaBean实例中，存放到List里。
5. MapHandler：将结果集中的第一行数据封装到一个Map里，key是列名，value就是对应的值。
6. MapListHandler：将结果集中的每一行数据都封装到一个Map里，然后再存放到List。
7. ColumnListHandler：将结果集中某一列的数据存放到List中。
8. KeyedHandler(name)：将结果集中的每一行数据都封装到一个Map里(List)，再把这些map再存到一个map里，其key为指定的列。
9. ……

表头的定义

表头是有序的,所以使用LinkedHashMap<String, ExcelHeader> header这种数据结构,其中ExcelHeader是我封装的一个类,如下,可以看到其功能是存储对应的表头展示名称,以及转换器

@Data
public class ExcelHeader {/*** 展示名称*/private String displayHeader;/*** 对应数据转换器*/private Function convert;private ExcelHeader(String displayHeader,Function convert){this.displayHeader = displayHeader;this.convert = convert;}public static ExcelHeader create(String displayHeader){return new ExcelHeader(displayHeader, Function.identity());}public static ExcelHeader create(String displayHeader,Function convert){return new ExcelHeader(displayHeader, convert);}
}

那么其使用就很简单了

   //表格对应的表头,以及该表头的数据处理LinkedHashMap<String, ExcelHeader> header = new LinkedHashMap<>();header.put("id", ExcelHeader.create("用户id"));header.put("username", ExcelHeader.create("用户名"));header.put("email", ExcelHeader.create("用户邮箱"));header.put("avatar", ExcelHeader.create("用户头像"));//对于日期使用转换器,转换器为java8的Function函数实现header.put("last_login_date", ExcelHeader.create("用户上次登录时间", FuncitionConvertUtil.date2String));header.put("status", ExcelHeader.create("用户id"));header.put("role", ExcelHeader.create("用户id"));//对于日期使用转换器,转换器为java8的Function函数实现header.put("gmt_create", ExcelHeader.create("用户id",FuncitionConvertUtil.date2String));

导出表

数据以及表头都有了,那么只剩下的导出操作也就是数据的组合.

    //导出表/*** 查出来的map直接导出表格*/ExcelExportUtil.fromMap(result).displayHeader(header).excelType(ExcelExportUtil.ExcelFileType.XLS).build("用户表").writeTo("/tmp/test1.xls");/*** 导出多张表*/ExcelExportUtil.fromMap(result).displayHeader(header).excelType(ExcelExportUtil.ExcelFileType.XLSX).build("用户表1").andFormMap(result).displayHeader(header).build("用户表2").writeTo("/tmp/test2.xlsx");

核心的导出逻辑为build()方法

  @SuppressWarnings("unchecked")public ExcelExportUtil build(String sheetName) {buildReady();//创建表,如果已存在excelBook,那么新增表Sheet sheet = sheetName == null ? workbook.createSheet() : workbook.createSheet(sheetName);//写表头int rowNum = 0;Row headerRow = sheet.createRow(rowNum++);int[] tempCol = {0};this.headerAndConvert.forEach((k,v) -> headerRow.createCell(tempCol[0]++).setCellValue(v.getDisplayHeader()));//写表数据for (Map<String, Object> colData : this.mapData) {Row row = sheet.createRow(rowNum++);tempCol[0] = 0;this.headerAndConvert.forEach((k,v) -> {Cell cell = row.createCell(tempCol[0]++);Object value = v.getConvert().apply(colData.get(k));convertObjValue(cell, value);});}return this;}

导出表格展示

test1.xls

test2.xlsx

下载Demo如何运行?

环境具备后,那么只需要对应的代码导出IDE,然后查看QueryAndExportByH2这个单元测试类,执行里面的testQueryAndExport()方法即可,注意导出表的时候需要指定路径名

项目结构截图

总结

本文的主要目的是表达迷茫的时候不知道自己该做什么,那么就从自己身边的需求开始,分析自己所遇到的痛点,然后用你喜欢的方式去解决这个痛点,那么这个过程就是你的进步.

最后给出github地址,如果你有能力则直接看源代码即可,如果你从此网站下载了Demo,那么有问题可以随时与我交流,邮件为 niudear@foxmail.com

由需求而产生的一款db导出excel的工具

代码地址如下：
http://www.demodashi.com/demo/12062.html

注：本文著作权归作者，由demo大师发表，拒绝转载，转载需要作者授权