返回

Excel导出：Spring Boot 应用中的数据提取与输出

后端

  1. 引入依赖

首先，我们需要在 Spring Boot 项目中引入必要的依赖。在 pom.xml 文件中，添加以下依赖：

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

2. Excel 工具类

接下来，我们需要创建一个 Excel 工具类，用于操作和导出 Excel 文件。这个工具类的基本结构如下：

import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.List;

public class ExcelUtil {

    public static ByteArrayOutputStream exportExcel(List<Object[]> dataList, String[] headers) throws IOException {
        // 创建工作簿
        Workbook workbook = new XSSFWorkbook();

        // 创建工作表
        Sheet sheet = workbook.createSheet("数据导出");

        // 创建标题行
        Row headerRow = sheet.createRow(0);
        for (int i = 0; i < headers.length; i++) {
            Cell cell = headerRow.createCell(i);
            cell.setCellValue(headers[i]);
        }

        // 填充数据
        for (int i = 0; i < dataList.size(); i++) {
            Row dataRow = sheet.createRow(i + 1);
            for (int j = 0; j < dataList.get(i).length; j++) {
                Cell cell = dataRow.createCell(j);
                cell.setCellValue(dataList.get(i)[j].toString());
            }
        }

        // 将工作簿写入字节数组输出流
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        workbook.write(baos);

        // 关闭工作簿
        workbook.close();

        // 返回字节数组输出流
        return baos;
    }
}

3. 控制层实现

在控制层，我们需要编写一个方法来处理数据导出请求，并使用 Excel 工具类导出数据。控制器的基本结构如下：

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.List;

@RestController
public class DataExportController {

    @Autowired
    private ExcelUtil excelUtil;

    @GetMapping("/exportExcel")
    public void exportExcel(HttpServletResponse response) throws IOException {
        // 查询需要导出的数据
        List<Object[]> dataList = ...

        // 设置导出的数据表头
        String[] headers = ...

        // 使用 Excel 工具类导出数据
        ByteArrayOutputStream baos = excelUtil.exportExcel(dataList, headers);

        // 设置响应头
        response.setHeader("Content-Disposition", "attachment; filename=数据导出.xlsx");
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");

        // 将导出的数据写入响应输出流
        response.getOutputStream().write(baos.toByteArray());
    }
}

4. 前端页面

在前端页面上，我们需要集成一个导出按钮，当用户点击该按钮时，即可触发数据导出功能。前端页面的基本结构如下：

<!DOCTYPE html>
<html>
<head>
    
</head>
<body>
    <a href="/exportExcel">导出数据</a>
</body>
</html>

至此，我们就完成了 Spring Boot 中的数据导出功能的实现。当用户点击导出按钮时，控制器中的方法就会被调用，并使用 Excel 工具类导出数据。导出的数据将以 Excel 文件的形式下载到用户的本地计算机。

结语

希望这篇文章对您有所帮助。如果您有任何问题或建议，请随时与我们联系。