java下载excel文件通用工具类

原创

mob64ca12e5502a 2024-05-30 07:28:18 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e5502a的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java下载Excel文件通用工具类

在Java开发中，有时候我们需要生成并下载Excel文件。为了方便开发者快速实现这一功能，我们可以编写一个通用的工具类来处理Excel文件的生成和下载。在本文中，我们将介绍如何使用Java编写一个通用的Excel文件下载工具类，并提供代码示例。

Excel文件下载工具类设计

我们将设计一个ExcelUtils工具类，该工具类将包含生成Excel文件和下载Excel文件的方法。下面是ExcelUtils工具类的设计关系图。

erDiagram
    ExcelUtils {
        String generateExcelData()
        void downloadExcelFile()
    }

Excel文件生成方法

首先，我们需要编写一个生成Excel文件的方法。该方法将接收数据并将其写入Excel文件中。下面是一个简单的示例代码：

public class ExcelUtils {

    public static String generateExcelData(List<String> data) {
        Workbook workbook = new XSSFWorkbook();
        Sheet sheet = workbook.createSheet("Sheet1");

        int rowNum = 0;
        for (String item : data) {
            Row row = sheet.createRow(rowNum++);
            Cell cell = row.createCell(0);
            cell.setCellValue(item);
        }

        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        try {
            workbook.write(outputStream);
            workbook.close();
        } catch (IOException e) {
            e.printStackTrace();
        }

        return Base64.getEncoder().encodeToString(outputStream.toByteArray());
    }
}

在上面的示例中，我们使用Apache POI库创建一个新的Excel工作簿，并向其添加数据。最后，将工作簿写入字节数组输出流中，并将其转换为Base64编码字符串返回。

Excel文件下载方法

接下来，我们需要编写一个下载Excel文件的方法。该方法将接收Excel文件数据并将其作为附件下载。下面是一个简单的示例代码：

public class ExcelUtils {

    public static void downloadExcelFile(String base64Data, HttpServletResponse response) {
        byte[] data = Base64.getDecoder().decode(base64Data);

        try {
            response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
            response.setHeader("Content-disposition", "attachment; filename=example.xlsx");
            response.getOutputStream().write(data);
            response.getOutputStream().flush();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

在上面的示例中，我们首先将Base64编码的Excel文件数据解码为字节数组，然后设置响应的Content-Type和Content-Disposition头，最后将数据写入响应输出流中。

使用ExcelUtils工具类

现在，我们可以使用ExcelUtils工具类来生成并下载Excel文件了。下面是一个简单的示例代码：

public class Main {

    public static void main(String[] args) {
        List<String> data = Arrays.asList("Item 1", "Item 2", "Item 3");

        String base64Data = ExcelUtils.generateExcelData(data);

        HttpServletResponse response = new MockHttpServletResponse();
        ExcelUtils.downloadExcelFile(base64Data, response);
    }
}

在上面的示例中，我们创建了一个包含一些数据的列表，并使用ExcelUtils工具类生成Excel文件数据。然后，我们模拟了HttpServletResponse对象，并将Excel文件数据作为附件下载。