使用POI导出EXCEL工具类并解决导出数据量大的问题

POI导出工具类

　　工作中常常会遇到一些图表需要导出的功能，在这里自己写了一个工具类方便以后使用（使用POI实现）。

项目依赖

       <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.6</version>
        </dependency>        

package com.adcc.eoss.util;

import org.apache.poi.hssf.usermodel.*;

import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;

/**
 * Created by LMQ on 2019/3/18.
 */
public class ExcelUtil {

    public static HSSFWorkbook getHSSFWorkbook(String sheetName, String[] title, String[][] values) {
        return getHSSFWorkbook(sheetName, title, values, null);
    }

    /**
     * 导出Excel
     *
     * @param sheetName sheet名称
     * @param title     标题
     * @param values    内容
     * @param wb        HSSFWorkbook对象
     * @return
     */
    public static HSSFWorkbook getHSSFWorkbook(String sheetName, String[] title, String[][] values, HSSFWorkbook wb) {

        // 第一步，创建一个HSSFWorkbook，对应一个Excel文件
        if (wb == null) {
            wb = new HSSFWorkbook();
        }

        // 第二步，在workbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet(sheetName);

        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制
        HSSFRow row = sheet.createRow(0);

        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式
        style.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER); // 创建一个垂直居中格式

        //声明列对象
        HSSFCell cell = null;

        //创建标题
        for (int i = 0; i < title.length; i++) {
            cell = row.createCell(i);
            cell.getCellStyle().setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
            cell.getCellStyle().setAlignment(HSSFCellStyle.ALIGN_CENTER);
            cell.setCellValue(title[i]);
            cell.setCellStyle(style);
        }

        //创建内容
        for (int i = 0; i < values.length; i++) {
            row = sheet.createRow(i + 1);
            for (int j = 0; j < values[i].length; j++) {
                //将内容按顺序赋给对应的列对象
                row.createCell(j).setCellValue(values[i][j]);
            }
        }
        return wb;
    }

    //响应到客户端
    public static void returnClient(HttpServletResponse response, String fileName, HSSFWorkbook wb) {
        try {
            setResponseHeader(response, fileName);
            OutputStream os = response.getOutputStream();
            wb.write(os);
            os.flush();
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    //发送响应流方法
    public static void setResponseHeader(HttpServletResponse response, String fileName) {
        try {
            try {
                fileName = new String(fileName.getBytes(), "ISO8859-1");
            } catch (UnsupportedEncodingException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
            response.setContentType("application/octet-stream;charset=ISO8859-1");
            response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
            response.addHeader("Pargam", "no-cache");
            response.addHeader("Cache-Control", "no-cache");
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }
}

通过这个工具类我们只需要传递对应参数就能够实现简单的EXCEL导出功能。

下面是一个导出的例子

 /**
     * 导出用户信息
     *
     * @param map
     * @param response
     */
    public void exportToExcel(Map<String, Object> map, HttpServletResponse response) throws Exception {
        // 查询得到用户数据
        List<UserVO> vos = findByConditionReturnExcel(map);
        // 定义表格头信息
        String[] titles = new String[]{"用户名称", "状态", "创建时间", "创建人", "最后登录时间", "所在公司"};
        // 为表格内容赋值
        String[][] values = new String[vos.size()][titles.length];
        if (vos != null && vos.size() > 0) {
            for (int i = 0; i < vos.size(); i++) {
                values[i][0] = vos.get(i).getUserName();
                values[i][1] = vos.get(i).getUserState();
                values[i][2] = vos.get(i).getCreateDate();
                values[i][3] = vos.get(i).getCreateUser();
                values[i][4] = vos.get(i).getLastLoginTime();
                values[i][5] = vos.get(i).getSourceCompanyName();
            }
        }
        // 导出
        ExcelUtil.returnClient(response, "user.xls", ExcelUtil.getHSSFWorkbook("用户管理", titles, values));
    }

HttpServletResponse 对象可在控制层通过参数传递进来即可。

那么问题来了，对于不是通过客户端浏览器的导出操作，我们就无法用 HttpServletResponse，其实这也是我在做一个统计工具的时候遇到的问题，
这个统计用的是图形界面开发SWT实现。那么我们就无法用这种客户端响应的方式实现导出。那么这个工具类不是就没用了？当时查了网上的资料，其实
只需更改响应方式即可。后来我用输出流修改了这个工具类。改动如下（因为只是个统计小工具也就没用到缓冲流之类的，只是简单做了输出）
fileName可定义具体的输出位置如： 'D:/统计/'+文件名

 //通过输出流响应
    public static void returnClient(String fileName, HSSFWorkbook wb) throws Exception {
        FileOutputStream os = null;
        try {
            File file = new File(fileName);
            os = new FileOutputStream(file);
            wb.write(os);
            os.flush();
            os.close();
        } catch (Exception e) {
            throw e;
        } finally {
            if (os != null) {
                try {
                    // 关闭流
                    os.close();
                } catch (IOException e) {
                    throw e;
                }
            }
        }
    }

导出数据量大的解决办法

使用poi导出excel的时候如果数据过多，超过65535条会报错，因为excel2003一个sheet表最多导出65535条，excel2007是10万4000多条限制。

Invalid row number (65536) outside allowable range (0..65535)
解决方案：一个sheet最多可以导出65535条，我们可以分成多个sheet导出。
当时我写了一个特别low的方法，但是也算实现的功能。下面分享一下我的办法
当时是这样想的，比如从数据库查询出一百万条数据，我把这数据分成多份，用同一个HSSFWorkbook对象，执行工具类中getHSSFWorkbook方法不就行了
于是就有了以下实现（主要是懒得改工具类-_-）

 /**
     * 将一个集合拆分成多个
     *
     * @param list 需要拆分的集合
     * @param num  每个集合的数量
     * @return
     */
    public Map<String, List<CDM>> splitList(List<CDM> list, Integer num) {

        // list 长度
        int listSize = list.size();
        // 用map来存放集合
        HashMap<String, List<CDM>> listHashMap = new HashMap<String, List<CDM>>();
        // 单个集合对象
        List<CDM> childList = new ArrayList<>();
        // 遍历要拆分集合按定义的num数量存放到childList
        for (int i = 0; i < listSize; i++) {
            childList.add(list.get(i));
            if (((i + 1) % num == 0) || (i + 1 == listSize)) {
                // 存入map
                listHashMap.put(String.valueOf(i), childList);
                childList = new ArrayList<>();
            }
        }
        return listHashMap;
    }

以上是拆分集合的方法，通过这个方法即可实现集合拆分，之后在业务代码这样写就行了
wb为HSSFWorkbook对象，我们在代码里把它在循环外声明就行 HSSFWorkbook wb = new HSSFWorkbook();

 if (list.size() > 65535) {
　　　　　　　　　　// 每个小集合放60000万个，个数可以自己定义
                Map<String, List<CDM>> childListMap = splitList(list, 60000);
　　　　　　　　　　// 循环map,循环为wb添加sheet wb也就是我们的HSSFWorkbook对象
                for (Map.Entry<String, List<CDM>> entry : childListMap.entrySet()) {
                    String[][] values = new String[entry.getValue().size()][titles.length];
                    for (int i = 0; i < entry.getValue().size(); i++) {
                        values[i][0] = entry.getValue().get(i).getId();
                        values[i][1] = entry.getValue().get(i).getName();
　　                    //....
　　　　　　　　　　　　}
                    wb = ExcelUtil.getHSSFWorkbook("统计" + entry.getKey(), titles, values, wb);
                }

这样就可以支持大数据量导出了。