java, poi, excel

时间:2023-03-08 15:53:34

工作需要用java操作Excel,现在网上搜索了一下,决定选取POI包来操作。pom内容如下:

        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.8</version>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.8</version>
        </dependency>

两个版本需要一致,否则会出错。

excel 03和07版本对应不同的操作类,分别是HSSFWorkbook和XSSFWorkbook。他们都继承自接口Workbook,所以在打开文件的时候,判断是哪一个版本,然后使用Workbook操作:

private static Workbook createWorkBook(String path) {
        try {
            InputStream in = new FileInputStream(path);
            if (!in.markSupported()) {
                in = new PushbackInputStream(in, 8);
            }
            if (POIFSFileSystem.hasPOIFSHeader(in)) {
                return new HSSFWorkbook(in);
            }
            if (POIXMLDocument.hasOOXMLHeader(in)) {
                return new XSSFWorkbook(OPCPackage.open(in));
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (InvalidFormatException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        throw new IllegalArgumentException("打开excel文件(" + path + ")错误");
    }

相应的Sheet,Row 和 Cell也都使用最上层的接口类:

private static Sheet getSheetByName(Workbook workbook, String s) {
        Sheet sheet = workbook.getSheet(s);
        if (sheet == null)
            sheet = workbook.createSheet(s);
        return sheet;
    }

private static Row getRowFromSheet(Sheet sheet, int rowNum) {
        Row row;
        row = sheet.getRow(rowNum);
        if (row != null)
            return row;
        else {
            row = sheet.createRow(rowNum);
            return row;
        }
    }

以Sheet为例, workbook.getSheet(i)是获得第i个sheet表单,如果不存在,则返回空。 workbook.createSheet(i)是创建表单,会覆盖原有表单。

从Cell中读取数据,有时候使用poi读取出来的单元格式会和在excel中看到的不一样。所以在读取数据时需要多判断一下类型,在poi源码的基础上修改一下:

private static String handleCellV1(Cell cell, int type) { // 读取日期形式单元格数据,返回字符串数据
        String var1 = null;
        int cellType = cell.getCellType();
        switch (cellType) {
            case 0: // numeric
                if (DateUtil.isCellDateFormatted(cell)) { // 如果是日期格式
                    SimpleDateFormat sdf = null;
                    if (type == 1) {
                        sdf = new SimpleDateFormat(DATE); // String DATE = "yyyy-MM-dd";
                    } else {
                        sdf = new SimpleDateFormat(TIME);
                    }
                    Date date = cell.getDateCellValue();
                    var1 = sdf.format(cell.getDateCellValue());
                } else {                                        // 有时候也会变成数字格式
                    String res = String.valueOf(cell.getNumericCellValue());
                    BigDecimal bd = new BigDecimal(res);    // 数字格式在excel中都是double,所以需要去掉小数点后的数(直接int??)
                    res = bd.toPlainString();
                    String[] ss = res.split("\\.");

                    res = ss[0];
                    var1 = res;
                }
                break;
            case 1: // text
                var1 = cell.getStringCellValue();
                break;
            default: // #unknown cell type
                var1 = "Unknown Cell Type: " + cell.getCellType();
                var1 = null;
        }
        return var1;
    }

写入excel单元格数据,可以使普通的写入:

Row row = sheet.createRow(rowNum);
row.createCell().setCellValue("日期");

也可以写入公式:

row.createCell().setCellFormula("IF(B" + rowNum + "=\"初始\",(SUMIFS(F:F,E:E,E" + rowNum + ",C:C,C" + rowNum + ",D:D,D" + rowNum + ",B:B,\"初始\")" +
                "-SUMIFS(F:F,E:E,E" + rowNum + ",C:C,C" + rowNum + ",D:D,D" + rowNum + ",B:B,\"平\"))" +
                "/COUNTIFS(E:E,E" + rowNum + ",C:C,C" + rowNum + ",D:D,D" + rowNum + ",B:B,B" + rowNum + "),IF(B" + rowNum + "=\"开\",F" + rowNum + ",F" + rowNum + "))");

写入公示后读取公式的结果摸索了一阵,网上有两种说法:

FormulaEvaluator evaluator = new XSSFFormulaEvaluator((XSSFWorkbook)workbook);
 CellValue cellValue = evaluator.evaluate(cell1);
String res = cellValue.getStringValue();
try {
value = String.valueOf(cell.getNumericCellValue());
 } catch (IllegalStateException e) {
value = String.valueOf(cell.getRichStringCellValue());
}

尝试了之后发现都不能正确读出来。

后来无意中发现,需要手动把写入公式的excel文件打开再保存一下,就可以正常读取。保存之后,在知道类型的情况下直接使用简单语句:

row.getCell(9).getNumericCellValue();

写入excel文件:

FileOutputStream fout = null;
        Workbook workbook = null;// 创建工作薄

        try {
            workbook = createWorkBook(dir + filename);
//            workbook = createWorkBook(dir+"test123.xlsx");
//            workbook = new XSSFWorkbook();
            Sheet sheet0 = getSheetByName(workbook, sheet1Name);
            Sheet sheet1 = getSheetByName(workbook, sheet2Name);

           //       ....      //

            fout = new FileOutputStream(dir + filename);
//            fout = new FileOutputStream(dir + "test123.xlsx");
            workbook.write(fout);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if (fout != null) {
                    fout.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }