利用poi3.9做的excel导出工具

一、先看看所生成的文件效果图

二、准备

本文需要六个jar包：

dom4j-1.6.1.jar
ojdbc14.jar
poi-3.9-20121203.jar
poi-ooxml-3.9-20121203.jar
poi-ooxml-schemas-3.9-20121203.jar
xmlbeans-2.3.0.jar

除了ojdbc14.jar是用来访问数据库的，其它的都是导出excel所需要的poi相关jar包。

注：本文是以poi3.9版本写的，利用了SXSSFWorkbook这个Workbook，这个可以分批写入，防止内存溢出。这个类只有在3.8及3.8以上版本才有。

三、写个连接数据库的工具

写得很简单，用的是oracle数据库

代码如下：

package com.utils;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;

public class DBUtil {
	private static String driver="oracle.jdbc.driver.OracleDriver";
	private static String url="jdbc:oracle:thin:@127.0.0.1:1521:xe";
	private static String user="test";
	private static String password="test";
	public static Connection getConnection(){
		Connection conn = null;
		try {
			Class.forName(driver);
			conn = DriverManager.getConnection(url,user,password);
		} catch (ClassNotFoundException e) {
			e.printStackTrace();
		} catch (SQLException e) {
			e.printStackTrace();
		}
		return conn;
	}
	
	public static void main(String[] args) {
		System.out.println(DBUtil.getConnection());//测试连接
	}
}

四、写导出工具类

代码如下：

package com.utils;

import java.io.FileOutputStream;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;

public class POIExport {
	/**
	 * 根据所传入的参数生成一个Workbook
	 * @param sql 查询数据的sql
	 * @param columns sql中列名字符串，以英文逗号分隔，不区分大小写
	 * @param headers 表头字符串数组，如果是多表头，在需要合并的地方写"null"，如：test1,test2,null,null,test3。
	 * 这表示test2将占三列。如果test2下面没有null，则占一行三列，有n个null，则占n行3列
	 * @param splitStr 分割表头字符串的分割符
	 * @return
	 */
	public static Workbook export(String sql,String columns,String[] headers, String splitStr){
		SXSSFWorkbook wb = new SXSSFWorkbook(1000);//创建excel文档,内存中保留 1000 条数据，以免内存溢出
		Font font = wb.createFont();//字体
		font.setBoldweight(Font.BOLDWEIGHT_BOLD);//加粗
		CellStyle cellStylehead = wb.createCellStyle();//表头样式
		cellStylehead.setFont(font);//设置字体样式
		cellStylehead.setAlignment(CellStyle.ALIGN_CENTER);//水平对齐
		cellStylehead.setVerticalAlignment(CellStyle.VERTICAL_CENTER);//垂直对齐
		cellStylehead.setWrapText(true);//自动换行
		//设置边框
		cellStylehead.setBorderTop(CellStyle.BORDER_THIN);
		cellStylehead.setBorderRight(CellStyle.BORDER_THIN);
		cellStylehead.setBorderBottom(CellStyle.BORDER_THIN);
		cellStylehead.setBorderLeft(CellStyle.BORDER_THIN);
		//表体样式
		CellStyle cellStyleBody = wb.createCellStyle();//表体单元格样式
		cellStyleBody.setAlignment(CellStyle.ALIGN_LEFT);//水平对齐
		cellStyleBody.setVerticalAlignment(CellStyle.VERTICAL_CENTER);//垂直对齐
		cellStyleBody.setWrapText(true);//自动换行
		//设置边框
		cellStyleBody.setBorderTop(CellStyle.BORDER_THIN);
		cellStyleBody.setBorderRight(CellStyle.BORDER_THIN);
		cellStyleBody.setBorderBottom(CellStyle.BORDER_THIN);
		cellStyleBody.setBorderLeft(CellStyle.BORDER_THIN);
		Sheet sheet = wb.createSheet("sheet1");//创建一个sheet
		sheet.setDefaultColumnWidth(15);//设置默认列宽
		//写表头
		createHeader(wb,sheet,cellStylehead,headers,splitStr);
		//写表体
		int beginRowNumber = headers.length;//表体开始行
		String[] cols = columns.split(",");//切分sql列名
		int cellSize = cols.length;//列数
		Connection conn = null;
		Statement stmt = null;
		ResultSet rs = null;
		conn = DBUtil.getConnection();
		try {
			stmt = conn.createStatement();
			rs = stmt.executeQuery(sql);
			int count = 0;//记录行号
			while(rs.next()){
				Row row = sheet.createRow(count+beginRowNumber);
				row.setHeightInPoints(14);//设置行高
				for(int j=0;j<cellSize;j++){
					Cell cell = row.createCell(j);
					Object obj = rs.getObject(cols[j]);
					String cv = obj==null?"":obj.toString();//取得对应列中的值
					cell.setCellValue(cv);//设置单元格的值
					cell.setCellStyle(cellStyleBody);//设置样式
				}
				count++;
			}
			System.out.println("共写入数据："+count+"条");
		} catch (SQLException e) {
			e.printStackTrace();
		} finally {
			try {
				if(rs!=null){
					rs.close();
				}
			} catch (SQLException e) {
				e.printStackTrace();
			}
			try {
				if(stmt!=null){
					stmt.close();
				}
			} catch (SQLException e) {
				e.printStackTrace();
			}
			try {
				if(conn!=null){
					conn.close();
				}
			} catch (SQLException e) {
				e.printStackTrace();
			}
		}
		return wb;
	}

	/**
	 * 创建excel表头
	 * @param wb excel的workbook
	 * @param sheet excel的sheet
	 * @param cellStylehead excel的样式
	 * @param headers	表头字符串数组
	 * @param splitStr	表头字符串切割符
	 */
	private static void createHeader(SXSSFWorkbook wb, Sheet sheet,
			CellStyle cellStylehead, String[] headers, String splitStr) {
		//遍历创建单元格
		for(int i=0;i<headers.length;i++){
			Row row = sheet.createRow(i);
			String[] header = headers[i].split(splitStr);
			for(int r=0;r<header.length;r++){
				Cell cell = row.createCell(r);
				cell.setCellValue(header[r]);
				cell.setCellStyle(cellStylehead);
			}
		}
		//遍历合并单元格，如果是单表头则跳过
		if(headers.length>1){
			int[][][] mergeDatas = parseHeader(headers,splitStr);
			for(int i=0;i<mergeDatas.length;i++){
				int[][] mergeData = mergeDatas[i];
				for(int j=0;j<mergeData.length;j++){
					int[] merges = mergeData[j];
					int mergesR = merges[1]-merges[0];
					int mergesC = merges[3]-merges[2];
					if(mergesR!=0||mergesC!=0){
						//合并单元格
						sheet.addMergedRegion(new CellRangeAddress(merges[0],merges[1],merges[2],merges[3]));
						//合并单元格后重新设置单元格的样式
						setMergedStyle(merges[0],merges[1],merges[2],merges[3],sheet,cellStylehead);
					}
				}
			}
		}
	}
	/**
	 * 设置合并单元格的样式
	 * @param i	first row(0-based)
	 * @param j	last row(0-based)
	 * @param k	first column(0-based)
	 * @param l	last column(0-based)
	 * @param sheet	excel的sheet
	 * @param cellStylehead	excel的样式
	 */
	private static void setMergedStyle(int i, int j, int k, int l, Sheet sheet, CellStyle cellStylehead) {
		for(int mm=i;mm<=j;mm++){
			Row row = sheet.getRow(mm);
			if(row==null){
				row = sheet.createRow(mm);
			}
			for(int nn=k;nn<=l;nn++){
				Cell cell = row.getCell(nn);
				if(cell==null){
					cell = row.createCell(nn);
				}
				cell.setCellStyle(cellStylehead);
			}
		}
	}
	/**
	 * 解析复杂表头，表头为多行且有合并的情况。表头要符合约定格式。
	 * 格式例子：
	 * String h21 = "test,test,test,null,null,null,null,null,test,null,null,null,null,null,test";
	 * String h22 = "null,null,test,null,null,test,null,null,test,null,test,null,test,null,null";
	 * String h23 = "null,null,test,test,test,test,test,test,test,test,test,test,test,test,null";
	 * 这里表头有三行，且有合并情况。一行为一个字符串，字符串以英文逗号隔开，在需要合并的地方填写"null"字符串
	 * @param headers 要解析的表头字符串数组
	 * @param splitStr 切割表头字符串的分割符
	 * @return 返回一个三维数组，第三重数据中保存着合并数据
	 */
	private static int[][][] parseHeader(String[] headers, String splitStr){
		//依据表头建立一个二维数组
		String[][] doubleAry = new String[headers.length][];
		//依据表头建立一个三维数组，用来保存合并单元格所需要的数据，单元格所需要的数据有四个，依次是开始合并的行，结束合并的行，开始合并的列，结束合并的列
		//sheet.addMergedRegion(new CellRangeAddress(merges[0],merges[1],merges[2],merges[3]));
		int[][][] rcs = new int[headers.length][][];
		//遍历赋值
		for(int i=0;i<headers.length;i++){
			String[] header = headers[i].split(splitStr);
			doubleAry[i] = header;
			rcs[i] = new int[header.length][4];
		}
		//遍历二维数组
		for(int i=0;i<doubleAry.length;i++){
			String[] sub = doubleAry[i];
			for(int j=0;j<sub.length;j++){
				int sum = 0;//计算行标识
				//计算i,j需要合并的行数
				if(!sub[j].equals("null")){//如果单元格不为null
					for(int m=i;m<doubleAry.length;m++){
						String rs = doubleAry[m][j];
						if(rs.equals("null")){//如果单元格为null
							doubleAry[m][j] = "null2";//把null重赋值为null2，防止在计算列时交错了
							sum++;//个数加1
						}
					}
				}
				rcs[i][j][0] = i;//赋合并的开始行，当前行
				rcs[i][j][1] = i+sum;//赋合并的结束行，当前行加上它下面为null的行数
				int sum2 = 0;//计算列标识
				//计算i,j需要合并的列数
				if(!sub[j].equals("null")){//如果单元格不为null
					for(int m=j+1;m<doubleAry[i].length;m++){
						String rs = doubleAry[i][m];
						if(rs.equals("null")){//如果单元格为null
							sum2++;
						}else{
							break;//一定要break，不然会算错
						}
					}
				}
				rcs[i][j][2] = j;//赋合并的开始列，当前列
				rcs[i][j][3] = j+sum2;//赋合并的结束列，当前列加上它右边为null的列数
			}
		}
		return rcs;
	}
	public static void main(String[] args) throws Exception {
		StringBuilder sb = new StringBuilder();
		sb.append("select to_char(sysdate,'yyyymmdd') c1,       \n");
		sb.append("'类别'||round(dbms_random.value(0,4)) c2,    \n");
		sb.append("round(dbms_random.value(0,10000),2) c3,      \n");
		sb.append("round(dbms_random.value(0,100),2)||'%' c4,   \n");
		sb.append("round(dbms_random.value(0,10000),2) c5,      \n");
		sb.append("round(dbms_random.value(0,10000),2) c6,      \n");
		sb.append("round(dbms_random.value(0,10000),2) c7,      \n");
		sb.append("round(dbms_random.value(0,10000),2) c8       \n");
		sb.append(" from dual connect by level < 100             \n");//注：改这里的数字，可得到不同行数的数据
		String sql = sb.toString();
		System.out.println("sql:\n"+sql);
		String columns = "c1,c2,c3,c4,c5,c6,c7,c8";
		System.out.println("columns:\n"+columns);
		String[] headers = new String[2];
		headers[0] = "日期,类别,3G,null,4G,null,null,null";
		headers[1] = "null,null,月流量,同比,月流量,上行流量,下行流量,赠送流量";
		System.out.println("表头：");
		for(int i=0;i<headers.length;i++){
			String[] header = headers[i].split(",");
			for(int j=0;j<header.length;j++){
				System.out.print(header[j]+"\t");
			}
			System.out.println();
		}
		System.out.println("生成文件开始。。。");
		long t1 = System.currentTimeMillis();
		Workbook wb = export(sql,columns,headers,",");
		long t11 = System.currentTimeMillis();
		System.out.println("生成Workbook共花费："+(t11-t1)+"毫秒");
		//写入文件
		FileOutputStream out = new FileOutputStream("f:/test/exp.xlsx");
		wb.write(out);
		out.close();
		long t2 = System.currentTimeMillis();
		System.out.println("把Workbook写入文件共花费："+(t2-t11)+"毫秒");
		System.out.println("生成文件结束，共花费："+(t2-t1)+"毫秒");
		//测试10万行*8列，所花时间13620毫秒
		//测试100万行*8列，所花时间121443毫秒
	}
}

五、如有什么bug请批评指正。

我把源码打包了，如果不想浪费2分，可以直接复制以上代码。如果不想复制，可以点这里下载：

http://download.csdn.net/detail/yunsyz/8345949

附文件目录图：