Java POI - 将工作簿转换为文件

Question

我需要 return 工作簿但转换为 Java File 对象。现在我只知道如何用 workbook.write(outputStream) 将工作簿写入磁盘。但是，我想 return File 对象以便直接使用它进行进一步处理（上传到亚马逊）。

我的代码：

public File addErrorColumnToExcel(MultipartFile file, HashMap<Integer, String> errors, int newColumnIndex) throws IOException {
    Workbook workbook = null;
    Sheet sheet = null;
    Row row;
    Cell newErrorCell;
    String errorText;

    try {
      workbook = new XSSFWorkbook(new BufferedInputStream(file.getInputStream()));
      sheet = workbook.getSheetAt(FIRST_SHEET_INDEX);
    } catch (IOException e) {
      logger.error("cannot get the file: " + file.getOriginalFilename() + e);
    }

    //do some logic

    return workbook; //but I need to convert this to a java File
  }

Answer 1

您可以创建一个临时文件并在 returning 之前将数据写入该文件。不要忘记在处理后删除文件。这是一个简化的片段：

public File getEmptyExcelFile() throws IOException {
    try (Workbook workbook = new XSSFWorkbook()) {

        workbook.createSheet("test"); // do some logic

        File outputFile = File.createTempFile("temp", ".xlsx");
        try (FileOutputStream fos = new FileOutputStream(outputFile)) {
            workbook.write(fos);
        }
        return outputFile;
    }
}

请注意，我对 XSSFWorkbook 和 FileOutputStream 使用了 try-with-resources。

另一个想法是 return 字节数组 byte[] 而不是 File。这是一个比较干净的解决方案，不会将文件写入磁盘。

public byte[] getEmptyExcelFileAsBytes() throws IOException {
    try (Workbook workbook = new XSSFWorkbook()) {

        workbook.createSheet("test"); // do some logic

        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        workbook.write(baos);
        return baos.toByteArray();
    }
}

Java POI - 将工作簿转换为文件

Java POI - convert Workbook to File

java

excel

file

apache-poi