1. 程式人生 > >apache poi根據模板導出excel

apache poi根據模板導出excel

use 一個 factor exception cep 得到 format 拷貝 return

需要預先新建編輯好一個excel文件,設置好樣式。

編輯好輸出的數據,根據excel坐標一一對應。

支持列表數據輸出,列表中列合並。

代碼如下:

package com.icourt.util;

import org.apache.commons.collections4.CollectionUtils;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddress;

import java.io.*; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Map.Entry; import java.util.regex.Matcher; import java.util.regex.Pattern; /** * 描述:poi根據模板導出excel,根據excel坐標賦值,如(B1) */ public class ExcelExportUtil { //模板map private
Map<String, Workbook> tempWorkbook = new HashMap<String, Workbook>(); //模板輸入流map private Map<String, InputStream> tempStream = new HashMap<String, InputStream>(); /** * 功能:按模板向Excel中相應地方填充數據 */ public void writeData(String templateFilePath, Map<String, Object> dataMap, int
sheetNo) throws IOException, InvalidFormatException { if (dataMap == null || dataMap.isEmpty()) { return; } //讀取模板 Workbook wbModule = getTempWorkbook(templateFilePath); //數據填充的sheet Sheet wsheet = wbModule.getSheetAt(sheetNo); for (Entry<String, Object> entry : dataMap.entrySet()) { String point = entry.getKey(); Object data = entry.getValue(); TempCell cell = getCell(point, data, wsheet); //指定坐標賦值 setCell(cell, wsheet); } //設置生成excel中公式自動計算 wsheet.setForceFormulaRecalculation(true); } /** * 功能:按模板向Excel中列表填充數據.只支持列合並 */ public void writeDateList(String templateFilePath, String[] heads, List<Map<Integer, Object>> datalist, int sheetNo) throws IOException, InvalidFormatException { if (heads == null || heads.length <= 0 || CollectionUtils.isEmpty(datalist)) { return; } //讀取模板 Workbook wbModule = getTempWorkbook(templateFilePath); //數據填充的sheet Sheet wsheet = wbModule.getSheetAt(sheetNo); //列表數據模板cell List<TempCell> tempCells = new ArrayList<TempCell>(heads.length); for (String point : heads) { TempCell tempCell = getCell(point, null, wsheet); //取得合並單元格位置 -1:表示不是合並單元格 int pos = isMergedRegion(wsheet, tempCell.getRow(), tempCell.getColumn()); if (pos > -1) { CellRangeAddress range = wsheet.getMergedRegion(pos); tempCell.setColumnSize(range.getLastColumn() - range.getFirstColumn()); } tempCells.add(tempCell); } //賦值 for (int i = 0; i < datalist.size(); i++) {//數據行 Map<Integer, Object> dataMap = datalist.get(i); for (int j = 0; j < tempCells.size(); j++) {// TempCell tempCell = tempCells.get(j); tempCell.setData(dataMap.get(j + 1)); setCell(tempCell, wsheet); tempCell.setRow(tempCell.getRow() + 1); } } } /** * 功能:獲取輸入工作區 */ private Workbook getTempWorkbook(String templateFilePath) throws IOException, InvalidFormatException { if (!tempWorkbook.containsKey(templateFilePath)) { InputStream inputStream = getInputStream(templateFilePath); tempWorkbook.put(templateFilePath, WorkbookFactory.create(inputStream)); } return tempWorkbook.get(templateFilePath); } /** * 功能:獲得模板輸入流 */ private InputStream getInputStream(String templateFilePath) throws FileNotFoundException { if (!tempStream.containsKey(templateFilePath)) { tempStream.put(templateFilePath, new FileInputStream((templateFilePath))); } return tempStream.get(templateFilePath); } /** * 功能:獲取單元格數據,樣式(根據坐標:B3) */ private TempCell getCell(String point, Object data, Sheet sheet) { TempCell tempCell = new TempCell(); //得到列 字母 String lineStr = ""; String reg = "[A-Z]+"; Pattern p = Pattern.compile(reg); Matcher m = p.matcher(point); while (m.find()) { lineStr = m.group(); } //將列字母轉成列號 根據ascii轉換 char[] ch = lineStr.toCharArray(); int column = 0; for (int i = 0; i < ch.length; i++) { char c = ch[i]; int post = ch.length - i - 1; int r = (int) Math.pow(10, post); column = column + r * ((int) c - 65); } tempCell.setColumn(column); //得到行號 reg = "[1-9]+"; p = Pattern.compile(reg); m = p.matcher(point); while (m.find()) { tempCell.setRow((Integer.parseInt(m.group()) - 1)); } //獲取模板指定單元格樣式,設置到tempCell(寫列表數據的時候用) Row rowIn = sheet.getRow(tempCell.getRow()); if (rowIn == null) { rowIn = sheet.createRow(tempCell.getRow()); } Cell cellIn = rowIn.getCell(tempCell.getColumn()); if (cellIn == null) { cellIn = rowIn.createCell(tempCell.getColumn()); } tempCell.setCellStyle(cellIn.getCellStyle()); tempCell.setData(data); return tempCell; } /** * 功能:給指定坐標單元格賦值 */ private void setCell(TempCell tempCell, Sheet sheet) { if (tempCell.getColumnSize() > -1) { CellRangeAddress rangeAddress = mergeRegion(sheet, tempCell.getRow(), tempCell.getRow(), tempCell.getColumn(), tempCell.getColumn() + tempCell.getColumnSize()); setRegionStyle(tempCell.getCellStyle(), rangeAddress, sheet); } Row rowIn = sheet.getRow(tempCell.getRow()); if (rowIn == null) { copyRows(tempCell.getRow() - 1, tempCell.getRow() - 1, tempCell.getRow(), sheet);//復制上一行 rowIn = sheet.getRow(tempCell.getRow()); } Cell cellIn = rowIn.getCell(tempCell.getColumn()); if (cellIn == null) { cellIn = rowIn.createCell(tempCell.getColumn()); } //根據data類型給cell賦值 if (tempCell.getData() instanceof String) { cellIn.setCellValue((String) tempCell.getData()); } else if (tempCell.getData() instanceof Integer) { cellIn.setCellValue((int) tempCell.getData()); } else if (tempCell.getData() instanceof Double) { cellIn.setCellValue((double) tempCell.getData()); } else { cellIn.setCellValue((String) tempCell.getData()); } //樣式 if (tempCell.getCellStyle() != null && tempCell.getColumnSize() == -1) { cellIn.setCellStyle(tempCell.getCellStyle()); } } /** * 功能:寫到輸出流並移除資源 */ public void writeAndClose(String templateFilePath, OutputStream os) throws IOException, InvalidFormatException { if (getTempWorkbook(templateFilePath) != null) { getTempWorkbook(templateFilePath).write(os); tempWorkbook.remove(templateFilePath); } if (getInputStream(templateFilePath) != null) { getInputStream(templateFilePath).close(); tempStream.remove(templateFilePath); } } /** * 功能:判斷指定的單元格是否是合並單元格 */ private Integer isMergedRegion(Sheet sheet, int row, int column) { for (int i = 0; i < sheet.getNumMergedRegions(); i++) { CellRangeAddress range = sheet.getMergedRegion(i); int firstColumn = range.getFirstColumn(); int lastColumn = range.getLastColumn(); int firstRow = range.getFirstRow(); int lastRow = range.getLastRow(); if (row >= firstRow && row <= lastRow) { if (column >= firstColumn && column <= lastColumn) { return i; } } } return -1; } /** * 功能:合並單元格 */ private CellRangeAddress mergeRegion(Sheet sheet, int firstRow, int lastRow, int firstCol, int lastCol) { CellRangeAddress rang = new CellRangeAddress(firstRow, lastRow, firstCol, lastCol); sheet.addMergedRegion(rang); return rang; } /** * 功能:設置合並單元格樣式 */ private void setRegionStyle(CellStyle cs, CellRangeAddress region, Sheet sheet) { for (int i = region.getFirstRow(); i <= region.getLastRow(); i++) { Row row = sheet.getRow(i); if (row == null) row = sheet.createRow(i); for (int j = region.getFirstColumn(); j <= region.getLastColumn(); j++) { Cell cell = row.getCell(j); if (cell == null) { cell = row.createCell(j); cell.setCellValue(""); } cell.setCellStyle(cs); } } } /** * 功能:copy rows */ private void copyRows(int startRow, int endRow, int pPosition, Sheet sheet) { int pStartRow = startRow - 1; int pEndRow = endRow - 1; int targetRowFrom; int targetRowTo; int columnCount; CellRangeAddress region = null; int i; int j; if (pStartRow == -1 || pEndRow == -1) { return; } // 拷貝合並的單元格 for (i = 0; i < sheet.getNumMergedRegions(); i++) { region = sheet.getMergedRegion(i); if ((region.getFirstRow() >= pStartRow) && (region.getLastRow() <= pEndRow)) { targetRowFrom = region.getFirstRow() - pStartRow + pPosition; targetRowTo = region.getLastRow() - pStartRow + pPosition; CellRangeAddress newRegion = region.copy(); newRegion.setFirstRow(targetRowFrom); newRegion.setFirstColumn(region.getFirstColumn()); newRegion.setLastRow(targetRowTo); newRegion.setLastColumn(region.getLastColumn()); sheet.addMergedRegion(newRegion); } } // 設置列寬 for (i = pStartRow; i <= pEndRow; i++) { Row sourceRow = sheet.getRow(i); columnCount = sourceRow.getLastCellNum(); if (sourceRow != null) { Row newRow = sheet.createRow(pPosition - pStartRow + i); newRow.setHeight(sourceRow.getHeight()); for (j = 0; j < columnCount; j++) { Cell templateCell = sourceRow.getCell(j); if (templateCell != null) { Cell newCell = newRow.createCell(j); copyCell(templateCell, newCell); } } } } } /** * 功能:copy cell,不copy值 */ private void copyCell(Cell srcCell, Cell distCell) { distCell.setCellStyle(srcCell.getCellStyle()); if (srcCell.getCellComment() != null) { distCell.setCellComment(srcCell.getCellComment()); } int srcCellType = srcCell.getCellType(); distCell.setCellType(srcCellType); } /** * 描述:臨時單元格數據 */ class TempCell { private int row; private int column; private CellStyle cellStyle; private Object data; //用於列表合並,表示幾列合並 private int columnSize = -1; public int getColumn() { return column; } public void setColumn(int column) { this.column = column; } public int getRow() { return row; } public void setRow(int row) { this.row = row; } public CellStyle getCellStyle() { return cellStyle; } public void setCellStyle(CellStyle cellStyle) { this.cellStyle = cellStyle; } public Object getData() { return data; } public void setData(Object data) { this.data = data; } public int getColumnSize() { return columnSize; } public void setColumnSize(int columnSize) { this.columnSize = columnSize; } } public static void main(String[] args) throws FileNotFoundException, IOException, InvalidFormatException { String templateFilePath = ExcelExportUtil.class.getClassLoader().getResource("plugin/ProTiming.xlsx").getPath(); File file = new File("/Users/sql/Downloads/test/data.xlsx"); OutputStream os = new FileOutputStream(file); ExcelExportUtil excel = new ExcelExportUtil(); Map<String, Object> dataMap = new HashMap<String, Object>(); dataMap.put("B1", "03_Alpha_項目工作時間統計表"); dataMap.put("B2", "統計時間:2017/01/01 - 2017/03/31"); excel.writeData(templateFilePath, dataMap, 0); List<Map<Integer, Object>> datalist = new ArrayList<Map<Integer, Object>>(); Map<Integer, Object> data = new HashMap<Integer, Object>(); data.put(1, "3/10/17"); data.put(2, "18:50"); data.put(3, "19:00"); data.put(4, "李子鵬"); data.put(5, "新增項目鍵值對接口,供任務計時調用"); data.put(6, "代碼開發"); data.put(7, "3.17"); datalist.add(data); data = new HashMap<Integer, Object>(); data.put(1, "3/10/17"); data.put(2, "18:50"); data.put(3, "19:00"); data.put(4, "李子鵬"); data.put(5, "新增項目鍵值對接口,供任務計時調用"); data.put(6, "代碼開發"); data.put(7, "3.17"); datalist.add(data); data = new HashMap<Integer, Object>(); data.put(1, "3/10/17"); data.put(2, "18:50"); data.put(3, "19:00"); data.put(4, "李子鵬"); data.put(5, "新增項目鍵值對接口,供任務計時調用"); data.put(6, "代碼開發"); data.put(7, "3.17"); datalist.add(data); data = new HashMap<Integer, Object>(); data.put(1, "3/10/17"); data.put(2, "18:50"); data.put(3, "19:00"); data.put(4, "李子鵬"); data.put(5, "新增項目鍵值對接口,供任務計時調用"); data.put(6, "代碼開發"); data.put(7, "3.17"); datalist.add(data); data = new HashMap<Integer, Object>(); data.put(1, "3/10/17"); data.put(2, "18:50"); data.put(3, "19:00"); data.put(4, "李子鵬"); data.put(5, "新增項目鍵值對接口,供任務計時調用"); data.put(6, "代碼開發"); data.put(7, "3.17"); datalist.add(data); data = new HashMap<Integer, Object>(); data.put(1, "3/10/17"); data.put(2, "18:50"); data.put(3, "19:00"); data.put(4, "李子鵬"); data.put(5, "新增項目鍵值對接口,供任務計時調用"); data.put(6, "代碼開發"); data.put(7, "3.17"); datalist.add(data); data = new HashMap<Integer, Object>(); data.put(1, "3/10/17"); data.put(2, "18:50"); data.put(3, "19:00"); data.put(4, "李子鵬"); data.put(5, "新增項目鍵值對接口,供任務計時調用"); data.put(6, "代碼開發"); data.put(7, "3.17"); datalist.add(data); data = new HashMap<Integer, Object>(); data.put(1, "3/10/17"); data.put(2, "18:50"); data.put(3, "19:00"); data.put(4, "李子鵬"); data.put(5, "新增項目鍵值對接口,供任務計時調用"); data.put(6, "代碼開發"); data.put(7, "3.17"); datalist.add(data); data = new HashMap<Integer, Object>(); data.put(1, "3/10/17"); data.put(2, "18:50"); data.put(3, "19:00"); data.put(4, "李子鵬"); data.put(5, "新增項目鍵值對接口,供任務計時調用新增項目鍵值對接口,供任務計時調用新增項目鍵值對接口,供任務計時調用新增項目鍵值對接口,供任務計時調用新增項目鍵值對接口,供任務計時調用新增項目鍵值對接口,供任務計時調用新增項目鍵值對接口,供任務計時調用新增項目鍵值對接口,供任務計時調用"); data.put(6, "代碼開發"); data.put(7, "3.17"); datalist.add(data); data = new HashMap<Integer, Object>(); data.put(1, ""); data.put(2, ""); data.put(3, ""); data.put(4, ""); data.put(5, ""); data.put(6, ""); data.put(7, ""); datalist.add(data); String[] heads = new String[]{"B4", "C4", "D4", "E4", "F4", "G4", "H4"}; excel.writeDateList(templateFilePath, heads, datalist, 0); //寫到輸出流並移除資源 excel.writeAndClose(templateFilePath, os); os.flush(); os.close(); } }

大體思路:

最主要是制作好模版

代碼根據模版,讀取設置好的列的格式,在循環數據行,讀取模版中的對應的行,存在該行就取得,不存在看是否需要copy某一行,不需要就手動創建無制定格式的行,後面在為該行的每一列對應的給個單元格制定格式和數據。

apache poi根據模板導出excel