| | |
| | | package com.doumee.core.annotation.excel; |
| | | |
| | | import com.doumee.core.constants.Constants; |
| | | import com.doumee.core.constants.ResponseStatus; |
| | | import com.doumee.core.exception.BusinessException; |
| | | import com.doumee.dao.business.model.Goodsorder; |
| | | import lombok.AllArgsConstructor; |
| | | import lombok.Data; |
| | | import org.apache.poi.ss.usermodel.*; |
| | | import org.apache.poi.ss.util.CellRangeAddress; |
| | | import org.apache.poi.xssf.streaming.SXSSFWorkbook; |
| | | import org.springframework.core.annotation.AnnotationConfigurationException; |
| | | |
| | |
| | | /** |
| | | * Excel导出实现 |
| | | * @author Eva.Caesar Liu |
| | | * @date 2022/03/15 09:54 |
| | | * @date 2023/02/14 11:14 |
| | | */ |
| | | @Data |
| | | public class ExcelExporter<T> { |
| | |
| | | ExcelExporter<T> excelExporter = new ExcelExporter<>(); |
| | | excelExporter.setModelClass(modelClass); |
| | | return excelExporter; |
| | | } |
| | | /** |
| | | * 导出到指定输出流 |
| | | * @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); |
| | | 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(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); |
| | | |
| | | } |
| | | } |
| | | |
| | | |
| | | 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(); |
| | | } |
| | | } |
| | | } |
| | | } |
| | | |
| | | /** |
| | |
| | | ColumnInfo column = columns.get(columnIndex); |
| | | Cell cell = row.createCell(columnIndex); |
| | | cell.setCellValue(getCellData(column, data.get(rowIndex))); |
| | | |
| | | // 设置数据单元格 |
| | | configDataCell(sxssfWorkbook, cell, column.columnConfig); |
| | | |
| | | } |
| | | } |
| | | 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(); |
| | | } |
| | | } |
| | | } |
| | | } |
| | | |
| | | // public void exportList (List<List<String>> data, String fileName, String sheetName, 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.exportList(data, sheetName, response.getOutputStream()); |
| | | // } catch (IOException e) { |
| | | // throw new BusinessException(ResponseStatus.EXPORT_EXCEL_ERROR, e); |
| | | // } |
| | | // } |
| | | |
| | | |
| | | /** |
| | | * 导出到指定输出流 |
| | | * @param os 输出流 |
| | | */ |
| | | public static void exportList (List<List<String>> dataList ,String sheetName, OutputStream os) { |
| | | SXSSFWorkbook sxssfWorkbook; |
| | | try { |
| | | sxssfWorkbook = new SXSSFWorkbook(); |
| | | Sheet sheet = sxssfWorkbook.createSheet(sheetName); |
| | | // 创建列头 |
| | | sheet.createFreezePane(0, 1); |
| | | Row header = sheet.createRow(0); |
| | | CellStyle hstyle = configHeaderCellStatic(sxssfWorkbook); |
| | | for (int i = 0; i < dataList.size(); i++) { |
| | | Cell cell = header.createCell(i); |
| | | cell.setCellValue(dataList.get(i).get(Constants.ZERO)); |
| | | // 列宽设置 |
| | | sheet.setColumnWidth(i, dataList.get(i).get(Constants.ZERO).length() * 2 * 256); |
| | | // 设置列头单元格 |
| | | cell.setCellStyle(hstyle); |
| | | } |
| | | |
| | | //总行数 |
| | | Integer rowSize = dataList.get(Constants.ZERO).size(); |
| | | //总列数 |
| | | Integer columnSize = dataList.size(); |
| | | for (int i = 1; i < rowSize; i++) { |
| | | Row row = sheet.createRow(i); |
| | | for (int j = 0; j < columnSize; j++) { |
| | | Cell cell = row.createCell(j); |
| | | cell.setCellValue(dataList.get(j).get(i)); |
| | | } |
| | | } |
| | | // |
| | | // List<String> headerList =dataList.get(0); |
| | | // for (int i = 0; i < headerList.size(); i++) { |
| | | // Cell cell = header.createCell(i); |
| | | // cell.setCellValue(headerList.get(i)); |
| | | // // 列宽设置 |
| | | // sheet.setColumnWidth(i, headerList.get(i).length() * 2 * 256); |
| | | // // 设置列头单元格 |
| | | // cell.setCellStyle(hstyle); |
| | | // } |
| | | // // 创建数据记录 |
| | | // for (int rowIndex = 1; rowIndex < dataList.size(); rowIndex++) { |
| | | // Row row = sheet.createRow(rowIndex ); |
| | | // List<String> rowList = dataList.get(rowIndex); |
| | | // for (int i = 0; i < rowList.size(); i++) { |
| | | // Cell cell = row.createCell(i); |
| | | // cell.setCellValue(rowList.get(i)); |
| | | // // 列宽设置 |
| | | // cell.setCellStyle(cstyle); |
| | | // } |
| | | // } |
| | | sxssfWorkbook.write(os); |
| | | os.close(); |
| | | } catch (Exception e) { |
| | |
| | | throw new BusinessException(ResponseStatus.EXPORT_EXCEL_ERROR, e); |
| | | } |
| | | } |
| | | /** |
| | | * 导出至响应流 |
| | | * @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 response HttpServletResponse对象 |
| | | */ |
| | | 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); |
| | | } |
| | | |
| | | /**O |
| | | * 获取列集合 |
| | | */ |
| | | private List<ColumnInfo> getColumns () { |
| | |
| | | style.setFont(font); |
| | | // 边框 |
| | | configCellBorder(style); |
| | | style.setWrapText(true); |
| | | cell.setCellStyle(style); |
| | | } |
| | | |
| | |
| | | configCellBorder(style); |
| | | cell.setCellStyle(style); |
| | | } |
| | | private static CellStyle configHeaderCellStatic (SXSSFWorkbook workbook) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(HorizontalAlignment.CENTER); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | // 设置背景 |
| | | style.setFillPattern(FillPatternType.SOLID_FOREGROUND); |
| | | style.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex()); |
| | | // 字体 |
| | | Font font = workbook.createFont(); |
| | | font.setFontHeightInPoints((short) 12); |
| | | style.setFont(font); |
| | | // 设置边框 |
| | | configCellBorder(style); |
| | | return style; |
| | | } |
| | | private static CellStyle configCellStatic (SXSSFWorkbook workbook) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(HorizontalAlignment.CENTER); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | // 设置背景 |
| | | style.setFillPattern(FillPatternType.SOLID_FOREGROUND); |
| | | // 字体 |
| | | Font font = workbook.createFont(); |
| | | font.setFontHeightInPoints((short) 12); |
| | | style.setFont(font); |
| | | // 设置边框 |
| | | configCellBorder(style); |
| | | return 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.GREY_25_PERCENT.getIndex()); |
| | | // 字体 |
| | | Font font = workbook.createFont(); |
| | | font.setFontHeightInPoints((short)18); |
| | | 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 configCellBorder (CellStyle style) { |
| | | private static void configCellBorder (CellStyle style) { |
| | | style.setBorderTop(BorderStyle.THIN); |
| | | style.setBorderRight(BorderStyle.THIN); |
| | | style.setBorderBottom(BorderStyle.THIN); |