POI中HSSFWorkbook匯出excel封裝的工具類
阿新 • • 發佈:2018-11-17
在實際中匯出excel非常常見,於是自己封裝了一個匯出資料到excel的工具類,先附上程式碼,最後會寫出例項和解釋。
程式碼中依賴了slf4j日誌包,commons-io包的IOUtils關閉流,commons-lang和commons-collections包等包。
package cn.xm.exam.utils; import java.io.File; import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.IOException;import java.io.OutputStream; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import org.apache.commons.collections.MapUtils; import org.apache.commons.io.IOUtils; import org.apache.commons.lang.ArrayUtils; 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.HSSFRow; import org.apache.poi.hssf.usermodel.HSSFSheet; import org.apache.poi.hssf.usermodel.HSSFWorkbook; importorg.slf4j.Logger; import org.slf4j.LoggerFactory; public class HSSFWorkExcel { private static final Logger LOGGER = LoggerFactory.getLogger(HSSFWorkExcel.class); private String[] headerNames; private HSSFWorkbook workBook; private HSSFSheet sheet; public HSSFWorkExcel(String[] headerNames, String sheetName) { this.headerNames = headerNames; // 建立一個工作簿 workBook = new HSSFWorkbook(); // 建立一個工作表sheet sheet = workBook.createSheet(sheetName); initHeader(); } /** * 初始化表頭資訊 */ private void initHeader() { // 建立第一行 HSSFRow row = sheet.createRow(0); HSSFCell cell = null; // 建立表頭 for (int i = 0; i < headerNames.length; i++) { cell = row.createCell(i); cell.setCellValue(headerNames[i]); setCellStyle(cell); } } /** * 設定單元格樣式 * * @param cell * 單元格 */ public void setCellStyle(HSSFCell cell) { // 設定樣式 HSSFCellStyle cellStyle = workBook.createCellStyle(); cellStyle.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 設定字型居中 // 設定字型 HSSFFont font = workBook.createFont(); font.setFontName("宋體"); font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);// 字型加粗 font.setFontHeightInPoints((short) 13); cellStyle.setFont(font); cell.setCellStyle(cellStyle); } /** * 建立行內容(每一行的資料裝在list中) * * @param datas * 每一行的資料 * @param rowIndex * 行號(從1開始) */ public void createTableRow(List<String> datas, int rowIndex) { // 建立第i行 HSSFRow row = sheet.createRow(rowIndex); HSSFCell cell = null; // 寫入資料 for (int index = 0, length = datas.size(); index < length; index++) { // 引數代表第幾列 cell = row.createCell(index); cell.setCellType(HSSFCell.CELL_TYPE_STRING); cell.setCellValue(datas.get(index)); } } /** * * @param datas * 資料,每一個map都是一行 * @param keys * key[i]代表從map中獲取keys[i]的值作為第i列的值,如果傳的是null預設取表頭 */ public void createTableRows(List<Map<String, Object>> datas, String[] keys) { for (int i = 0, length_1 = datas.size(); i < length_1; i++) { if (ArrayUtils.isEmpty(keys)) { keys = headerNames; } // 建立行(從第二行開始) Map<String, Object> data = datas.get(i); HSSFRow row = sheet.createRow(i + 1); HSSFCell cell = null; for (int j = 0, length_2 = keys.length; j < length_2; j++) { // 單元格獲取map中的key String key = keys[j]; String value = MapUtils.getString(data, key, ""); cell = row.createCell(j); cell.setCellType(HSSFCell.CELL_TYPE_STRING); cell.setCellValue(value); } } } /** * 根據表頭自動調整列寬度 */ public void autoAllSizeColumn() { for (int i = 0, length = headerNames.length; i < length; i++) { sheet.autoSizeColumn(i); } } /** * 將資料寫出到excel中 * * @param outputStream */ public void exportExcel(OutputStream outputStream) { // 匯出之前先自動設定列寬 this.autoAllSizeColumn(); try { workBook.write(outputStream); } catch (IOException e) { LOGGER.error(" exportExcel error", e); } finally { IOUtils.closeQuietly(outputStream); } } }
上面的程式碼邏輯非常簡單,建立例項的時候就初始化表頭資訊和建立sheet。
向excel中填充資料的方式有兩種,就是上面的createTableRow方法和createTableRows方法。
createTableRow(List,int)方法就是多次呼叫此方法,list中資料,int是行號。list中資料依次作為第i行的列資料。
createTableRows(List<Map>,String[])這個方法應該是非常常用的一種方式。我們從資料庫查詢到的資料大多數對映為Map放入list中,因此上面的方法就比較常用。List<Map>引數就是所有的資料,一個Map代表一行,String[]是Map中的key,也就是陣列的第一個元素對應的key作為第一列,第二個元素是作為第二列。如果map資料的key正好與表頭一致我們可以傳一個null。(因為Map是基於陣列+連結串列,且存入的是無序的,所以無法直接通過map中的key確定列。除非傳的資料是LinkedHashMap)
下面是自己的兩個測試程式碼:
(1)測試List<String>寫入資料
public static void test1() { HSSFWorkExcel hssfWorkExcel = new HSSFWorkExcel(new String[] { "姓名", "年齡" }, "人員基本資訊"); for (int i = 0; i < 10; i++) { List<String> data = new ArrayList<>(); data.add("namesssssssssssssss水水水水水水水水水水水水水水水水水水水ssssssssssssssss" + i); data.add("" + (i + 20)); hssfWorkExcel.createTableRow(data, i + 1); } try { hssfWorkExcel.exportExcel(new FileOutputStream(new File("e:/test.xls"))); } catch (FileNotFoundException e) { e.printStackTrace(); } }
結果:
(2)測試List<Map>寫入資料
public static void test2() { HSSFWorkExcel hssfWorkExcel = new HSSFWorkExcel(new String[] { "姓名", "年齡" }, "人員基本資訊"); List<Map<String, Object>> datas = new ArrayList<>(); for (int i = 0; i < 10; i++) { Map data = new HashMap<>(); data.put("name", "tttttttttttttt" + i); data.put("age", "age" + i); datas.add(data); } hssfWorkExcel.createTableRows(datas, new String[] { "name", "age" }); try { hssfWorkExcel.exportExcel(new FileOutputStream(new File("e:/test1.xls"))); } catch (FileNotFoundException e) { e.printStackTrace(); } }
結果: