1. 程式人生 > >Java之——匯出Excel通用工具類

Java之——匯出Excel通用工具類

一、概述

相信大家在工作過程中,都會遇到這樣一個需求,就是將相關的資料列表匯出成excel,那麼,有沒有通用的匯出方式呢,這裡,就帶著大家一起來用Java實現一個通用的匯出Excel的工具。

二、專案實現

1、構建pom.xml

我們的工程是利用Maven來構建的,專案具體搭建過程大家可以參見網上其他資料,這裡我們僅給出最核心的Maven配置

<dependency>
	<groupId>org.apache.poi</groupId>
	<artifactId>poi-scratchpad</artifactId>
	<version>3.11-beta2</version>
</dependency>
<dependency>
	<groupId>org.apache.poi</groupId>
	<artifactId>poi-ooxml</artifactId>
	<version>3.11-beta2</version>
</dependency>
<dependency>
	<groupId>org.apache.poi</groupId>
	<artifactId>poi-ooxml-schemas</artifactId>
	<version>3.11-beta2</version>
</dependency>
<dependency>
	<groupId>org.apache.poi</groupId>
	<artifactId>poi-excelant</artifactId>
	<version>3.11-beta2</version>
</dependency>

2、編寫ExportExcelUtil類

這個類使我們整個工具的核心,它實現了Excel檔案的匯出功能,具體程式碼如下:

package com.lyz.utils.excel.poi;
import java.io.IOException;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.text.SimpleDateFormat;
import java.util.Collection;
import java.util.Date;
import java.util.Iterator;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.commons.lang3.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.hssf.util.HSSFColor;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFCellStyle;
import org.apache.poi.xssf.usermodel.XSSFColor;
import org.apache.poi.xssf.usermodel.XSSFFont;
import org.apache.poi.xssf.usermodel.XSSFRichTextString;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;


/**
 * 匯出Excel
 * @author liuyazhuang
 *
 * @param <T>
 */
public class ExportExcelUtil<T>{
	
	// 2007 版本以上 最大支援1048576行
	public  final static String  EXCEl_FILE_2007 = "2007";
	// 2003 版本 最大支援65536 行
	public  final static String  EXCEL_FILE_2003 = "2003";
	
	/**
	 * <p>
	 * 匯出無頭部標題行Excel <br>
	 * 時間格式預設:yyyy-MM-dd hh:mm:ss <br>
	 * </p>
	 * 
	 * @param title 表格標題
	 * @param dataset 資料集合
	 * @param out 輸出流
	 * @param version 2003 或者 2007,不傳時預設生成2003版本
	 */
	public void exportExcel(String title, Collection<T> dataset, OutputStream out, String version) {
		if(StringUtils.isEmpty(version) || EXCEL_FILE_2003.equals(version.trim())){
			exportExcel2003(title, null, dataset, out, "yyyy-MM-dd HH:mm:ss");
		}else{
			exportExcel2007(title, null, dataset, out, "yyyy-MM-dd HH:mm:ss");
		}
	}

	/**
	 * <p>
	 * 匯出帶有頭部標題行的Excel <br>
	 * 時間格式預設:yyyy-MM-dd hh:mm:ss <br>
	 * </p>
	 * 
	 * @param title 表格標題
	 * @param headers 頭部標題集合
	 * @param dataset 資料集合
	 * @param out 輸出流
	 * @param version 2003 或者 2007,不傳時預設生成2003版本
	 */
	public void exportExcel(String title,String[] headers, Collection<T> dataset, OutputStream out,String version) {
		if(StringUtils.isBlank(version) || EXCEL_FILE_2003.equals(version.trim())){
			exportExcel2003(title, headers, dataset, out, "yyyy-MM-dd HH:mm:ss");
		}else{
			exportExcel2007(title, headers, dataset, out, "yyyy-MM-dd HH:mm:ss");
		}
	}

	/**
	 * <p>
	 * 通用Excel匯出方法,利用反射機制遍歷物件的所有欄位,將資料寫入Excel檔案中 <br>
	 * 此版本生成2007以上版本的檔案 (檔案字尾:xlsx)
	 * </p>
	 * 
	 * @param title
	 *            表格標題名
	 * @param headers
	 *            表格頭部標題集合
	 * @param dataset
	 *            需要顯示的資料集合,集合中一定要放置符合JavaBean風格的類的物件。此方法支援的
	 *            JavaBean屬性的資料型別有基本資料型別及String,Date
	 * @param out
	 *            與輸出裝置關聯的流物件,可以將EXCEL文件匯出到本地檔案或者網路中
	 * @param pattern
	 *            如果有時間資料,設定輸出格式。預設為"yyyy-MM-dd hh:mm:ss"
	 */
	@SuppressWarnings({ "unchecked", "rawtypes" })
	public void exportExcel2007(String title, String[] headers, Collection<T> dataset, OutputStream out, String pattern) {
		// 宣告一個工作薄
		XSSFWorkbook workbook = new XSSFWorkbook();
		// 生成一個表格
		XSSFSheet sheet = workbook.createSheet(title);
		// 設定表格預設列寬度為15個位元組
		sheet.setDefaultColumnWidth(20);
		// 生成一個樣式
		XSSFCellStyle style = workbook.createCellStyle();
		// 設定這些樣式
		style.setFillForegroundColor(new XSSFColor(java.awt.Color.gray));
		style.setFillPattern(XSSFCellStyle.SOLID_FOREGROUND);
		style.setBorderBottom(XSSFCellStyle.BORDER_THIN);
		style.setBorderLeft(XSSFCellStyle.BORDER_THIN);
		style.setBorderRight(XSSFCellStyle.BORDER_THIN);
		style.setBorderTop(XSSFCellStyle.BORDER_THIN);
		style.setAlignment(XSSFCellStyle.ALIGN_CENTER);
		// 生成一個字型
		XSSFFont font = workbook.createFont();
		font.setBoldweight(XSSFFont.BOLDWEIGHT_BOLD);
		font.setFontName("宋體"); 
		font.setColor(new XSSFColor(java.awt.Color.BLACK));
		font.setFontHeightInPoints((short) 11);
		// 把字型應用到當前的樣式
		style.setFont(font);
		// 生成並設定另一個樣式
		XSSFCellStyle style2 = workbook.createCellStyle();
		style2.setFillForegroundColor(new XSSFColor(java.awt.Color.WHITE));
		style2.setFillPattern(XSSFCellStyle.SOLID_FOREGROUND);
		style2.setBorderBottom(XSSFCellStyle.BORDER_THIN);
		style2.setBorderLeft(XSSFCellStyle.BORDER_THIN);
		style2.setBorderRight(XSSFCellStyle.BORDER_THIN);
		style2.setBorderTop(XSSFCellStyle.BORDER_THIN);
		style2.setAlignment(XSSFCellStyle.ALIGN_CENTER);
		style2.setVerticalAlignment(XSSFCellStyle.VERTICAL_CENTER);
		// 生成另一個字型
		XSSFFont font2 = workbook.createFont();
		font2.setBoldweight(XSSFFont.BOLDWEIGHT_NORMAL);
		// 把字型應用到當前的樣式
		style2.setFont(font2);

		// 產生表格標題行
		XSSFRow row = sheet.createRow(0);
		XSSFCell cellHeader;
		for (int i = 0; i < headers.length; i++) {
			cellHeader = row.createCell(i);
			cellHeader.setCellStyle(style);
			cellHeader.setCellValue(new XSSFRichTextString(headers[i]));
		}

		// 遍歷集合資料,產生資料行
		Iterator<T> it = dataset.iterator();
		int index = 0;
		T t;
		Field[] fields;
		Field field;
		XSSFRichTextString richString;
		Pattern p = Pattern.compile("^//d+(//.//d+)?$");
		Matcher matcher;
		String fieldName;
		String getMethodName;
		XSSFCell cell;
		Class tCls;
		Method getMethod;
		Object value;
		String textValue;
		SimpleDateFormat sdf = new SimpleDateFormat(pattern);
		while (it.hasNext()) {
			index++;
			row = sheet.createRow(index);
			t = (T) it.next();
			// 利用反射,根據JavaBean屬性的先後順序,動態呼叫getXxx()方法得到屬性值
			fields = t.getClass().getDeclaredFields();
			for (int i = 0; i < fields.length; i++) {
				cell = row.createCell(i);
				cell.setCellStyle(style2);
				field = fields[i];
				fieldName = field.getName();
				getMethodName = "get" + fieldName.substring(0, 1).toUpperCase()
						+ fieldName.substring(1);
				try {
					tCls = t.getClass();
					getMethod = tCls.getMethod(getMethodName, new Class[] {});
					value = getMethod.invoke(t, new Object[] {});
					// 判斷值的型別後進行強制型別轉換
					textValue = null;
					if (value instanceof Integer) {
						cell.setCellValue((Integer) value);
					} else if (value instanceof Float) {
						textValue = String.valueOf((Float) value);
						cell.setCellValue(textValue);
					} else if (value instanceof Double) {
						textValue = String.valueOf((Double) value);
						cell.setCellValue(textValue);
					} else if (value instanceof Long) {
						cell.setCellValue((Long) value);
					}
					if (value instanceof Boolean) {
						textValue = "是";
						if (!(Boolean) value) {
							textValue = "否";
						}
					} else if (value instanceof Date) {
						textValue = sdf.format((Date) value);
					} else {
						// 其它資料型別都當作字串簡單處理
						if (value != null) {
							textValue = value.toString();
						}
					}
					if (textValue != null) {
						matcher = p.matcher(textValue);
						if (matcher.matches()) {
							// 是數字當作double處理
							cell.setCellValue(Double.parseDouble(textValue));
						} else {
							richString = new XSSFRichTextString(textValue);
							cell.setCellValue(richString);
						}
					}
				} catch (SecurityException e) {
					e.printStackTrace();
				} catch (NoSuchMethodException e) {
					e.printStackTrace();
				} catch (IllegalArgumentException e) {
					e.printStackTrace();
				} catch (IllegalAccessException e) {
					e.printStackTrace();
				} catch (InvocationTargetException e) {
					e.printStackTrace();
				} finally {
					// 清理資源
				}
			}
		}
		try {
			workbook.write(out);
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	
	
	/**
	 * <p>
	 * 通用Excel匯出方法,利用反射機制遍歷物件的所有欄位,將資料寫入Excel檔案中 <br>
	 * 此方法生成2003版本的excel,檔名字尾:xls <br>
	 * </p>
	 * 
	 * @param title
	 *            表格標題名
	 * @param headers
	 *            表格頭部標題集合
	 * @param dataset
	 *            需要顯示的資料集合,集合中一定要放置符合JavaBean風格的類的物件。此方法支援的
	 *            JavaBean屬性的資料型別有基本資料型別及String,Date
	 * @param out
	 *            與輸出裝置關聯的流物件,可以將EXCEL文件匯出到本地檔案或者網路中
	 * @param pattern
	 *            如果有時間資料,設定輸出格式。預設為"yyyy-MM-dd hh:mm:ss"
	 */
	@SuppressWarnings({ "unchecked", "rawtypes" })
	public void exportExcel2003(String title, String[] headers, Collection<T> dataset, OutputStream out, String pattern) {
		// 宣告一個工作薄
		HSSFWorkbook workbook = new HSSFWorkbook();
		// 生成一個表格
		HSSFSheet sheet = workbook.createSheet(title);
		// 設定表格預設列寬度為15個位元組
		sheet.setDefaultColumnWidth(20);
		// 生成一個樣式
		HSSFCellStyle style = workbook.createCellStyle();
		// 設定這些樣式
		style.setFillForegroundColor(HSSFColor.GREY_50_PERCENT.index);
		style.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);
		style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
		style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
		style.setBorderRight(HSSFCellStyle.BORDER_THIN);
		style.setBorderTop(HSSFCellStyle.BORDER_THIN);
		style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
		// 生成一個字型
		HSSFFont font = workbook.createFont();
		font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
		font.setFontName("宋體"); 
		font.setColor(HSSFColor.WHITE.index);
		font.setFontHeightInPoints((short) 11);
		// 把字型應用到當前的樣式
		style.setFont(font);
		// 生成並設定另一個樣式
		HSSFCellStyle style2 = workbook.createCellStyle();
		style2.setFillForegroundColor(HSSFColor.WHITE.index);
		style2.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);
		style2.setBorderBottom(HSSFCellStyle.BORDER_THIN);
		style2.setBorderLeft(HSSFCellStyle.BORDER_THIN);
		style2.setBorderRight(HSSFCellStyle.BORDER_THIN);
		style2.setBorderTop(HSSFCellStyle.BORDER_THIN);
		style2.setAlignment(HSSFCellStyle.ALIGN_CENTER);
		style2.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
		// 生成另一個字型
		HSSFFont font2 = workbook.createFont();
		font2.setBoldweight(HSSFFont.BOLDWEIGHT_NORMAL);
		// 把字型應用到當前的樣式
		style2.setFont(font2);

		// 產生表格標題行
		HSSFRow row = sheet.createRow(0);
		HSSFCell cellHeader;
		for (int i = 0; i < headers.length; i++) {
			cellHeader = row.createCell(i);
			cellHeader.setCellStyle(style);
			cellHeader.setCellValue(new HSSFRichTextString(headers[i]));
		}

		// 遍歷集合資料,產生資料行
		Iterator<T> it = dataset.iterator();
		int index = 0;
		T t;
		Field[] fields;
		Field field;
		HSSFRichTextString richString;
		Pattern p = Pattern.compile("^//d+(//.//d+)?$");
		Matcher matcher;
		String fieldName;
		String getMethodName;
		HSSFCell cell;
		Class tCls;
		Method getMethod;
		Object value;
		String textValue;
		SimpleDateFormat sdf = new SimpleDateFormat(pattern);
		while (it.hasNext()) {
			index++;
			row = sheet.createRow(index);
			t = (T) it.next();
			// 利用反射,根據JavaBean屬性的先後順序,動態呼叫getXxx()方法得到屬性值
			fields = t.getClass().getDeclaredFields();
			for (int i = 0; i < fields.length; i++) {
				cell = row.createCell(i);
				cell.setCellStyle(style2);
				field = fields[i];
				fieldName = field.getName();
				getMethodName = "get" + fieldName.substring(0, 1).toUpperCase()
						+ fieldName.substring(1);
				try {
					tCls = t.getClass();
					getMethod = tCls.getMethod(getMethodName, new Class[] {});
					value = getMethod.invoke(t, new Object[] {});
					// 判斷值的型別後進行強制型別轉換
					textValue = null;
					if (value instanceof Integer) {
						cell.setCellValue((Integer) value);
					} else if (value instanceof Float) {
						textValue = String.valueOf((Float) value);
						cell.setCellValue(textValue);
					} else if (value instanceof Double) {
						textValue = String.valueOf((Double) value);
						cell.setCellValue(textValue);
					} else if (value instanceof Long) {
						cell.setCellValue((Long) value);
					}
					if (value instanceof Boolean) {
						textValue = "是";
						if (!(Boolean) value) {
							textValue = "否";
						}
					} else if (value instanceof Date) {
						textValue = sdf.format((Date) value);
					} else {
						// 其它資料型別都當作字串簡單處理
						if (value != null) {
							textValue = value.toString();
						}
					}
					if (textValue != null) {
						matcher = p.matcher(textValue);
						if (matcher.matches()) {
							// 是數字當作double處理
							cell.setCellValue(Double.parseDouble(textValue));
						} else {
							richString = new HSSFRichTextString(textValue);
							cell.setCellValue(richString);
						}
					}
				} catch (SecurityException e) {
					e.printStackTrace();
				} catch (NoSuchMethodException e) {
					e.printStackTrace();
				} catch (IllegalArgumentException e) {
					e.printStackTrace();
				} catch (IllegalAccessException e) {
					e.printStackTrace();
				} catch (InvocationTargetException e) {
					e.printStackTrace();
				} finally {
					// 清理資源
				}
			}
		}
		try {
			workbook.write(out);
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
}

為了演示匯出功能,這裡我們建立了一個Student學生類。

3、建立Student類

package com.lyz.utils.excel.poi;

/**
 * 例子JavaBean
 * @author liuyazhuang
 *
 */
public class Student {
    private int id;
    private String name;
    private String sex;

    public Student(int id, String name, String sex) {
        this.id = id;
        this.name = name;
        this.sex = sex;
    }

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getSex() {
        return sex;
    }

    public void setSex(String sex) {
        this.sex = sex;
    }
}

4、建立測試類TestExportExcelUtil

這個類,主要是用來測試我們的工具類。具體程式碼如下:

package com.lyz.test;

import java.io.FileOutputStream;
import java.util.ArrayList;
import java.util.List;

import com.lyz.utils.excel.poi.ExportExcelUtil;
import com.lyz.utils.excel.poi.Student;

/**
 * 測試檔案匯出
 * @author liuyazhuang
 *
 */
public class TestExportExcelUtil {
	
	public static void main(String[] args) throws Exception{
		ExportExcelUtil<Student> util = new ExportExcelUtil<Student>();
		 // 準備資料
        List<Student> list = new ArrayList<>();
        for (int i = 0; i < 10; i++) {
        	 list.add(new Student(111,"張三asdf","男"));
             list.add(new Student(111,"李四asd","男"));
             list.add(new Student(111,"王五","女"));
        }
        String[] columnNames = { "ID", "姓名", "性別" };
        util.exportExcel("使用者匯出", columnNames, list, new FileOutputStream("E:/test.xls"), ExportExcelUtil.EXCEL_FILE_2003);
	}
}

5、測試

我們執行TestExportExcelUtil類,會發現在我們的E盤下生成了test.xls檔案,具體如下:

三、專案擴充套件

以上實現均是在本地磁碟生成excel檔案,但更多的時候,我們需要通過點選網頁上的某個按鈕來自動生成並下載Excel文件,那麼,這又如何實現呢?下面我們就一起來實現這個功能。

1、擴充套件ExportExcelUtil類

根據Java的繼承思想,我們不在ExportExcelUtil類上修改新增,我們建立一個ExportExcelUtil類的子類ExportExcelWrapper,這個類繼承ExportExcelUtil的所有功能,同時,擴充套件了網頁生成Excel的功能。具體程式碼如下:

package com.lyz.utils.excel.poi;

import java.net.URLEncoder;
import java.util.Collection;

import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang3.StringUtils;

/**
 * 包裝類
 * @author liuyazhuang
 *
 * @param <T>
 */
public class ExportExcelWrapper<T> extends ExportExcelUtil<T> {
	/**
	 * <p>
	 * 匯出帶有頭部標題行的Excel <br>
	 * 時間格式預設:yyyy-MM-dd hh:mm:ss <br>
	 * </p>
	 * 
	 * @param title 表格標題
	 * @param headers 頭部標題集合
	 * @param dataset 資料集合
	 * @param out 輸出流
	 * @param version 2003 或者 2007,不傳時預設生成2003版本
	 */
	public void exportExcel(String fileName, String title, String[] headers, Collection<T> dataset, HttpServletResponse response,String version) {
		try {
			response.setContentType("application/vnd.ms-excel");  
    		response.addHeader("Content-Disposition", "attachment;filename="+ URLEncoder.encode(fileName, "UTF-8") + ".xls");
			if(StringUtils.isBlank(version) || EXCEL_FILE_2003.equals(version.trim())){
				exportExcel2003(title, headers, dataset, response.getOutputStream(), "yyyy-MM-dd hh:mm:ss");
			}else{
				exportExcel2007(title, headers, dataset, response.getOutputStream(), "yyyy-MM-dd hh:mm:ss");
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
}

這樣,我們可以在Controller層呼叫ExportExcelWrapper將相關的資料和HttpServletResponse傳遞進來,即可實現通過網頁生生並下載Excel文件了。

2、編寫Controller類

@Controller("test")
@RequestMapping("/test")
public class TestController {
	@RequestMapping("/get/excel")
	public void getExcel(HttpServletRequest request, HttpServletResponse response) throws Exception {
		// 準備資料
		List<Student> list = new ArrayList<>();
		for (int i = 0; i < 10; i++) {
			 list.add(new Student(111,"張三asdf","男"));
		     list.add(new Student(111,"李四asd","男"));
		     list.add(new Student(111,"王五","女"));
		}
		String[] columnNames = { "ID", "姓名", " 性別"};
		String fileName = "excel1";
		ExportExcelWrapper<Student> util = new ExportExcelWrapper<Student>();
		util.exportExcel(fileName, fileName, columnNames, list, response, ExportExcelUtil.EXCEL_FILE_2003);
	}
}

3、編寫index.html

這個網頁很簡單,具體實現如下:

<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8">
<title>匯出Excel</title>
</head>
<body>
<form id="form_login" action="http://127.0.0.1:8080/test/get/excel" method="post">
  
</form>
<button id="btn-submit" onclick="beforeSubmit()">Submit</button>
<script type="text/javascript">
    var loginForm = document.getElementById('form_login');
    function beforeSubmit() {
        loginForm.submit();
    }
</script>
</body>
</html>

4、測試

我們將程式釋出到Tomcat,並點選網上的按鈕,效果如下:

我們開啟excel1.xls檔案如下:

至此,我們的工具就編寫完成了。