sdfghjtyrt的风格浩特人挖宝 / 憋错料

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);
	}
}

csdn下载频道有人说我是骗子，没有传完整的项目，请那位大神仔细想想，公司开发的商业软件可能会全给您上传到网上么？我已经把完成这个功能的核心代码传上去了，只做交流使用，您那么懒，你父母知道么？

sdfghjtyrt的风格浩特人挖宝

时间： 2024-11-13 08:17:41

sdfghjtyrt的风格浩特人挖宝

sdfghjtyrt的风格浩特人挖宝的相关文章

2015第19周五《象与骑象人》《瞬变》读书笔记

一款简易的CSS3扁平化风格联系表单

红军的基因--为什么有些人做不成电商？

人际沟通风格介绍一

谈判的四种风格

关于扁平化界面风格的设计美学讨论

POI导出EXCEL经典实现

万法简史：肯恩·威尔伯的典范革命

诗外功夫