| | |
| | | import org.apache.commons.lang3.CharUtils; |
| | | import org.apache.commons.lang3.StringUtils; |
| | | import org.apache.commons.lang3.math.NumberUtils; |
| | | import org.apache.poi.hssf.usermodel.HSSFRichTextString; |
| | | import org.apache.poi.hssf.usermodel.HSSFWorkbook; |
| | | import org.apache.poi.ss.usermodel.*; |
| | | import org.apache.poi.ss.util.CellRangeAddress; |
| | | import org.apache.poi.xssf.streaming.SXSSFWorkbook; |
| | | import org.apache.poi.xssf.usermodel.XSSFWorkbook; |
| | | import org.slf4j.Logger; |
| | |
| | | public void export (List<T> data, String fileName, HttpServletResponse response) { |
| | | this.export(data, fileName, DEFAULT_SHEET_NAME, response); |
| | | } |
| | | /** |
| | | * 导出至响应流 |
| | | * @param data 数据 |
| | | * @param fileName Excel文件名 |
| | | * @param response HttpServletResponse对象 |
| | | */ |
| | | public void exportWithFirstAndEnd (List<T> data, String fileName,String first,String end, HttpServletResponse response) { |
| | | this.exportWithFirstAndEnd(data, fileName, DEFAULT_SHEET_NAME, first,end,response); |
| | | } |
| | | |
| | | /** |
| | | * 导出至响应流 |
| | | * @param data 数据 |
| | | * @param fileName Excel文件名 |
| | | * @param sheetName Sheet名称 |
| | | * @param response HttpServletResponse对象 |
| | | */ |
| | | public void exportWithFirstAndEnd (List<T> data, String fileName, String sheetName, String first,String end ,HttpServletResponse response) { |
| | | try { |
| | | String encodeFileName = URLEncoder.encode(fileName, Charset.forName("UTF-8").toString()) + ".xlsx"; |
| | | response.setHeader("Content-Disposition","attachment;filename=" + encodeFileName); |
| | | response.setContentType("application/octet-stream"); |
| | | response.setHeader("eva-opera-type", "download"); |
| | | response.setHeader("eva-download-filename", encodeFileName); |
| | | this.exportWithFirstAndEnd(data, sheetName,first,end, response.getOutputStream()); |
| | | } catch (IOException e) { |
| | | throw new BusinessException(ResponseStatus.EXPORT_EXCEL_ERROR, e); |
| | | } |
| | | } |
| | | /** |
| | | * 导出到指定输出流 |
| | | * @param data 数据 |
| | | * @param sheetName Sheet名称 |
| | | */ |
| | | public void exportWithFirstAndEnd (List<T> data, String sheetName,String first,String end, OutputStream os) { |
| | | SXSSFWorkbook sxssfWorkbook; |
| | | try { |
| | | sxssfWorkbook = new SXSSFWorkbook(); |
| | | Sheet sheet = sxssfWorkbook.createSheet(sheetName); |
| | | // 创建列头 |
| | | sheet.createFreezePane(0, 2); |
| | | sheet.addMergedRegion(new CellRangeAddress(0 ,0,0,this.getColumns().size()-1)); |
| | | Row title = sheet.createRow(0); |
| | | title.setHeight((short) 1000); |
| | | Cell c = title.createCell(0); |
| | | c.setCellValue(first); |
| | | configFirstCell(sxssfWorkbook,c); |
| | | |
| | | Row header = sheet.createRow(1); |
| | | header.setHeight((short)600); |
| | | List<ColumnInfo> columns = this.getColumns(); |
| | | for (int columnIndex = 0; columnIndex < columns.size(); columnIndex++) { |
| | | ColumnInfo column = columns.get(columnIndex); |
| | | Cell cell = header.createCell(columnIndex); |
| | | cell.setCellValue(new HSSFRichTextString(column.columnConfig.name())); |
| | | // 列宽设置 |
| | | if (column.columnConfig.width() == -1) { |
| | | sheet.setColumnWidth(columnIndex, column.columnConfig.name().length() * 2 * 256); |
| | | } else { |
| | | sheet.setColumnWidth(columnIndex, column.columnConfig.width() * 2 * 256); |
| | | } |
| | | // 设置列头单元格 |
| | | configHeaderCell(sxssfWorkbook, cell, column.columnConfig); |
| | | } |
| | | // 创建数据记录 |
| | | for (int rowIndex = 0; rowIndex < data.size(); rowIndex++) { |
| | | Row row = sheet.createRow(rowIndex + 2); |
| | | for (int columnIndex = 0; columnIndex < columns.size(); columnIndex++) { |
| | | ColumnInfo column = columns.get(columnIndex); |
| | | Cell cell = row.createCell(columnIndex); |
| | | cell.setCellValue(getCellData(column, data.get(rowIndex))); |
| | | // 设置数据单元格 |
| | | configDataCell(sxssfWorkbook, cell, column.columnConfig); |
| | | } |
| | | } |
| | | if(StringUtils.isNotBlank(end)){ |
| | | sheet.addMergedRegion(new CellRangeAddress(data.size()+2 ,data.size()+2,0,this.getColumns().size()-1)); |
| | | Row endRow = sheet.createRow(data.size()+2); |
| | | // endRow.setHeight((short) 600); |
| | | Cell c1 = endRow.createCell(0); |
| | | c1.setCellValue(end); |
| | | configEndCell(sxssfWorkbook,c1); |
| | | } |
| | | sxssfWorkbook.write(os); |
| | | os.close(); |
| | | } catch (Exception e) { |
| | | throw new BusinessException(ResponseStatus.EXPORT_EXCEL_ERROR, e); |
| | | } finally { |
| | | if (os != null) { |
| | | try { |
| | | os.close(); |
| | | } catch (IOException e) { |
| | | e.printStackTrace(); |
| | | } |
| | | } |
| | | } |
| | | } |
| | | /** |
| | | * 配置数据单元格 |
| | | */ |
| | | private void configDataCell (SXSSFWorkbook workbook, Cell cell, ExcelColumn columnConfig) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(columnConfig.align()); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | // 设置背景 |
| | | style.setFillPattern(FillPatternType.SOLID_FOREGROUND); |
| | | style.setFillForegroundColor(columnConfig.dataBackgroundColor().getIndex()); |
| | | // 字体 |
| | | Font font = workbook.createFont(); |
| | | font.setFontHeightInPoints(columnConfig.fontSize()); |
| | | // 字体颜色 |
| | | font.setColor(columnConfig.color().getIndex()); |
| | | // 粗体 |
| | | font.setBold(columnConfig.bold()); |
| | | // 斜体 |
| | | font.setItalic(columnConfig.italic()); |
| | | style.setFont(font); |
| | | // 边框 |
| | | configCellBorder(style); |
| | | style.setWrapText(true); |
| | | cell.setCellStyle(style); |
| | | } |
| | | /** |
| | | * 配置列头单元格 |
| | | */ |
| | | private void configFirstCell (SXSSFWorkbook workbook, Cell cell ) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(HorizontalAlignment.CENTER); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | // 设置背景 |
| | | style.setFillPattern(FillPatternType.SOLID_FOREGROUND); |
| | | style.setFillForegroundColor(IndexedColors.PALE_BLUE.getIndex()); |
| | | // 字体 |
| | | Font font = workbook.createFont(); |
| | | font.setFontHeightInPoints((short)24); |
| | | font.setBold(true); |
| | | style.setFont(font); |
| | | // 设置边框 |
| | | configCellBorder(style); |
| | | cell.setCellStyle(style); |
| | | } |
| | | /** |
| | | * 配置列头单元格 |
| | | */ |
| | | private void configEndCell (SXSSFWorkbook workbook, Cell cell ) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(HorizontalAlignment.RIGHT); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | // 设置背景 |
| | | style.setFillPattern(FillPatternType.SOLID_FOREGROUND); |
| | | style.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex()); |
| | | // 字体 |
| | | Font font = workbook.createFont(); |
| | | font.setFontHeightInPoints((short)14); |
| | | style.setFont(font); |
| | | // 设置边框 |
| | | configCellBorder(style); |
| | | cell.setCellStyle(style); |
| | | } |
| | | /** |
| | | * 配置列头单元格 |
| | | */ |
| | | private void configHeaderCell (SXSSFWorkbook workbook, Cell cell, ExcelColumn columnConfig) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(columnConfig.align()); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | // 设置背景 |
| | | style.setFillPattern(FillPatternType.SOLID_FOREGROUND); |
| | | // style.setFillForegroundColor(columnConfig.backgroundColor().getIndex()); |
| | | style.setFillForegroundColor(IndexedColors.LIGHT_CORNFLOWER_BLUE.getIndex()); |
| | | // 字体 |
| | | Font font = workbook.createFont(); |
| | | font.setFontHeightInPoints((short)11); |
| | | font.setColor(columnConfig.color().index); |
| | | font.setBold(true); |
| | | style.setFont(font); |
| | | // 设置边框 |
| | | style.setWrapText(true); |
| | | configCellBorder(style); |
| | | cell.setCellStyle(style); |
| | | } |
| | | /** |
| | | * 获取列集合 |
| | | */ |
| | | private List<ColumnInfo> getColumns () { |