| | |
| | | |
| | | 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.commons.lang3.BooleanUtils; |
| | | 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; |
| | | import org.slf4j.LoggerFactory; |
| | | import org.springframework.core.annotation.AnnotationConfigurationException; |
| | | import org.springframework.web.multipart.MultipartFile; |
| | | |
| | | import javax.servlet.http.HttpServletResponse; |
| | | import java.io.IOException; |
| | | import java.io.InputStream; |
| | | import java.io.OutputStream; |
| | | import java.lang.reflect.Constructor; |
| | | import java.lang.reflect.Field; |
| | | import java.lang.reflect.Method; |
| | | import java.math.BigDecimal; |
| | | import java.net.URLEncoder; |
| | | import java.nio.charset.Charset; |
| | | import java.text.SimpleDateFormat; |
| | | import java.util.*; |
| | | import java.util.stream.Collectors; |
| | | import java.util.stream.Stream; |
| | | |
| | | /** |
| | | * Excel导出实现 |
| | | * @author Eva.Caesar Liu |
| | | * @date 2022/04/18 18:12 |
| | | * @date 2023/02/14 11:14 |
| | | */ |
| | | @Data |
| | | public class ExcelExporter<T> { |
| | |
| | | |
| | | private ExcelExporter(){} |
| | | |
| | | |
| | | |
| | | private final static Logger log = LoggerFactory.getLogger(ExcelExporter.class); |
| | | |
| | | private final static String EXCEL2003 = "xls"; |
| | | private final static String EXCEL2007 = "xlsx"; |
| | | |
| | | /** |
| | | * 构造器 |
| | | * @param modelClass 实体Class对象 |
| | |
| | | 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(); |
| | | } |
| | | } |
| | | } |
| | | } |
| | | |
| | | /** |
| | |
| | | SXSSFWorkbook sxssfWorkbook; |
| | | try { |
| | | sxssfWorkbook = new SXSSFWorkbook(); |
| | | |
| | | CellStyle style = sxssfWorkbook.createCellStyle(); |
| | | Sheet sheet = sxssfWorkbook.createSheet(sheetName); |
| | | // 创建列头 |
| | | sheet.createFreezePane(0, 1); |
| | |
| | | sheet.setColumnWidth(columnIndex, column.columnConfig.width() * 2 * 256); |
| | | } |
| | | // 设置列头单元格 |
| | | configHeaderCell(sxssfWorkbook, cell, column.columnConfig,style); |
| | | configHeaderCell(sxssfWorkbook, cell, column.columnConfig); |
| | | } |
| | | // 创建数据记录 |
| | | for (int rowIndex = 0; rowIndex < data.size(); rowIndex++) { |
| | |
| | | ColumnInfo column = columns.get(columnIndex); |
| | | Cell cell = row.createCell(columnIndex); |
| | | cell.setCellValue(getCellData(column, data.get(rowIndex))); |
| | | |
| | | // 设置数据单元格 |
| | | configDataCell(sxssfWorkbook, cell, column.columnConfig,style); |
| | | 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(); |
| | | } |
| | | } |
| | | } |
| | | } |
| | | /** |
| | | * 导出到指定输出流 |
| | | * @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); |
| | | CellStyle cstyle = configCellStatic(sxssfWorkbook); |
| | | 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); |
| | |
| | | throw new BusinessException(ResponseStatus.EXPORT_EXCEL_ERROR, e); |
| | | } |
| | | } |
| | | |
| | | /** |
| | | * 导出至响应流 |
| | | * @param data 数据 |
| | | * @param fileName Excel文件名 |
| | | * @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); |
| | | } |
| | | |
| | | /** |
| | | * 导出至响应流 |
| | | * @param data 数据 |
| | |
| | | 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(); |
| | | } |
| | | } |
| | | } |
| | | /** |
| | | * 导出至响应流 |
| | | * @param data 数据 |
| | | * @param fileName Excel文件名 |
| | | * @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 () { |
| | | Map<Integer, ColumnInfo> sortedFields = new TreeMap<>(); |
| | | Field[] fields = modelClass.getDeclaredFields(); |
| | | int index = 0; |
| | | for (Field field : fields) { |
| | | ExcelColumn excelColumn = field.getAnnotation(ExcelColumn.class); |
| | | if (excelColumn == null) { |
| | | continue; |
| | | } |
| | | if (sortedFields.get(excelColumn.index()) != null) { |
| | | throw new AnnotationConfigurationException("Excel column contains the same index."); |
| | | } |
| | | sortedFields.put(excelColumn.index() == -1 ? index : excelColumn.index(), new ColumnInfo(excelColumn, field)); |
| | | index++; |
| | | } |
| | | return new ArrayList<>(sortedFields.values()); |
| | | } |
| | | |
| | | /** |
| | | * 配置数据单元格 |
| | | */ |
| | |
| | | style.setWrapText(true); |
| | | 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()); |
| | | // 字体 |
| | | Font font = workbook.createFont(); |
| | | font.setFontHeightInPoints(columnConfig.fontSize()); |
| | | style.setFont(font); |
| | | // 设置边框 |
| | | 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; |
| | | } |
| | | /** |
| | | * 配置列头单元格 |
| | | */ |
| | |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | // 设置背景 |
| | | style.setFillPattern(FillPatternType.SOLID_FOREGROUND); |
| | | style.setFillForegroundColor(IndexedColors.PALE_BLUE.getIndex()); |
| | | style.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex()); |
| | | // 字体 |
| | | Font font = workbook.createFont(); |
| | | font.setFontHeightInPoints((short)24); |
| | | font.setBold(true); |
| | | font.setFontHeightInPoints((short)18); |
| | | style.setFont(font); |
| | | // 设置边框 |
| | | configCellBorder(style); |
| | |
| | | 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 () { |
| | | Map<Integer, ColumnInfo> sortedFields = new TreeMap<>(); |
| | | Field[] fields = modelClass.getDeclaredFields(); |
| | | int index = 0; |
| | | for (Field field : fields) { |
| | | ExcelColumn excelColumn = field.getAnnotation(ExcelColumn.class); |
| | | if (excelColumn == null) { |
| | | continue; |
| | | } |
| | | if (sortedFields.get(excelColumn.index()) != null) { |
| | | throw new AnnotationConfigurationException("Excel column contains the same index."); |
| | | } |
| | | sortedFields.put(excelColumn.index() == -1 ? index : excelColumn.index(), new ColumnInfo(excelColumn, field)); |
| | | index++; |
| | | } |
| | | return new ArrayList<>(sortedFields.values()); |
| | | } |
| | | |
| | | public static void main(String[] args) { |
| | | ExcelExporter m = new ExcelExporter(); |
| | | // ExcelExporter.build(UserSalaryListVO.class).getColumns(); |
| | | } |
| | | /** |
| | | * 配置数据单元格 |
| | | */ |
| | | private void configDataCell (SXSSFWorkbook workbook, Cell cell, ExcelColumn columnConfig, CellStyle style) { |
| | | // 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); |
| | | cell.setCellStyle(style); |
| | | } |
| | | |
| | | /** |
| | | * 配置列头单元格 |
| | | */ |
| | | private void configHeaderCell (SXSSFWorkbook workbook, Cell cell, ExcelColumn columnConfig,CellStyle style) { |
| | | // CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(columnConfig.align()); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | // 设置背景 |
| | | style.setFillPattern(FillPatternType.SOLID_FOREGROUND); |
| | | style.setFillForegroundColor(columnConfig.backgroundColor().getIndex()); |
| | | // 字体 |
| | | Font font = workbook.createFont(); |
| | | font.setFontHeightInPoints(columnConfig.fontSize()); |
| | | 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); |
| | |
| | | |
| | | private Field field; |
| | | } |
| | | |
| | | |
| | | |
| | | public static <T> List<T> readExcel(String path, Class<T> cls, MultipartFile file) { |
| | | |
| | | String fileName = file.getOriginalFilename(); |
| | | if (!fileName.matches("^.+\\.(?i)(xls)$") && !fileName.matches("^.+\\.(?i)(xlsx)$")) { |
| | | log.error("上传文件格式不正确"); |
| | | } |
| | | List<T> dataList = new ArrayList<>(); |
| | | Workbook workbook = null; |
| | | try { |
| | | InputStream is = file.getInputStream(); |
| | | if (fileName.endsWith(EXCEL2007)) { |
| | | workbook = new XSSFWorkbook(is); |
| | | } |
| | | if (fileName.endsWith(EXCEL2003)) { |
| | | workbook = new HSSFWorkbook(is); |
| | | } |
| | | if (workbook != null) { |
| | | //类映射 注解 value-->bean columns |
| | | Map<String, List<Field>> classMap = new HashMap<>(); |
| | | List<Field> fields = Stream.of(cls.getDeclaredFields()).collect(Collectors.toList()); |
| | | fields.forEach( |
| | | field -> { |
| | | ExcelColumn annotation = field.getAnnotation(ExcelColumn.class); |
| | | if (annotation != null) { |
| | | String value = annotation.name(); |
| | | if (StringUtils.isBlank(value)) { |
| | | return;//return起到的作用和continue是相同的 语法 |
| | | } |
| | | if (!classMap.containsKey(value)) { |
| | | classMap.put(value, new ArrayList<>()); |
| | | } |
| | | field.setAccessible(true); |
| | | classMap.get(value).add(field); |
| | | } |
| | | } |
| | | ); |
| | | //索引-->columns |
| | | Map<Integer, List<Field>> reflectionMap = new HashMap<>(16); |
| | | //默认读取第一个sheet |
| | | Sheet sheet = workbook.getSheetAt(0); |
| | | |
| | | boolean firstRow = true; |
| | | for (int i = sheet.getFirstRowNum(); i <= sheet.getLastRowNum(); i++) { |
| | | Row row = sheet.getRow(i); |
| | | //首行 提取注解 |
| | | if (firstRow) { |
| | | for (int j = row.getFirstCellNum(); j <= row.getLastCellNum(); j++) { |
| | | Cell cell = row.getCell(j); |
| | | String cellValue = getCellValue(cell); |
| | | if (classMap.containsKey(cellValue)) { |
| | | reflectionMap.put(j, classMap.get(cellValue)); |
| | | } |
| | | } |
| | | firstRow = false; |
| | | } else { |
| | | //忽略空白行 |
| | | if (row == null) { |
| | | continue; |
| | | } |
| | | try { |
| | | T t = cls.newInstance(); |
| | | //判断是否为空白行 |
| | | boolean allBlank = true; |
| | | for (int j = row.getFirstCellNum(); j <= row.getLastCellNum(); j++) { |
| | | if (reflectionMap.containsKey(j)) { |
| | | Cell cell = row.getCell(j); |
| | | String cellValue = getCellValue(cell); |
| | | if (StringUtils.isNotBlank(cellValue)) { |
| | | allBlank = false; |
| | | } |
| | | List<Field> fieldList = reflectionMap.get(j); |
| | | fieldList.forEach( |
| | | x -> { |
| | | try { |
| | | handleField(t, cellValue, x); |
| | | } catch (Exception e) { |
| | | log.error(String.format("reflect field:%s value:%s exception!", x.getName(), cellValue), e); |
| | | } |
| | | } |
| | | ); |
| | | } |
| | | } |
| | | if (!allBlank) { |
| | | dataList.add(t); |
| | | } else { |
| | | log.warn(String.format("row:%s is blank ignore!", i)); |
| | | } |
| | | } catch (Exception e) { |
| | | log.error(String.format("parse row:%s exception!", i), e); |
| | | } |
| | | } |
| | | } |
| | | } |
| | | } catch (Exception e) { |
| | | log.error(String.format("parse excel exception!"), e); |
| | | } finally { |
| | | if (workbook != null) { |
| | | try { |
| | | workbook.close(); |
| | | } catch (Exception e) { |
| | | log.error(String.format("parse excel exception!"), e); |
| | | } |
| | | } |
| | | } |
| | | return dataList; |
| | | } |
| | | |
| | | private static <T> void handleField(T t, String value, Field field) throws Exception { |
| | | Class<?> type = field.getType(); |
| | | if (type == null || type == void.class || StringUtils.isBlank(value)) { |
| | | return; |
| | | } |
| | | if (type == Object.class) { |
| | | field.set(t, value); |
| | | //数字类型 |
| | | } else if (type.getSuperclass() == null || type.getSuperclass() == Number.class) { |
| | | if (type == int.class || type == Integer.class) { |
| | | field.set(t, NumberUtils.toInt(value)); |
| | | } else if (type == long.class || type == Long.class) { |
| | | field.set(t, NumberUtils.toLong(value)); |
| | | } else if (type == byte.class || type == Byte.class) { |
| | | field.set(t, NumberUtils.toByte(value)); |
| | | } else if (type == short.class || type == Short.class) { |
| | | field.set(t, NumberUtils.toShort(value)); |
| | | } else if (type == double.class || type == Double.class) { |
| | | field.set(t, NumberUtils.toDouble(value)); |
| | | } else if (type == float.class || type == Float.class) { |
| | | field.set(t, NumberUtils.toFloat(value)); |
| | | } else if (type == char.class || type == Character.class) { |
| | | field.set(t, CharUtils.toChar(value)); |
| | | } else if (type == boolean.class) { |
| | | field.set(t, BooleanUtils.toBoolean(value)); |
| | | } else if (type == BigDecimal.class) { |
| | | field.set(t, new BigDecimal(value)); |
| | | } |
| | | } else if (type == Boolean.class) { |
| | | field.set(t, BooleanUtils.toBoolean(value)); |
| | | } else if (type == Date.class) { |
| | | field.set(t, value); |
| | | } else if (type == String.class) { |
| | | field.set(t, value); |
| | | } else { |
| | | Constructor<?> constructor = type.getConstructor(String.class); |
| | | field.set(t, constructor.newInstance(value)); |
| | | } |
| | | } |
| | | |
| | | private static String getCellValue(Cell cell) { |
| | | if (cell == null) { |
| | | return ""; |
| | | }else{ |
| | | return StringUtils.trimToEmpty(cell.getStringCellValue()); |
| | | } |
| | | /* if (cell.getCellType() == Cell.CELL_TYPE_NUMERIC) { |
| | | if (HSSFDateUtil.isCellDateFormatted(cell)) { |
| | | return HSSFDateUtil.getJavaDate(cell.getNumericCellValue()).toString(); |
| | | } else { |
| | | return new BigDecimal(cell.getNumericCellValue()).toString(); |
| | | } |
| | | } else if (cell.getCellType() == Cell.CELL_TYPE_STRING) { |
| | | return StringUtils.trimToEmpty(cell.getStringCellValue()); |
| | | } else if (cell.getCellType() == Cell.CELL_TYPE_FORMULA) { |
| | | return StringUtils.trimToEmpty(cell.getCellFormula()); |
| | | } else if (cell.getCellType() == Cell.CELL_TYPE_BLANK) { |
| | | return ""; |
| | | } else if (cell.getCellType() == Cell.CELL_TYPE_BOOLEAN) { |
| | | return String.valueOf(cell.getBooleanCellValue()); |
| | | } else if (cell.getCellType() == Cell.CELL_TYPE_ERROR) { |
| | | return "ERROR"; |
| | | } else { |
| | | return cell.toString().trim(); |
| | | }*/ |
| | | } |
| | | |
| | | /* public static <T> void writeExcel(HttpServletResponse response, List<T> dataList, Class<T> cls, String fileName) { |
| | | Field[] fields = cls.getDeclaredFields(); |
| | | List<Field> fieldList = Arrays.stream(fields) |
| | | .filter(field -> { |
| | | ExcelColumn annotation = field.getAnnotation(ExcelColumn.class); |
| | | if (annotation != null && annotation.col() > 0) { |
| | | field.setAccessible(true); |
| | | return true; |
| | | } |
| | | return false; |
| | | }).sorted(Comparator.comparing(field -> { |
| | | int col = 0; |
| | | ExcelColumn annotation = field.getAnnotation(ExcelColumn.class); |
| | | if (annotation != null) { |
| | | col = annotation.col(); |
| | | } |
| | | return col; |
| | | })).collect(Collectors.toList()); |
| | | |
| | | Workbook wb = new XSSFWorkbook(); |
| | | Sheet sheet = wb.createSheet("Sheet1"); |
| | | AtomicInteger ai = new AtomicInteger(); |
| | | { |
| | | Row row = sheet.createRow(ai.getAndIncrement()); |
| | | AtomicInteger aj = new AtomicInteger(); |
| | | //写入头部 |
| | | fieldList.forEach(field -> { |
| | | ExcelColumn annotation = field.getAnnotation(ExcelColumn.class); |
| | | String columnName = ""; |
| | | if (annotation != null) { |
| | | columnName = annotation.value(); |
| | | } |
| | | Cell cell = row.createCell(aj.getAndIncrement()); |
| | | CellStyle cellStyle = wb.createCellStyle(); |
| | | cellStyle.setFillForegroundColor(IndexedColors.WHITE.getIndex()); |
| | | Font font = wb.createFont(); |
| | | cellStyle.setFont(font); |
| | | cell.setCellStyle(cellStyle); |
| | | cell.setCellValue(columnName); |
| | | }); |
| | | } |
| | | if (CollectionUtils.isNotEmpty(dataList)) { |
| | | dataList.forEach(t -> { |
| | | Row row1 = sheet.createRow(ai.getAndIncrement()); |
| | | AtomicInteger aj = new AtomicInteger(); |
| | | fieldList.forEach(field -> { |
| | | Class<?> type = field.getType(); |
| | | Object value = ""; |
| | | try { |
| | | value = field.get(t); |
| | | } catch (Exception e) { |
| | | e.printStackTrace(); |
| | | } |
| | | Cell cell = row1.createCell(aj.getAndIncrement()); |
| | | if (value != null) { |
| | | if (type == Date.class) { |
| | | cell.setCellValue(value.toString()); |
| | | } else { |
| | | cell.setCellValue(value.toString()); |
| | | } |
| | | cell.setCellValue(value.toString()); |
| | | } |
| | | }); |
| | | }); |
| | | } |
| | | //冻结窗格 |
| | | wb.getSheet("Sheet1").createFreezePane(0, 1, 0, 1); |
| | | //浏览器下载excel |
| | | buildExcelDocument(fileName, wb, response); |
| | | }*/ |
| | | |
| | | /** |
| | | * 浏览器下载excel |
| | | * |
| | | * @param fileName |
| | | * @param wb |
| | | * @param response |
| | | */ |
| | | /* |
| | | private static void buildExcelDocument(String fileName, Workbook wb, HttpServletResponse response) { |
| | | try { |
| | | response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); |
| | | response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "utf-8")); |
| | | response.flushBuffer(); |
| | | wb.write(response.getOutputStream()); |
| | | } catch (IOException e) { |
| | | e.printStackTrace(); |
| | | } |
| | | }*/ |
| | | |
| | | /** |
| | | * 生成excel文件 |
| | | * |
| | | * @param path 生成excel路径 |
| | | * @param wb |
| | | */ |
| | | /* private static void buildExcelFile(String path, Workbook wb) { |
| | | File file = new File(path); |
| | | if (file.exists()) { |
| | | file.delete(); |
| | | } |
| | | try { |
| | | wb.write(new FileOutputStream(file)); |
| | | } catch (Exception e) { |
| | | e.printStackTrace(); |
| | | } |
| | | }*/ |
| | | |
| | | } |