POI 操作 Excel 工具

poi 3.8 及以上版本使用 SXSSFWorkbook 性能会更高点，内存占用会更少

• HSSF：是操作 Excel97-2003 版本，扩展名为.xls。
• XSSF：是操作 Excel2007 版本开始，扩展名为.xlsx。
• SXSSF：是在 XSSF 基础上，POI3.8 版本开始提供的一种支持低内存占用的操作方式，扩展名为.xlsx。

• SXSSFWorkbook 需要指定的字体库支持。

ExcelUtils.java

import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.DateUtil;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

/**
 * poi 3.8 及以上版本使用 SXSSFWorkbook 性能会更高点，内存占用会更少
 * <pre>
 *     HSSF：是操作 Excel97-2003 版本，扩展名为.xls。
 *     XSSF：是操作 Excel2007 版本开始，扩展名为.xlsx。
 *     SXSSF：是在 XSSF 基础上，POI3.8 版本开始提供的一种支持低内存占用的操作方式，扩展名为.xlsx。
 * </pre>
 *
 * @author yangli
 */
public class ExcelUtils {
    private static final String XLS = ".xls";
    private static final String XLSX = ".xlsx";

    /**
     * 读取 excel 文件，使用SXSSFWorkbook
     *
     * @param file           MultipartFile
     * @param rowIgnoreCount 略过空行和前 <code>rowIgnoreCount</code> 行
     * @return 列表
     */
    public static List<Map<String, String>> read(MultipartFile file, int rowIgnoreCount) {
        InputStream is = null;
        Workbook wb = null;
        try {
            is = file.getInputStream();
            if (file.getOriginalFilename().toLowerCase().endsWith(XLS)) {
                wb = new HSSFWorkbook(is);
            } else if (file.getOriginalFilename().toLowerCase().endsWith(XLSX)) {
                // wb = new XSSFWorkbook(is);
                // 压缩临时文件
                wb = new SXSSFWorkbook(new XSSFWorkbook(is), 100, true);
            } else {
                throw new RuntimeException("不支持该 Excel 类型");
            }
            if (wb instanceof HSSFWorkbook) {
                return ExcelUtils.read(wb, rowIgnoreCount);
            } else {
                // 如果使用 SXSSFWorkbook，需要 getXSSFWorkbook，不然会读取为空
                return ExcelUtils.read(((SXSSFWorkbook) wb).getXSSFWorkbook(), rowIgnoreCount);
            }
            // return ExcelUtils.read(wb, rowIgnoreCount);
        } catch (IOException e) {
            throw new RuntimeException("文件处理异常[" + e.getMessage() + "]");
        } finally {
            if (wb != null) {
                if (wb instanceof SXSSFWorkbook) {
                    // 删除临时文件，否则磁盘可能会被写满
                    ((SXSSFWorkbook) wb).dispose();
                }
            }
            IOUtils.closeQuietly(is);
            IOUtils.closeQuietly(wb);
        }
    }

    /**
     * 读取 excel 文件
     *
     * @param wb             Workbook
     * @param rowIgnoreCount 略过空行和前 <code>rowIgnoreCount</code> 行
     * @return 列表
     */
    public static List<Map<String, String>> read(Workbook wb, int rowIgnoreCount) {
        List<Map<String, String>> list = new ArrayList<>();
        int sheetSize = wb.getNumberOfSheets();
        // 遍历 sheet 页
        for (int i = 0; i < sheetSize; i++) {
            Sheet sheet = wb.getSheetAt(i);
            int rowSize = sheet.getLastRowNum() + 1;
            // 遍历行
            for (int j = 1; j < rowSize; j++) {
                Row row = sheet.getRow(j);
                // 略过空行和前 rowIgnoreCount 行
                if (row == null || j < rowIgnoreCount) {
                    continue;
                }
                short lastCellNum = row.getLastCellNum();
                if (lastCellNum > 100) {
                    throw new RuntimeException("列数超过 50");
                }
                boolean isAllBlank = true;
                Map<String, String> cell = new LinkedHashMap<>();
                for (int cellNum = 0; cellNum < lastCellNum; cellNum++) {
                    String value = ExcelUtils.getValue(row, cellNum);
                    cell.put(String.valueOf(cellNum), value);
                    if (StringUtils.isNotBlank(value)) {
                        isAllBlank = false;
                    }
                }
                // 如果该行单元格全部为空，则过滤掉该行数据
                if (isAllBlank) {
                    continue;
                }
                list.add(cell);
            }
        }
        return list;
    }

    /**
     * 根据数据格式返回数据
     */
    private static String getValue(Row row, int cellNum) {
        Cell cell = row.getCell(cellNum);
        if (cell == null) {
            return "";
        } else {
            CellType cellType = cell.getCellType();
            if (cellType == CellType.BOOLEAN) {
                return String.valueOf(cell.getBooleanCellValue());
            } else if (cellType == CellType.NUMERIC) {
                if (DateUtil.isCellDateFormatted(cell)) {
                    return cell.getLocalDateTimeCellValue().format(DateTimeFormatter.ofPattern("yyyy-MM-dd"));
                }
                // if ("General".equals(cell.getCellStyle().getDataFormatString())) {
                //     return new DecimalFormat("0").format(cell.getNumericCellValue());
                // } else if ("m/d/yy".equals(cell.getCellStyle().getDataFormatString())) {
                //     return new SimpleDateFormat("yyyy-MM-dd").format(cell.getDateCellValue());
                // } else {
                //     return new DecimalFormat("0").format(cell.getNumericCellValue());
                // }
                if ("m/d/yy".equals(cell.getCellStyle().getDataFormatString())) {
                    return new SimpleDateFormat("yyyy-MM-dd").format(cell.getDateCellValue());
                } else {
                    double value = cell.getNumericCellValue();
                    String strValue = String.valueOf(value);
                    String[] split = strValue.split("\\.", -1);
                    if (split.length >= 2) {
                        String s = split[split.length - 1];
                        // 最后一位为 0 时
                        if ("0".equals(s)) {
                            return new DecimalFormat("0").format(value);
                        }
                    }
                    return strValue;
                }
            } else if (cellType == CellType.FORMULA) {
                return cell.getCellFormula();
            } else if (cellType == CellType.BLANK) {
                return "";
            } else {
                return cell.getStringCellValue().trim();
            }
        }
    }
}