JAVA导出数据到excel中大数据量的解决方法

web项目中需要有将数据导出excel的操作需求。曾经尝试过几种方法：

1、poi导出到excel

他是操作对象然后将结果写到excel中。

2、拼excel的html形式文件，将文件扩展名改为xls

这中方式是利用excel能够另存为html格式文件，文件包含excel的头信息。

以前用过的方法，没有考虑的问题

1、使用poi等导出时，没有考虑数据量的问题，大数据量无法满足，有个几千行jvm就哭了。

2、后改为拼html文件时，没有考虑excel中sheet的最大行数65536，超过这个行数就excel无法打开了。

综合考虑，并从网上找了些其他人的想法，最后用如下方案

继续沿用拼excel的html形式文件的方法，但在导出时，根据最大行数重新建立新的excel文件；

数据从数据库取出使用纯jdbc方式读数据，边读边向文件中写；

待所有数据写完，将各个小excel文件进行打包，输出到输出流中。

定义导出接口

import java.io.OutputStream;
import java.util.Collection;

/**
 * 大数据量导出
 *
 * @author lisen
 *
 */
public interface ExportBigData {
    /**
     * 导出sql查询出的数据,输出为压缩后的格式
     *
     * @param os
     *            输出流
     * @param maxRow
     *            每个excel的最大行数
     * @param sql
     *            查询语句
     * @param sqlParams
     *            查询参数
     */
    public void exportToZip(final OutputStream os, final int maxRow,
            final String sql, final Object... sqlParams);

    /**
     * 导出sql查询出的数据,输出为压缩后的格式
     *
     * @param titles
     *            标题
     * @param os
     *            输出流
     * @param maxRow
     *            每个excel的最大行数
     * @param sql
     *            查询语句
     * @param sqlParams
     *            查询参数
     */
    public void exportToZip(final Collection<String> titles,
            final OutputStream os, final int maxRow, final String sql,
            final Object... sqlParams);

    public void exportToZip(final String[] titles, final OutputStream os,
            final int maxRow, final String sql, final Object... sqlParams);

    public void exportToZip(final String[] titles, final OutputStream os,
            final String sql, final Object... sqlParams);

    public void exportToZip(final Collection<String> titles,
            final OutputStream os, final String sql, final Object... sqlParams);

    public void exportToZip(final OutputStream os, final String sql,
            final Object... sqlParams);
}

大数据量导出抽象实现

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.springframework.dao.DataAccessException;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.ResultSetExtractor;
import org.springframework.jdbc.support.JdbcUtils;

import com.cnaec.common.Constant;

/**
 * 大数据量导出 抽象类
 *
 * @author lisen
 * @date 2013-10-30 上午10:40:33
 */
public abstract class AbstractExportBigData implements ExportBigData {
    /**
     * 每个文件的最大行数 超过请求按默认算
     */
    public static final int MAXROWS = 50000;
    /**
     * 用于数据查询
     */
    private JdbcTemplate jdbcTemplate;

    /**
     * 构造函数
     *
     * @param jdbcTemplate
     */
    public AbstractExportBigData(JdbcTemplate jdbcTemplate) {
        this.jdbcTemplate = jdbcTemplate;
    }

    /**
     * 临时文件前缀
     *
     * @return
     */
    abstract protected String getPrefix();

    /**
     * 临时文件后缀
     *
     * @return
     */
    abstract protected String getSuffix();

    /**
     * 删除临时文件
     *
     * @param fileList
     */
    protected void cleanTempFile(List<File> fileList) {
        for (File file : fileList) {
            System.out.println("-----删除文件," + file.getPath());
            file.delete();
        }
    }

    /**
     * 创建临时文件
     *
     * @return
     * @throws IOException
     */
    protected File createTempFile() throws IOException {
        return File.createTempFile(getPrefix(), getSuffix());
    }

    /**
     * 数据输出
     *
     * @param data
     * @param fos
     * @throws IOException
     */
    protected void writeToOutputStream(String data, FileOutputStream fos)
            throws IOException {
        IOUtils.write(data, fos, Constant.ENCODING);
    }

    /**
     * 文件开头的写入
     *
     * @param fos
     * @throws IOException
     */
    abstract protected void writeHeaderToOutputStream(FileOutputStream fos)
            throws IOException;

    /**
     * 文件结尾的写入
     *
     * @param fos
     */
    abstract protected void writeFooterToOutputStream(FileOutputStream fos)
            throws IOException;

    /**
     * 一行数据的写入
     *
     * @param rs
     * @param fos
     * @throws SQLException
     * @throws IOException
     */
    abstract protected void writeOneRowToOutputStream(ResultSet rs,
            FileOutputStream fos) throws SQLException, IOException;

    /**
     * 写数据标题
     *
     * @param titles
     * @param fos
     * @throws IOException
     */
    abstract protected void writeTitleToOutputStream(Collection<String> titles,
            FileOutputStream fos) throws IOException;

    /**
     * 打包 压缩成zip
     *
     * @param os
     *            压缩输出流
     * @param fileList
     *            被压缩的文件列表
     * @throws IOException
     */
    protected void doZip(OutputStream os, List<File> fileList)
            throws IOException {
        if (fileList != null && fileList.size() > 0) {
            byte[] buf = new byte[1024];
            ZipOutputStream out = new ZipOutputStream(os);
            for (File file : fileList) {
                FileInputStream in = new FileInputStream(file);
                out.putNextEntry(new ZipEntry(file.getName()));
                int len;
                while ((len = in.read(buf)) > 0) {
                    out.write(buf, 0, len);
                }
                out.closeEntry();
                in.close();
            }
            System.out.println(os == null);
            out.close();
            System.out.println(os == null);
        }
    }

    /**
     * 获取单个文件最大行数
     *
     * @param maxRow
     * @return
     */
    protected int getMaxRows(int maxRow) {
        return maxRow < MAXROWS ? maxRow : MAXROWS;
    }

    protected String getColumnKey(String columnName) {
        return columnName;
    }

    protected Object getColumnValue(ResultSet rs, int index)
            throws SQLException {
        return JdbcUtils.getResultSetValue(rs, index);
    }

    @Override
    public void exportToZip(final Collection<String> titles,
            final OutputStream os, final int maxRow, final String sql,
            final Object... sqlParams) {
        // 每个文件最大行数
        final int max = getMaxRows(maxRow);
        jdbcTemplate.query(sql, sqlParams, new ResultSetExtractor() {
            @Override
            public Object extractData(ResultSet rs) throws SQLException,
                    DataAccessException {
                // 文件收集器
                List<File> fileList = new ArrayList<File>();
                // 行数记录器
                int i = 0;
                // 临时文件
                File file = null;
                FileOutputStream fos = null;
                try {
                    while (rs.next()) {
                        // 达到最大行数 或者 新建的 创建新文件
                        if (i == max || i == 0) {
                            // 如果不是新文件 为这个文件写入文件尾
                            if (file != null) {
                                // 写文件尾
                                writeFooterToOutputStream(fos);
                                // 关闭流
                                IOUtils.closeQuietly(fos);
                            }
                            // 创建临时文件
                            file = createTempFile();
                            // 打开流
                            fos = FileUtils.openOutputStream(file);
                            // 放进收集器里
                            fileList.add(file);
                            // 写文件头
                            writeHeaderToOutputStream(fos);
                            // 数据区标题栏
                            writeTitleToOutputStream(titles, fos);
                            i = 0;
                        }
                        i++;
                        // 写实际一行数据
                        writeOneRowToOutputStream(rs, fos);
                    }

                    if (file != null) {
                        // 写文件尾
                        writeFooterToOutputStream(fos);
                        // 关闭流
                        IOUtils.closeQuietly(fos);
                    }
                    // 打包
                    doZip(os, fileList);
                } catch (IOException e) {
                    // io异常
                } finally {
                    IOUtils.closeQuietly(fos);
                    // 清空临时文件
                    cleanTempFile(fileList);
                    fileList.clear();
                    fileList = null;
                }
                return null;
            }
        });
    }

    @Override
    public void exportToZip(final OutputStream os, int maxRow, String sql,
            Object... sqlParams) {
        exportToZip(Collections.EMPTY_LIST, os, maxRow, sql, sqlParams);
    }

    @Override
    public void exportToZip(final String[] titles, OutputStream os, int maxRow,
            String sql, Object... sqlParams) {
        exportToZip(Arrays.asList(titles), os, maxRow, sql, sqlParams);
    }

    @Override
    public void exportToZip(final String[] titles, final OutputStream os,
            final String sql, final Object... sqlParams) {
        exportToZip(titles, os, MAXROWS, sql, sqlParams);
    }

    @Override
    public void exportToZip(final OutputStream os, final String sql,
            final Object... sqlParams) {
        exportToZip(os, MAXROWS, sql, sqlParams);
    }

    @Override
    public void exportToZip(final Collection<String> titles,
            final OutputStream os, final String sql, final Object... sqlParams) {
        exportToZip(titles, os, MAXROWS, sql, sqlParams);
    }
}

EXCEL导出的实现

import java.io.FileOutputStream;
import java.io.IOException;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.Collection;
import java.util.List;

import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.web.util.HtmlUtils;

import com.avicinfo.v2.common.export.bigdata.AbstractExportBigData;
import com.avicinfo.v2.common.export.bigdata.ExportBigData;

/**
 * 用于excel大数据量导出
 *
 * @author lisen
 * @date 2013-10-30 上午10:41:36
 */
public class BigDataToExcelImpl extends AbstractExportBigData implements
        ExportBigData {

    StringBuffer headStr = new StringBuffer(
            "<html xmlns:x="urn:schemas-microsoft-com:office:excel">")
            .append("<head>")
            .append(
                    "<meta http-equiv="content-type" content="application/ms-excel; charset=UTF-8"/>")
            .append("<!--[if gte mso 9]><xml>").append("<x:ExcelWorkbook>")
            .append("<x:ExcelWorksheets>").append("<x:ExcelWorksheet>").append(
                    "<x:Name></x:Name>").append("<x:WorksheetOptions>").append(
                    "<x:Print>").append("<x:ValidPrinterInfo />").append(
                    "</x:Print>").append("</x:WorksheetOptions>").append(
                    "</x:ExcelWorksheet>").append("</x:ExcelWorksheets>")
            .append("</x:ExcelWorkbook>").append("</xml><![endif]-->").append(
                    "</head>").append("<body>").append("<table>");

    StringBuffer footStr = new StringBuffer("</table></body></html>");

    public BigDataToExcelImpl(JdbcTemplate jdbcTemplate) {
        super(jdbcTemplate);
    }

    /**
     * 临时文件前缀
     *
     * @return
     */
    protected String getPrefix() {
        return "excelTemp";
    }

    /**
     * 临时文件后缀
     *
     * @return
     */
    protected String getSuffix() {
        return ".xls";
    }

    /**
     * 文件开头的写入
     *
     * @param fos
     * @throws IOException
     */
    protected void writeHeaderToOutputStream(FileOutputStream fos)
            throws IOException {
        writeToOutputStream(headStr.toString(), fos);
    }

    /**
     * 文件结尾的写入
     *
     * @param fos
     */
    protected void writeFooterToOutputStream(FileOutputStream fos)
            throws IOException {
        writeToOutputStream(footStr.toString(), fos);
    }

    /**
     * 一行数据的写入
     *
     * @param rs
     * @param fos
     * @throws SQLException
     * @throws IOException
     */
    protected void writeOneRowToOutputStream(ResultSet rs, FileOutputStream fos)
            throws SQLException, IOException {
        // 获取metaData;
        ResultSetMetaData rsmd = rs.getMetaData();
        int columnCount = rsmd.getColumnCount();
        writeToOutputStream("<tr>", fos);
        for (int i = 1; i <= columnCount; i++) {
            // String key = getColumnKey(rsmd.getColumnName(i));
            Object obj = getColumnValue(rs, i);
            writeToOutputStream("<td>"
                    + HtmlUtils.htmlEscape(obj == null ? "" : obj.toString())
                    + "</td>", fos);
        }
        writeToOutputStream("</tr>", fos);
    }

    protected void fileOutputStreamStatus(List<FileOutputStream> foList)
            throws IOException {
        System.out.println("共有文件输出流：" + foList.size());
        for (FileOutputStream fo : foList) {
            System.out.println("文件输出流："
                    + (fo == null ? "已清空" : fo.toString() + " : "
                            + (fo.getFD().valid())));
        }
    }

    @Override
    protected void writeTitleToOutputStream(Collection<String> titles,
            FileOutputStream fos) throws IOException {
        if (titles != null && titles.size() > 0) {
            writeToOutputStream("<tr>", fos);
            for (String title : titles) {
                writeToOutputStream("<td>"
                        + HtmlUtils.htmlEscape(title == null ? "" : title)
                        + "</td>", fos);
            }
            writeToOutputStream("</tr>", fos);
        }
    }

}